English/kouyu_english/test/navigation_back_button_test.dart

import 'package:flutter/material.dart';
import 'package:flutter_test/flutter_test.dart';
import 'package:kouyu_english/core/app_state.dart';
import 'package:kouyu_english/core/generated_content.dart';
import 'package:kouyu_english/core/models.dart';
import 'package:kouyu_english/features/dialogue/dialogue_flow.dart';
import 'package:kouyu_english/features/review/review_page.dart';
import 'package:kouyu_english/features/shell/learning_shell.dart';
import 'package:shared_preferences/shared_preferences.dart';

import 'support/app_assets.dart';

void main() {
  setUp(() {
    SharedPreferences.setMockInitialValues({});
  });

  testWidgets('Onboarding PlacementPage back button returns to WelcomePage', (
    tester,
  ) async {
    await pumpLoadedApp(tester);
    await tester.pumpAndSettle();

    expect(find.text('每天 20 分钟，\n说出能用的英语。'), findsOneWidget);
    await tester.tap(find.text('继续'));
    await tester.pumpAndSettle();

    expect(find.text('从哪里开始？'), findsOneWidget);
    expect(find.byIcon(Icons.arrow_back), findsOneWidget);

    await tester.tap(find.byIcon(Icons.arrow_back));
    await tester.pumpAndSettle();

    expect(find.text('每天 20 分钟，\n说出能用的英语。'), findsOneWidget);
  });

  testWidgets('Profile sub-page back button returns to 我的', (tester) async {
    final state = AppState()..finishOnboarding();
    await tester.pumpWidget(MaterialApp(home: LearningShell(state: state)));
    await tester.pumpAndSettle();

    await tester.tap(find.text('我的'));
    await tester.pumpAndSettle();

    final preferences = find.text('学习偏好');
    await tester.ensureVisible(preferences);
    await tester.tap(preferences);
    await tester.pumpAndSettle();

    expect(find.text('每日学习时间'), findsOneWidget);
    await tester.tap(find.byType(BackButton));
    await tester.pumpAndSettle();

    expect(find.text('每日学习时间'), findsNothing);
    expect(find.text('学习偏好'), findsOneWidget);
  });

  testWidgets('LessonFlow back button returns to tab', (tester) async {
    final state = AppState()..finishOnboarding();
    await tester.pumpWidget(MaterialApp(home: LearningShell(state: state)));
    await tester.pumpAndSettle();

    expect(find.text('开始今天的学习'), findsOneWidget);
    await tester.tap(find.text('开始今天的学习'));
    await tester.pumpAndSettle();

    // In LessonFlow
    expect(find.byIcon(Icons.arrow_back), findsOneWidget);

    // Tap back button
    await tester.tap(find.byIcon(Icons.arrow_back));
    await tester.pumpAndSettle();

    expect(find.text('开始今天的学习'), findsOneWidget);
  });

  testWidgets('学习 tab shows today task and the lesson path', (tester) async {
    final state = AppState()..finishOnboarding();
    await tester.pumpWidget(MaterialApp(home: LearningShell(state: state)));
    await tester.pumpAndSettle();

    expect(find.byType(NavigationDestination), findsNWidgets(4));
    expect(find.text('今日新课'), findsOneWidget);
    expect(find.text('课程路径'), findsOneWidget);
    expect(find.textContaining('第 1 课 · '), findsWidgets);
  });

  testWidgets('DialogueSummaryPage has back button', (tester) async {
    var backCalled = false;
    await tester.pumpWidget(
      MaterialApp(
        home: DialogueSummaryPage(
          summary: const DialogueSummaryData(
            completedTasks: ['介绍姓名'],
            personalSentence: 'My name is Shen.',
            usedHelp: false,
          ),
          onHome: () {},
          onLesson: () {},
          onRetry: () {},
          onBack: () => backCalled = true,
        ),
      ),
    );
    await tester.pumpAndSettle();

    expect(find.text('对话完成'), findsNWidgets(2)); // AppBar title and Eyebrow
    expect(find.byIcon(Icons.arrow_back), findsOneWidget);

    await tester.tap(find.byIcon(Icons.arrow_back));
    await tester.pumpAndSettle();

    expect(backCalled, isTrue);
  });

  testWidgets('AdaptiveLessonPage has back button in all states', (
    tester,
  ) async {
    final state = AppState()..finishOnboarding();

    // 1. Empty state
    var finished = false;
    await tester.pumpWidget(
      MaterialApp(
        home: AdaptiveLessonPage(
          state: state,
          onFinished: () => finished = true,
        ),
      ),
    );
    await tester.pumpAndSettle();

    expect(find.text('AI 四技能补练'), findsNWidgets(2));
    expect(find.byIcon(Icons.arrow_back), findsOneWidget);
    await tester.tap(find.byIcon(Icons.arrow_back));
    expect(finished, isTrue);

    // 2. Active lesson state
    state.cacheApprovedAdaptiveLesson(
      const GeneratedLesson(
        lessonId: 'adapt-1',
        revision: 1,
        stageVersion: 'A0',
        abilityIds: ['greeting'],
        prerequisiteIds: [],
        targetItemIds: ['name'],
        receptiveChunks: ['My name is Mia.'],
        previewItemIds: ['name'],
        estimatedMinutes: 5,
        tasks: [
          GeneratedLessonTask(
            taskId: 't1',
            skill: 'listening',
            type: 'listen',
            prompt: '听并写出名字',
            stimulus: 'My name is Mia.',
            answer: 'Mia',
            targetItemIds: ['name'],
          ),
        ],
      ),
      auditor: 'test',
    );

    finished = false;
    await tester.pumpWidget(
      MaterialApp(
        home: AdaptiveLessonPage(
          state: state,
          onFinished: () => finished = true,
        ),
      ),
    );
    await tester.pumpAndSettle();

    expect(find.byIcon(Icons.arrow_back), findsOneWidget);
    await tester.tap(find.byIcon(Icons.arrow_back));
    expect(finished, isTrue);
  });

  testWidgets(
    'AssessmentPreparationPage and AssessmentPage have back buttons',
    (tester) async {
      final state = AppState()..finishOnboarding();
      await tester.pumpWidget(MaterialApp(home: LearningShell(state: state)));
      await tester.pumpAndSettle();

      // Navigate to "我的"
      await tester.tap(find.text('我的'));
      await tester.pumpAndSettle();

      final assessments = find.text('四技能评估');
      await tester.ensureVisible(assessments);
      await tester.tap(assessments);
      await tester.pumpAndSettle();

      // Open assessment pack A0-E1
      final packAButton = find.text('开始 A0-E1');
      await tester.ensureVisible(packAButton);
      await tester.pumpAndSettle();

      expect(packAButton, findsOneWidget);
      await tester.tap(packAButton);
      await tester.pumpAndSettle();

      // In AssessmentPreparationPage
      expect(find.text('评估准备'), findsOneWidget);
      expect(find.byIcon(Icons.arrow_back), findsOneWidget);

      // Start assessment
      await tester.tap(find.text('开始评估'));
      await tester.pumpAndSettle();

      // In AssessmentPage
      expect(find.byIcon(Icons.arrow_back), findsOneWidget);

      // Returns to 我的
      await tester.tap(find.byIcon(Icons.arrow_back));
      await tester.pumpAndSettle();
      expect(find.text('评估准备'), findsNothing);
      expect(find.text('四技能评估'), findsOneWidget);
    },
  );
}