drift/drift_dev/test/analysis/results/dart_test.dart

import 'package:drift_dev/src/analysis/options.dart';
import 'package:drift_dev/src/analysis/results/dart.dart';
import 'package:drift_dev/src/writer/writer.dart';
import 'package:test/test.dart';

import '../test_utils.dart';

void main() {
  late TestBackend tester;

  setUpAll(() async => tester = await TestBackend.init({}));
  tearDownAll(() => tester.dispose());

  group('from AST', () {
    final testUri = Uri.parse('package:a/test.dart');

    Future<void> checkTransformation(String sourceExpression,
        String expectedResult, Map<String, String> expectedImports) async {
      final expression =
          await tester.resolveExpression(testUri, sourceExpression, const []);
      final annotated = AnnotatedDartCode.ast(expression);

      final imports = TestImportManager();
      final writer = Writer(
        const DriftOptions.defaults(),
        generationOptions: GenerationOptions(
          imports: imports,
        ),
      );

      final code = writer.dartCode(annotated);

      expect(code, expectedResult);
      expectedImports.forEach((alias, import) {
        expect(imports.importAliases[Uri.parse(import)], alias);
      });
    }

    test('constructor invocation', () async {
      await checkTransformation('const Duration(seconds: 12)',
          'const i0.Duration(seconds: 12)', {'i0': 'dart:core'});
    });

    test('static invocation', () async {
      await checkTransformation(
          'Uri.parse("")', 'i0.Uri.parse("")', {'i0': 'dart:core'});
    });

    test('explicit extension invocation', () async {
      await checkTransformation(
          'IterableExtensions<String>([]).firstOrNull',
          'i0.IterableExtensions<i1.String>([]).firstOrNull',
          {'i0': 'dart:collection', 'i1': 'dart:core'});
    });
  });
}