pkg/analysis_server/test/src/services/correction/assist/convert_into_block_body_test.dart

// Copyright (c) 2018, 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.

// @dart = 2.9

import 'package:analysis_server/src/services/correction/assist.dart';
import 'package:analyzer_plugin/utilities/assist/assist.dart';
import 'package:test_reflective_loader/test_reflective_loader.dart';

import 'assist_processor.dart';

void main() {
  defineReflectiveSuite(() {
    defineReflectiveTests(ConvertIntoBlockBodyTest);
  });
}

@reflectiveTest
class ConvertIntoBlockBodyTest extends AssistProcessorTest {
  @override
  AssistKind get kind => DartAssistKind.CONVERT_INTO_BLOCK_BODY;

  Future<void> test_async() async {
    await resolveTestCode('''
class A {
  mmm() async => 123;
}
''');
    await assertHasAssistAt('mmm()', '''
class A {
  mmm() async {
    return 123;
  }
}
''');
  }

  Future<void> test_closure() async {
    await resolveTestCode('''
setup(x) {}
main() {
  setup(() => 42);
}
''');
    await assertHasAssistAt('() => 42', '''
setup(x) {}
main() {
  setup(() {
    return 42;
  });
}
''');
    assertExitPosition(after: '42;');
  }

  Future<void> test_closure_voidExpression() async {
    await resolveTestCode('''
setup(x) {}
main() {
  setup(() => print('done'));
}
''');
    await assertHasAssistAt('() => print', '''
setup(x) {}
main() {
  setup(() {
    print('done');
  });
}
''');
    assertExitPosition(after: "');");
  }

  Future<void> test_constructor() async {
    await resolveTestCode('''
class A {
  A.named();

  factory A() => A.named();
}
''');
    await assertHasAssistAt('A()', '''
class A {
  A.named();

  factory A() {
    return A.named();
  }
}
''');
  }

  Future<void> test_inExpression() async {
    await resolveTestCode('''
main() => 123;
''');
    await assertNoAssistAt('123;');
  }

  Future<void> test_method() async {
    await resolveTestCode('''
class A {
  mmm() => 123;
}
''');
    await assertHasAssistAt('mmm()', '''
class A {
  mmm() {
    return 123;
  }
}
''');
  }

  Future<void> test_noEnclosingFunction() async {
    await resolveTestCode('''
var v = 123;
''');
    await assertNoAssistAt('v =');
  }

  Future<void> test_notExpressionBlock() async {
    await resolveTestCode('''
fff() {
  return 123;
}
''');
    await assertNoAssistAt('fff() {');
  }

  Future<void> test_onArrow() async {
    await resolveTestCode('''
fff() => 123;
''');
    await assertHasAssistAt('=>', '''
fff() {
  return 123;
}
''');
  }

  Future<void> test_onName() async {
    await resolveTestCode('''
fff() => 123;
''');
    await assertHasAssistAt('fff()', '''
fff() {
  return 123;
}
''');
  }

  Future<void> test_throw() async {
    await resolveTestCode('''
class A {
  mmm() => throw 'error';
}
''');
    await assertHasAssistAt('mmm()', '''
class A {
  mmm() {
    throw 'error';
  }
}
''');
  }
}