FocusGram-Android/test/services/screen_time_service_test.dart

import 'package:flutter_test/flutter_test.dart';
import 'package:shared_preferences/shared_preferences.dart';
import 'package:focusgram/services/screen_time_service.dart';

void main() {
  TestWidgetsFlutterBinding.ensureInitialized();

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

  test('init loads persisted secondsByDate', () async {
    final prefs = await SharedPreferences.getInstance();
    await prefs.setString(
      ScreenTimeService.prefKey,
      '{"2026-01-01": 42, "2026-01-02": 7}',
    );

    final s = ScreenTimeService();
    await s.init();

    expect(s.secondsByDate['2026-01-01'], 42);
    expect(s.secondsByDate['2026-01-02'], 7);
  });

  test('resetAll clears stored data and in-memory map', () async {
    final prefs = await SharedPreferences.getInstance();
    await prefs.setString(ScreenTimeService.prefKey, '{"2026-01-01": 42}');

    final s = ScreenTimeService();
    await s.init();
    expect(s.secondsByDate.isNotEmpty, isTrue);

    await s.resetAll();
    expect(s.secondsByDate, isEmpty);

    final raw = prefs.getString(ScreenTimeService.prefKey);
    expect(raw, isNull);
  });

  test(
    'startTracking increments today seconds and stopTracking persists',
    () async {
      final s = ScreenTimeService();
      await s.init();

      final beforeTodayKey = DateTime.now();
      final todayKey =
          '${beforeTodayKey.year.toString().padLeft(4, '0')}-'
          '${beforeTodayKey.month.toString().padLeft(2, '0')}-'
          '${beforeTodayKey.day.toString().padLeft(2, '0')}';

      s.startTracking();

      // Wait ~2 seconds (test is unit-ish; still acceptable).
      await Future<void>.delayed(const Duration(seconds: 2));

      s.stopTracking();

      expect(s.secondsByDate[todayKey], isNotNull);
      expect(s.secondsByDate[todayKey]!, greaterThanOrEqualTo(2));

      final prefs = await SharedPreferences.getInstance();
      final stored = prefs.getString(ScreenTimeService.prefKey);
      expect(stored, isNotNull);
      expect(stored, contains(todayKey));
    },
  );
}