immich/mobile/test/unit/presentation/partner_page_test.dart

import 'package:flutter/material.dart';
import 'package:flutter_test/flutter_test.dart';
import 'package:hooks_riverpod/hooks_riverpod.dart';
import 'package:immich_mobile/domain/models/user.model.dart';
import 'package:immich_mobile/pages/library/partner/partner.page.dart';

import '../factories/partner_user_factory.dart';
import '../factories/user_factory.dart';
import '../presentation_context.dart';

void main() {
  late PresentationContext context;

  setUp(() async => context = await PresentationContext.create());
  tearDown(() async => await context.dispose());

  group('PartnerSharedByList', () {
    testWidgets('shows the empty-state add button when there are no partners', (tester) async {
      await tester.pumpTestWidget(PartnerSharedByList(partners: const [], onAdd: () {}, onRemove: (_) {}));

      expect(find.byType(ListView), findsNothing);
      expect(find.widgetWithIcon(ElevatedButton, Icons.person_add), findsOneWidget);
    });

    testWidgets('renders a tile per partner with name and email', (tester) async {
      final partner1 = PartnerFactory.create();
      final partner2 = PartnerFactory.create();
      await tester.pumpTestWidget(PartnerSharedByList(partners: [partner1, partner2], onAdd: () {}, onRemove: (_) {}));

      expect(find.byType(ListTile), findsNWidgets(2));
      expect(find.text(partner1.name), findsOneWidget);
      expect(find.text(partner1.email), findsOneWidget);
      expect(find.text(partner2.name), findsOneWidget);
      expect(find.text(partner2.email), findsOneWidget);
    });

    testWidgets('invokes onRemovePartner with the tapped partner', (tester) async {
      final partner1 = PartnerFactory.create(inTimeline: true);
      final partner2 = PartnerFactory.create();
      Partner? removed;
      await tester.pumpTestWidget(
        PartnerSharedByList(partners: [partner1, partner2], onAdd: () {}, onRemove: (p) => removed = p),
      );

      await tester.tap(find.byIcon(Icons.person_remove).first);
      await tester.pump();

      expect(removed, partner1);
    });
  });

  group('PartnerSelectionDialog', () {
    final dialogButtonKey = UniqueKey();

    Widget dialogWidget({void Function(User?)? onClosed}) {
      return Builder(
        builder: (context) => ElevatedButton(
          onPressed: () async {
            final selected = await showDialog<User>(context: context, builder: (_) => const PartnerSelectionDialog());
            onClosed?.call(selected);
          },
          child: Text(key: dialogButtonKey, 'open'),
        ),
      );
    }

    List<Override> withCandidates(List<User> candidates) => [
      candidatesStateProvider.overrideWith((ref) => Stream<Iterable<User>>.value(candidates)),
    ];

    testWidgets('renders an option per candidate fetched from the provider', (tester) async {
      final user = UserFactory.create();
      await tester.pumpTestWidget(dialogWidget(), overrides: withCandidates([user]));

      await tester.tap(find.byKey(dialogButtonKey));
      await tester.pumpAndSettle();

      expect(find.byType(SimpleDialogOption), findsOneWidget);
      expect(find.text(user.name), findsOneWidget);
    });

    testWidgets('shows no options when the provider returns no candidates', (tester) async {
      await tester.pumpTestWidget(dialogWidget(), overrides: withCandidates(const []));

      await tester.tap(find.byKey(dialogButtonKey));
      await tester.pumpAndSettle();

      expect(find.byType(SimpleDialogOption), findsNothing);
    });

    testWidgets('pops the selected candidate when an option is tapped', (tester) async {
      final user = UserFactory.create();
      User? selected;
      await tester.pumpTestWidget(dialogWidget(onClosed: (user) => selected = user), overrides: withCandidates([user]));

      await tester.tap(find.byKey(dialogButtonKey));
      await tester.pumpAndSettle();

      await tester.tap(find.text(user.name));
      await tester.pumpAndSettle();

      expect(selected, user);
    });
  });
}