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),
      ),
    );
  });
}