FocusGram-Android/test/services/dm_ghost_injection_test.dart

import 'package:flutter_test/flutter_test.dart';
import 'package:flutter_inappwebview/flutter_inappwebview.dart';
import 'package:focusgram/focus_settings.dart';
import 'package:focusgram/scripts/focus_scripts.dart';

void main() {
  group('FocusSettings — Field cleanup', () {
    test('only ghostMode remains (fullDmGhost and storyGhost removed)',
        () async {
      const settings = FocusSettings(ghostMode: true);

      expect(settings.ghostMode, isTrue);
      expect(settings.noAds, isTrue);
      expect(settings.noStories, isFalse);
      expect(settings.noReels, isFalse);
      expect(settings.noAutoplay, isFalse);
      expect(settings.noDMs, isFalse);

      // Verify fullDmGhost and storyGhost are NOT fields anymore
      // (these would be compile errors if they existed)
    });

    test('default ghostMode is false', () async {
      const settings = FocusSettings();
      expect(settings.ghostMode, isFalse);
    });
  });

  group('buildUserScripts — DM Ghost injection', () {
    test('injects kFullDmGhostJS when ghostMode is true', () async {
      final scripts = buildUserScripts(const FocusSettings(ghostMode: true));

      expect(scripts.length, equals(1));
      expect(
        scripts[0].injectionTime,
        equals(UserScriptInjectionTime.AT_DOCUMENT_START),
      );

      // Verify the comprehensive Full DM ghost JS is injected
      final src = scripts[0].source;
      expect(src, contains('__fgFullDmGhost=true'));
      expect(src, contains('__fgFullDmGhostPatched'));
      expect(src, contains('shouldBlockDmPath'));
      expect(src, contains('DM_URLS'));
      expect(src, contains('DM_OPS'));
      expect(src, contains('serviceWorker'));
      expect(src, contains('sendBeacon'));
    });

    test('does NOT inject ghost scripts when ghostMode is false', () async {
      final scripts =
          buildUserScripts(const FocusSettings(ghostMode: false));

      // Should have no DOCUMENT_START scripts
      final startScripts =
          scripts.where((s) =>
              s.injectionTime == UserScriptInjectionTime.AT_DOCUMENT_START);
      for (final s in startScripts) {
        expect(s.source.contains('__fgFullDmGhost'), isFalse);
      }
    });

    test('injects noAutoplay alongside DM Ghost', () async {
      final scripts = buildUserScripts(
        const FocusSettings(ghostMode: true, noAutoplay: true),
      );

      final startScripts =
          scripts.where((s) =>
              s.injectionTime == UserScriptInjectionTime.AT_DOCUMENT_START);
      expect(startScripts.length, equals(1));
      expect(startScripts.first.source, contains('__fgFullDmGhost=true'));
      expect(startScripts.first.source, contains('document.addEventListener'));
    });

    test('injects hideStoryTray at DOCUMENT_END when noStories is true',
        () async {
      final scripts = buildUserScripts(
        const FocusSettings(noStories: true),
      );

      final endScripts =
          scripts.where((s) =>
              s.injectionTime == UserScriptInjectionTime.AT_DOCUMENT_END);
      expect(endScripts.length, equals(1));
      expect(
        endScripts.first.source,
        contains('[data-pagelet="story_tray"]'),
      );
    });
  });
}