Fix Background/After sections for multiple Feature

Author: olexale

example/integration_test/sample_test.dart

@@ -61,6 +61,8 @@ String generateFeatureDart(
   );
 
   for (final feature in features) {
+    sb.writeln("  group('''${feature.first.value}''', () {");
+
     final hasBackground = _parseBackground(sb, feature);
     final hasAfter = _parseAfter(sb, feature);
 
@@ -90,13 +92,13 @@ bool _parseSetup(
 ) {
   var offset = lines.indexWhere((element) => element.type == elementType);
   if (offset != -1) {
-    sb.writeln('  Future<void> $title(WidgetTester tester) async {');
+    sb.writeln('    Future<void> $title(WidgetTester tester) async {');
     offset++;
     while (lines[offset].type == LineType.step) {
-      sb.writeln('    await ${getStepMethodCall(lines[offset].value)};');
+      sb.writeln('      await ${getStepMethodCall(lines[offset].value)};');
       offset++;
     }
-    sb.writeln('  }');
+    sb.writeln('    }');
   }
   return offset != -1;
 }
@@ -108,8 +110,6 @@ void _parseFeature(
   bool hasTearDown,
   String testMethodName,
 ) {
-  sb.writeln("  group('''${feature.first.value}''', () {");
-
   final scenarios = _splitScenarios(
     feature.skipWhile((value) => !_isNewScenario(value.type)).toList(),
   ).toList();

example/test/sample_test.dart

@@ -22,10 +22,10 @@ import './step/the_test_finishes.dart';
 import './step/the_app_is_running.dart';
 
 void main() {
-  Future<void> bddTearDown(WidgetTester tester) async {
-    await theTestFinishes(tester);
-  }
   group(\'\'\'Testing feature\'\'\', () {
+    Future<void> bddTearDown(WidgetTester tester) async {
+      await theTestFinishes(tester);
+    }
     testWidgets(\'\'\'Testing scenario\'\'\', (tester) async {
       try {
         await theAppIsRunning(tester);

lib/src/feature_generator.dart

@@ -22,10 +22,10 @@ import './step/the_server_always_return_errors.dart';
 import './step/the_app_is_running.dart';
 
 void main() {
-  Future<void> bddSetUp(WidgetTester tester) async {
-    await theServerAlwaysReturnErrors(tester);
-  }
   group(\'\'\'Testing feature\'\'\', () {
+    Future<void> bddSetUp(WidgetTester tester) async {
+      await theServerAlwaysReturnErrors(tester);
+    }
     testWidgets(\'\'\'Testing scenario\'\'\', (tester) async {
       await bddSetUp(tester);
       await theAppIsRunning(tester);

test/after_test.dart

@@ -14,6 +14,11 @@ Feature: Counter
         Given the app is running
         And I run {'func foo() {}; func bar() { print("hey!"); };'} code
         Then I see {'0'} text
+Feature: Counter 2
+    Background:
+        Given the app is running
+    Scenario: Initial counter value is 0
+        Given the app is running
 ''';
 
     const expectedFeatureDart = '''
@@ -29,13 +34,13 @@ import './step/i_run_code.dart';
 import 'package:bdd_sample/i_see_text.dart';
 
 void main() {
-  Future<void> bddSetUp(WidgetTester tester) async {
-    await theAppIsRunning(tester);
-  }
-  Future<void> bddTearDown(WidgetTester tester) async {
-    await _iDoNotSeeText(tester, '42');
-  }
   group(\'\'\'Counter\'\'\', () {
+    Future<void> bddSetUp(WidgetTester tester) async {
+      await theAppIsRunning(tester);
+    }
+    Future<void> bddTearDown(WidgetTester tester) async {
+      await _iDoNotSeeText(tester, '42');
+    }
     customTestWidgets(\'\'\'Initial counter value is 0\'\'\', (tester) async {
       try {
         await bddSetUp(tester);
@@ -47,6 +52,15 @@ void main() {
       }
     });
   });
+  group(\'\'\'Counter 2\'\'\', () {
+    Future<void> bddSetUp(WidgetTester tester) async {
+      await theAppIsRunning(tester);
+    }
+    customTestWidgets(\'\'\'Initial counter value is 0\'\'\', (tester) async {
+      await bddSetUp(tester);
+      await theAppIsRunning(tester);
+    });
+  });
 }
 ''';
 

test/background_test.dart

@@ -25,13 +25,13 @@ import './step/the_test_finishes.dart';
 import './step/the_app_is_running.dart';
 
 void main() {
-  Future<void> bddSetUp(WidgetTester tester) async {
-    await theServerAlwaysReturnErrors(tester);
-  }
-  Future<void> bddTearDown(WidgetTester tester) async {
-    await theTestFinishes(tester);
-  }
   group(\'\'\'Testing feature\'\'\', () {
+    Future<void> bddSetUp(WidgetTester tester) async {
+      await theServerAlwaysReturnErrors(tester);
+    }
+    Future<void> bddTearDown(WidgetTester tester) async {
+      await theTestFinishes(tester);
+    }
     testWidgets(\'\'\'Testing scenario\'\'\', (tester) async {
       try {
         await bddSetUp(tester);