sharedinbox/test/unit/undo_service_test.dart

import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:flutter_test/flutter_test.dart';
import 'package:mockito/annotations.dart';
import 'package:mockito/mockito.dart';
import 'package:sharedinbox/core/models/email.dart';
import 'package:sharedinbox/core/models/undo_action.dart';
import 'package:sharedinbox/core/repositories/email_repository.dart';
import 'package:sharedinbox/core/repositories/undo_repository.dart';
import 'package:sharedinbox/di.dart';

import 'undo_service_test.mocks.dart';

@GenerateMocks([EmailRepository, UndoRepository])
void main() {
  late ProviderContainer container;
  late MockEmailRepository mockEmailRepo;
  late MockUndoRepository mockUndoRepo;

  setUp(() {
    mockEmailRepo = MockEmailRepository();
    mockUndoRepo = MockUndoRepository();

    when(mockUndoRepo.saveAction(any)).thenAnswer((_) async {});
    when(mockUndoRepo.deleteAction(any)).thenAnswer((_) async {});
    when(
      mockUndoRepo.getHistory(limit: anyNamed('limit')),
    ).thenAnswer((_) async => []);

    container = ProviderContainer(
      overrides: [
        emailRepositoryProvider.overrideWithValue(mockEmailRepo),
        undoRepositoryProvider.overrideWithValue(mockUndoRepo),
      ],
    );
  });

  tearDown(() {
    container.dispose();
  });

  test('UndoService initial state is empty list', () {
    expect(container.read(undoServiceProvider), isEmpty);
  });

  test('pushAction maintains history and updates state', () async {
    final action1 = UndoAction(
      id: '1',
      accountId: 'acc1',
      type: UndoType.move,
      emailIds: ['e1'],
      sourceMailboxPath: 'INBOX',
    );
    final action2 = UndoAction(
      id: '2',
      accountId: 'acc1',
      type: UndoType.delete,
      emailIds: ['e2'],
      sourceMailboxPath: 'INBOX',
    );

    final notifier = container.read(undoServiceProvider.notifier);
    await notifier.init(); // Wait for persistent load

    await notifier.pushAction(action1);
    expect(container.read(undoServiceProvider), [action1]);

    await notifier.pushAction(action2);
    expect(container.read(undoServiceProvider), [action1, action2]);
  });

  test('undo pops history and updates state', () async {
    final action1 = UndoAction(
      id: '1',
      accountId: 'acc1',
      type: UndoType.move,
      emailIds: ['e1'],
      sourceMailboxPath: 'INBOX',
    );
    final action2 = UndoAction(
      id: '2',
      accountId: 'acc1',
      type: UndoType.delete,
      emailIds: ['e2'],
      sourceMailboxPath: 'INBOX',
    );

    when(mockEmailRepo.moveEmail(any, any)).thenAnswer((_) async {});
    when(
      mockEmailRepo.cancelPendingChange(any, any),
    ).thenAnswer((_) async => false);

    final notifier = container.read(undoServiceProvider.notifier);
    await notifier.init();
    await notifier.pushAction(action1);
    await notifier.pushAction(action2);

    await notifier.undo();
    expect(container.read(undoServiceProvider), [action1]);
    verify(mockEmailRepo.moveEmail('e2', 'INBOX')).called(1);

    await notifier.undo();
    expect(container.read(undoServiceProvider), isEmpty);
    verify(mockEmailRepo.moveEmail('e1', 'INBOX')).called(1);
  });

  test('undo with actionId removes and undos specific action', () async {
    final action1 = UndoAction(
      id: '1',
      accountId: 'acc1',
      type: UndoType.move,
      emailIds: ['e1'],
      sourceMailboxPath: 'INBOX',
    );
    final action2 = UndoAction(
      id: '2',
      accountId: 'acc1',
      type: UndoType.delete,
      emailIds: ['e2'],
      sourceMailboxPath: 'INBOX',
    );

    when(mockEmailRepo.moveEmail(any, any)).thenAnswer((_) async {});
    when(
      mockEmailRepo.cancelPendingChange(any, any),
    ).thenAnswer((_) async => false);

    final notifier = container.read(undoServiceProvider.notifier);
    await notifier.init();
    await notifier.pushAction(action1);
    await notifier.pushAction(action2);

    await notifier.undo(actionId: '1');
    expect(container.read(undoServiceProvider), [action2]);
    verify(mockEmailRepo.moveEmail('e1', 'INBOX')).called(1);
    verifyNever(mockEmailRepo.moveEmail('e2', 'INBOX'));
  });

  test('undo performs cancelPendingChange optimization', () async {
    final action = UndoAction(
      id: '1',
      accountId: 'acc1',
      type: UndoType.move,
      emailIds: ['e1'],
      sourceMailboxPath: 'INBOX',
    );

    when(mockEmailRepo.moveEmail(any, any)).thenAnswer((_) async {});
    when(
      mockEmailRepo.cancelPendingChange('e1', 'delete'),
    ).thenAnswer((_) async => false);
    when(
      mockEmailRepo.cancelPendingChange('e1', 'move'),
    ).thenAnswer((_) async => true);

    final notifier = container.read(undoServiceProvider.notifier);
    await notifier.init();
    await notifier.pushAction(action);

    await notifier.undo();
    verify(mockEmailRepo.moveEmail('e1', 'INBOX')).called(1);
    verify(mockEmailRepo.cancelPendingChange('e1', 'move')).called(2);
  });

  test('undo restores hard-deleted emails if provided', () async {
    final email = Email(
      id: 'e1',
      accountId: 'acc1',
      mailboxPath: 'Trash',
      uid: 10,
      from: [const EmailAddress(email: 'me@me.com')],
      to: [],
      cc: [],
      subject: 'hi',
      receivedAt: DateTime.now(),
      isSeen: true,
      isFlagged: false,
      hasAttachment: false,
    );
    final action = UndoAction(
      id: '1',
      accountId: 'acc1',
      type: UndoType.delete,
      emailIds: ['e1'],
      sourceMailboxPath: 'INBOX',
      originalEmails: [email],
    );

    when(
      mockEmailRepo.cancelPendingChange(any, any),
    ).thenAnswer((_) async => false);
    when(mockEmailRepo.restoreEmails(any)).thenAnswer((_) async {});
    when(mockEmailRepo.moveEmail(any, any)).thenAnswer((_) async {});

    final notifier = container.read(undoServiceProvider.notifier);
    await notifier.init();
    await notifier.pushAction(action);

    await notifier.undo();

    verify(mockEmailRepo.restoreEmails(any)).called(1);
    verify(mockEmailRepo.moveEmail('e1', 'INBOX')).called(1);
  });

  test('init loads persisted history from repository', () async {
    final persisted = UndoAction(
      id: '99',
      accountId: 'acc1',
      type: UndoType.move,
      emailIds: ['e99'],
      sourceMailboxPath: 'INBOX',
    );

    when(
      mockUndoRepo.getHistory(limit: anyNamed('limit')),
    ).thenAnswer((_) async => [persisted]);

    final notifier = container.read(undoServiceProvider.notifier);
    await notifier.init();

    expect(container.read(undoServiceProvider), [persisted]);
  });

  test('pushAction after restart appends to persisted history', () async {
    final persisted = UndoAction(
      id: '1',
      accountId: 'acc1',
      type: UndoType.move,
      emailIds: ['e1'],
      sourceMailboxPath: 'INBOX',
    );
    final newAction = UndoAction(
      id: '2',
      accountId: 'acc1',
      type: UndoType.delete,
      emailIds: ['e2'],
      sourceMailboxPath: 'INBOX',
    );

    when(
      mockUndoRepo.getHistory(limit: anyNamed('limit')),
    ).thenAnswer((_) async => [persisted]);

    final notifier = container.read(undoServiceProvider.notifier);
    await notifier.init();
    await notifier.pushAction(newAction);

    expect(container.read(undoServiceProvider), [persisted, newAction]);
  });

  test('pushAction concurrent with init waits for init to complete', () async {
    final persisted = UndoAction(
      id: '1',
      accountId: 'acc1',
      type: UndoType.move,
      emailIds: ['e1'],
      sourceMailboxPath: 'INBOX',
    );
    final raced = UndoAction(
      id: '2',
      accountId: 'acc1',
      type: UndoType.delete,
      emailIds: ['e2'],
      sourceMailboxPath: 'INBOX',
    );

    // Simulate slow DB load
    when(
      mockUndoRepo.getHistory(limit: anyNamed('limit')),
    ).thenAnswer(
      (_) => Future.delayed(
        const Duration(milliseconds: 10),
        () => [persisted],
      ),
    );

    final notifier = container.read(undoServiceProvider.notifier);
    final initFuture = notifier.init();
    // pushAction issued before init completes — it must still see persisted history
    final pushFuture = notifier.pushAction(raced);

    await Future.wait([initFuture, pushFuture]);

    expect(container.read(undoServiceProvider), [persisted, raced]);
  });
}