fluffychat/test/pangea/span_data_model_test.dart

import 'package:flutter_test/flutter_test.dart';

import 'package:fluffychat/pangea/choreographer/igc/replacement_type_enum.dart';
import 'package:fluffychat/pangea/choreographer/igc/span_data_model.dart';

void main() {
  test('SpanData.fromJson handles legacy correction type (maps to grammar)',
      () {
    final Map<String, dynamic> legacyJson = {
      'message': null,
      'short_message': null,
      'choices': <dynamic>[],
      'offset': 0,
      'length': 4,
      'full_text': 'Test',
      'type': {
        'type_name': 'correction',
      },
    };

    expect(() => SpanData.fromJson(legacyJson), returnsNormally);
    final SpanData span = SpanData.fromJson(legacyJson);
    // 'correction' is mapped to 'grammar' for backward compatibility
    expect(span.type, ReplacementTypeEnum.subjectVerbAgreement);
  });

  test('SpanData.fromJson handles legacy typeName object', () {
    final Map<String, dynamic> legacyJson = {
      'message': null,
      'short_message': null,
      'choices': <dynamic>[],
      'offset': 0,
      'length': 4,
      'full_text': 'Test',
      'type': {
        'typeName': 'itStart',
      },
    };

    expect(() => SpanData.fromJson(legacyJson), returnsNormally);
    final SpanData span = SpanData.fromJson(legacyJson);
    expect(span.type, ReplacementTypeEnum.itStart);
  });

  test('SpanData.fromJson handles did_you_mean string', () {
    final Map<String, dynamic> jsonData = {
      'message': null,
      'short_message': null,
      'choices': <dynamic>[],
      'offset': 0,
      'length': 4,
      'full_text': 'Test',
      'type': 'did_you_mean',
    };

    expect(() => SpanData.fromJson(jsonData), returnsNormally);
    final SpanData span = SpanData.fromJson(jsonData);
    expect(span.type, ReplacementTypeEnum.didYouMean);
  });

  test('SpanData.fromJson handles legacy vocabulary type (maps to wordChoice)',
      () {
    final Map<String, dynamic> legacyJson = {
      'message': null,
      'short_message': null,
      'choices': <dynamic>[],
      'offset': 0,
      'length': 4,
      'full_text': 'Test',
      'type': 'vocabulary',
    };

    expect(() => SpanData.fromJson(legacyJson), returnsNormally);
    final SpanData span = SpanData.fromJson(legacyJson);
    expect(span.type, ReplacementTypeEnum.other);
  });

  test('SpanData.fromJson handles new grammar type directly', () {
    final Map<String, dynamic> jsonData = {
      'message': null,
      'short_message': null,
      'choices': <dynamic>[],
      'offset': 0,
      'length': 4,
      'full_text': 'Test',
      'type': 'grammar',
    };

    expect(() => SpanData.fromJson(jsonData), returnsNormally);
    final SpanData span = SpanData.fromJson(jsonData);
    expect(span.type, ReplacementTypeEnum.subjectVerbAgreement);
  });

  test('SpanData.fromJson handles translation type', () {
    final Map<String, dynamic> jsonData = {
      'message': null,
      'short_message': null,
      'choices': <dynamic>[],
      'offset': 0,
      'length': 4,
      'full_text': 'Test',
      'type': 'translation',
    };

    expect(() => SpanData.fromJson(jsonData), returnsNormally);
    final SpanData span = SpanData.fromJson(jsonData);
    expect(span.type, ReplacementTypeEnum.translation);
  });

  group('SpanData.fromJson fullText fallback', () {
    test('uses full_text from JSON when present', () {
      final Map<String, dynamic> jsonData = {
        'message': null,
        'short_message': null,
        'choices': <dynamic>[],
        'offset': 0,
        'length': 4,
        'full_text': 'Text from span',
        'type': 'grammar',
      };

      final SpanData span = SpanData.fromJson(
        jsonData,
        parentFullText: 'Text from parent',
      );
      expect(span.fullText, 'Text from span');
    });

    test('uses parentFullText when full_text not in JSON', () {
      final Map<String, dynamic> jsonData = {
        'message': null,
        'short_message': null,
        'choices': <dynamic>[],
        'offset': 0,
        'length': 4,
        // Note: no full_text field
        'type': 'grammar',
      };

      final SpanData span = SpanData.fromJson(
        jsonData,
        parentFullText: 'Text from parent',
      );
      expect(span.fullText, 'Text from parent');
    });

    test('uses empty string when neither full_text nor parentFullText present',
        () {
      final Map<String, dynamic> jsonData = {
        'message': null,
        'short_message': null,
        'choices': <dynamic>[],
        'offset': 0,
        'length': 4,
        'type': 'grammar',
      };

      final SpanData span = SpanData.fromJson(jsonData);
      expect(span.fullText, '');
    });

    test('prefers sentence over full_text (legacy field name)', () {
      final Map<String, dynamic> jsonData = {
        'message': null,
        'short_message': null,
        'choices': <dynamic>[],
        'offset': 0,
        'length': 4,
        'sentence': 'Text from sentence field',
        'full_text': 'Text from full_text field',
        'type': 'grammar',
      };

      final SpanData span = SpanData.fromJson(jsonData);
      expect(span.fullText, 'Text from sentence field');
    });
  });
}