pkgs/code_builder/test/e2e/injection_test.dart

// Copyright (c) 2017, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:code_builder/code_builder.dart';
import 'package:test/test.dart';

import '../common.dart';

void main() {
  useDartfmt();

  test('should generate a complex generated file', () {
    // Imports from an existing Dart library.
    final $App = refer('App', 'package:app/app.dart');
    final $Module = refer('Module', 'package:app/module.dart');
    final $Thing = refer('Thing', 'package:app/thing.dart');

    final clazz = ClassBuilder()
      ..name = 'Injector'
      ..implements.add($App)
      ..fields.add(
        Field(
          (b) => b
            ..modifier = FieldModifier.final$
            ..name = '_module'
            ..type = $Module.type,
        ),
      )
      ..constructors.add(
        Constructor(
          (b) => b
            ..requiredParameters.add(
              Parameter(
                (b) => b
                  ..name = '_module'
                  ..toThis = true,
              ),
            ),
        ),
      )
      ..methods.add(
        Method(
          (b) => b
            ..name = 'getThing'
            ..body = $Thing.newInstance([
              refer('_module').property('get1').call([]),
              refer('_module').property('get2').call([]),
            ]).code
            ..returns = $Thing
            ..annotations.add(refer('override')),
        ),
      );

    expect(
      clazz.build(),
      equalsDart(r'''
        class Injector implements App {
          Injector(this._module);

          final Module _module;

          @override
          Thing getThing() => Thing(_module.get1(), _module.get2(), );
        }
      '''),
    );

    expect(
      clazz.build(),
      equalsDart(r'''
        class Injector implements _i1.App {
          Injector(this._module);

          final _i2.Module _module;

          @override
          _i3.Thing getThing() => _i3.Thing(_module.get1(), _module.get2(), );
        }
      ''', DartEmitter(allocator: Allocator.simplePrefixing())),
    );
  });
}