pkg/analysis_server/test/src/services/correction/assist/flutter_wrap_column_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.

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';

main() {
  defineReflectiveSuite(() {
    defineReflectiveTests(FlutterWrapColumnTest);
  });
}

@reflectiveTest
class FlutterWrapColumnTest extends AssistProcessorTest {
  @override
  AssistKind get kind => DartAssistKind.FLUTTER_WRAP_COLUMN;

  test_controlFlowCollections_if() async {
    addFlutterPackage();
    await resolveTestUnit('''
import 'package:flutter/widgets.dart';

Widget build(bool b) {
  return Row(
    children: [
      Text('aaa'),
      if (b) /*caret*/Text('bbb'),
      Text('ccc'),
    ],
  );
}
''');
    await assertHasAssist('''
import 'package:flutter/widgets.dart';

Widget build(bool b) {
  return Row(
    children: [
      Text('aaa'),
      if (b) Column(
        children: <Widget>[
          Text('bbb'),
        ],
      ),
      Text('ccc'),
    ],
  );
}
''');
  }

  test_coveredByWidget() async {
    addFlutterPackage();
    await resolveTestUnit('''
import 'package:flutter/widgets.dart';

class FakeFlutter {
  main() {
    return Container(
      child: /*caret*/Text('aaa'),
    );
  }
}
''');
    await assertHasAssist('''
import 'package:flutter/widgets.dart';

class FakeFlutter {
  main() {
    return Container(
      child: Column(
        children: <Widget>[
          Text('aaa'),
        ],
      ),
    );
  }
}
''');
  }

  test_coversWidgets() async {
    addFlutterPackage();
    await resolveTestUnit('''
import 'package:flutter/widgets.dart';

class FakeFlutter {
  main() {
    return Row(children: [
      Text('aaa'),
// start
      Text('bbb'),
      Text('ccc'),
// end
      Text('ddd'),
    ]);
  }
}
''');
    await assertHasAssist('''
import 'package:flutter/widgets.dart';

class FakeFlutter {
  main() {
    return Row(children: [
      Text('aaa'),
      Column(
        children: <Widget>[
          Text('bbb'),
          Text('ccc'),
        ],
      ),
      Text('ddd'),
    ]);
  }
}
''');
  }
}