pkgs/code_builder/test/specs/code/statement_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 emit a block of code', () {
    expect(
      Block.of([
        const Code('if (foo) {'),
        const Code('  print(true);'),
        const Code('}'),
      ]),
      equalsDart(r'''
        if (foo) {
          print(true);
        }
      '''),
    );
  });

  test('should emit a block of code including expressions', () {
    expect(
      Block.of([
        const Code('if (foo) {'),
        refer('print')([literalTrue]).statement,
        const Code('}'),
      ]),
      equalsDart(r'''
        if (foo) {
          print(true);
        }
      '''),
    );
  });

  test('should emit a block of code with lazily invoked generators', () {
    expect(
      Method((b) => b
        ..name = 'main'
        ..body = Block.of([
          const Code('if ('),
          lazyCode(() => refer('foo').code),
          const Code(') {'),
          refer('print')([literalTrue]).statement,
          const Code('}'),
        ])),
      equalsDart(r'''
        main() {
          if (foo) {
            print(true);
          }
        }
      '''),
    );
  });
}