@Tags(['analyzer'])

import 'package:analyzer/dart/ast/ast.dart';
import 'package:analyzer/dart/element/element.dart';
import 'package:build/build.dart';
import 'package:drift_dev/src/analyzer/dart/parser.dart';
import 'package:drift_dev/src/analyzer/runner/steps.dart';
import 'package:drift_dev/src/analyzer/session.dart';
import 'package:test/test.dart';

import '../../utils/test_backend.dart';

void main() {
  test('return expression of methods', () async {
    final backend = TestBackend({
      AssetId.parse('test_lib|lib/main.dart'): r'''
      class Test {
        String get getter => 'foo';
        String function() => 'bar';
        String invalid() {
         return 'baz';
        }
      }
    '''
    });

    final input = Uri.parse('package:test_lib/main.dart');
    final session = MoorSession(backend);
    final backendTask = backend.startTask(input);
    final task = session.startTask(backendTask);

    final library = await backendTask.resolveDart(input);
    final parser = DriftDartParser(ParseDartStep(
        task, session.registerFile(input), library, await task.helper));

    Future<MethodDeclaration> _loadDeclaration(Element element) async {
      final node = await parser.loadElementDeclaration(element);
      return node as MethodDeclaration;
    }

    Future<void> _verifyReturnExpressionMatches(
        Element element, String source) async {
      final node = await _loadDeclaration(element);
      expect(parser.returnExpressionOfMethod(node)!.toSource(), source);
    }

    final testClass = library.getType('Test');

    await _verifyReturnExpressionMatches(
        testClass!.getGetter('getter')!, "'foo'");
    await _verifyReturnExpressionMatches(
        testClass.getMethod('function')!, "'bar'");

    final invalidDecl = await _loadDeclaration(testClass.getMethod('invalid')!);
    expect(parser.returnExpressionOfMethod(invalidDecl), isNull);
    expect(parser.step.errors.errors, isNotEmpty);

    backend.finish();
  });
}