pkgs/code_builder/test/directive_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();

  final $LinkedHashMap = refer('LinkedHashMap', 'dart:collection');

  final library = Library(
    (b) => b
      ..directives.add(Directive.export('../relative.dart'))
      ..directives.add(Directive.export('package:foo/foo.dart'))
      ..directives.add(Directive.part('lib.g.dart'))
      ..body.add(
        Field(
          (b) => b
            ..name = 'relativeRef'
            ..modifier = FieldModifier.final$
            ..assignment = refer(
              'Relative',
              '../relative.dart',
            ).newInstance([]).code,
        ),
      )
      ..body.add(
        Field(
          (b) => b
            ..name = 'pkgRefFoo'
            ..modifier = FieldModifier.final$
            ..assignment = refer(
              'Foo',
              'package:foo/foo.dart',
            ).newInstance([]).code,
        ),
      )
      ..body.add(
        Field(
          (b) => b
            ..name = 'pkgRefBar'
            ..modifier = FieldModifier.final$
            ..assignment = refer(
              'Bar',
              'package:foo/bar.dart',
            ).newInstance([]).code,
        ),
      )
      ..body.add(
        Field(
          (b) => b
            ..name = 'collectionRef'
            ..modifier = FieldModifier.final$
            ..assignment = $LinkedHashMap.newInstance([]).code,
        ),
      ),
  );

  test('should emit a source file with imports in defined order', () {
    expect(
      library,
      equalsDart(r'''
          // ignore_for_file: no_leading_underscores_for_library_prefixes

          import '../relative.dart' as _i1;
          import 'package:foo/foo.dart' as _i2;
          import 'package:foo/bar.dart' as _i3;
          import 'dart:collection' as _i4;
          export '../relative.dart';
          export 'package:foo/foo.dart';
          part 'lib.g.dart';

          final relativeRef = _i1.Relative();
          final pkgRefFoo = _i2.Foo();
          final pkgRefBar = _i3.Bar();
          final collectionRef = _i4.LinkedHashMap();''', DartEmitter.scoped()),
    );
  });

  test('should emit a source file with ordered', () {
    expect(
      library,
      equalsDart(
        r'''
          // ignore_for_file: no_leading_underscores_for_library_prefixes

          import 'dart:collection' as _i4;

          import 'package:foo/bar.dart' as _i3;
          import 'package:foo/foo.dart' as _i2;

          import '../relative.dart' as _i1;

          export 'package:foo/foo.dart';
          export '../relative.dart';

          part 'lib.g.dart';

          final relativeRef = _i1.Relative();
          final pkgRefFoo = _i2.Foo();
          final pkgRefBar = _i3.Bar();
          final collectionRef = _i4.LinkedHashMap();''',
        DartEmitter.scoped(orderDirectives: true),
      ),
    );
  });
}