Add 'After' keyword

Author: olexale

lib/src/bdd_line.dart

@@ -13,6 +13,7 @@ enum LineType {
   background,
   scenario,
   step,
+  after,
   unknown,
 }
 
@@ -23,6 +24,9 @@ LineType _lineTypeFromString(String line) {
   if (backgroundMarkers.any((marker) => line.startsWith(marker))) {
     return LineType.background;
   }
+  if (afterMarkers.any((marker) => line.startsWith(marker))) {
+    return LineType.after;
+  }
   if (scenarioMarkers.any((marker) => line.startsWith(marker))) {
     return LineType.scenario;
   }
@@ -34,6 +38,7 @@ LineType _lineTypeFromString(String line) {
 
 const featureMarkers = ['Feature:'];
 const backgroundMarkers = ['Background:'];
+const afterMarkers = ['After:'];
 const scenarioMarkers = ['Scenario:', 'Example:'];
 const stepMarkers = ['Given', 'When', 'Then', 'And', 'But'];
 

lib/src/feature_generator.dart

@@ -1,3 +1,5 @@
+import 'dart:math';
+
 import 'package:bdd_widget_test/src/bdd_line.dart';
 import 'package:bdd_widget_test/src/step_file.dart';
 import 'package:bdd_widget_test/src/step_generator.dart';
@@ -29,36 +31,49 @@ String generateFeatureDart(List<BddLine> lines, List<StepFile> steps) {
 
   for (final feature in features) {
     final backgroundOffset = _parseBackground(sb, feature);
-    _parseFeature(sb, feature, backgroundOffset);
+    final afterOffset = _parseAfter(sb, feature);
+    final offset = _calculateOffset(backgroundOffset, afterOffset);
+    _parseFeature(sb, feature, offset);
   }
   sb.writeln('}');
   return sb.toString();
 }
 
-int _parseBackground(StringBuffer sb, List<BddLine> lines) {
-  var backgroundOffset =
-      lines.indexWhere((element) => element.type == LineType.background);
-  if (backgroundOffset != -1) {
-    sb.writeln('  setUp(() async {');
-    backgroundOffset++;
-    while (lines[backgroundOffset].type == LineType.step) {
-      sb.writeln(
-          '    await ${getStepMethodName(lines[backgroundOffset].value)}();');
-      backgroundOffset++;
+int _parseBackground(StringBuffer sb, List<BddLine> lines) =>
+    _parseSetup(sb, lines, LineType.background, 'setUp');
+
+int _parseAfter(StringBuffer sb, List<BddLine> lines) =>
+    _parseSetup(sb, lines, LineType.after, 'tearDown');
+
+int _calculateOffset(int backgroundOffset, int afterOffset) {
+  if (backgroundOffset == -1 && afterOffset == -1) {
+    return -1;
+  }
+  return max(backgroundOffset, afterOffset);
+}
+
+int _parseSetup(
+    StringBuffer sb, List<BddLine> lines, LineType elementType, String title) {
+  var offset = lines.indexWhere((element) => element.type == elementType);
+  if (offset != -1) {
+    sb.writeln('  $title(() async {');
+    offset++;
+    while (lines[offset].type == LineType.step) {
+      sb.writeln('    await ${getStepMethodName(lines[offset].value)}();');
+      offset++;
     }
     sb.writeln('  });');
   }
-  return backgroundOffset;
+  return offset;
 }
 
-void _parseFeature(
-    StringBuffer sb, List<BddLine> feature, int backgroundOffset) {
+void _parseFeature(StringBuffer sb, List<BddLine> feature, int offset) {
   sb.writeln('  group(\'${feature.first.value}\', () {');
 
   final scenarios = splitWhen(
-      feature.skip(backgroundOffset == -1
+      feature.skip(offset == -1
           ? 1 // Skip 'Feature:'
-          : backgroundOffset), // or 'Backround:'
+          : offset), // or 'Backround:' / 'After:'
       (e) => e.type == LineType.scenario).toList();
   for (final scenario in scenarios) {
     _parseScenario(sb, scenario);

test/after_test.dart

@@ -0,0 +1,39 @@
+import 'package:bdd_widget_test/src/feature_file.dart';
+import 'package:flutter_test/flutter_test.dart';
+
+void main() {
+  test('After steps appear after groups ', () {
+    const featureFile = '''
+Feature: Testing feature
+  After:
+    And the test finishes
+  Scenario: Testing scenario
+    Given the app is running
+''';
+
+    const expectedFeatureDart = '''
+// GENERATED CODE - DO NOT MODIFY BY HAND
+// ignore_for_file: unused_import, directives_ordering
+
+import 'package:flutter/material.dart';
+import 'package:flutter_test/flutter_test.dart';
+
+import './step/the_test_finishes.dart';
+import './step/the_app_is_running.dart';
+
+void main() {
+  tearDown(() async {
+    await theTestFinishes();
+  });
+  group('Testing feature', () {
+    testWidgets('Testing scenario', (tester) async {
+      await theAppIsRunning(tester);
+    });
+  });
+}
+''';
+
+    final feature = FeatureFile(path: 'test.feature', input: featureFile);
+    expect(feature.dartContent, expectedFeatureDart);
+  });
+}

test/setup_test.dart

@@ -0,0 +1,45 @@
+import 'package:bdd_widget_test/src/feature_file.dart';
+import 'package:flutter_test/flutter_test.dart';
+
+void main() {
+  test('After steps appear after groups ', () {
+    const featureFile = '''
+Feature: Testing feature
+  Background:
+    Given the server always return errors
+  After:
+    And the test finishes
+  Scenario: Testing scenario
+    Given the app is running
+''';
+
+    const expectedFeatureDart = '''
+// GENERATED CODE - DO NOT MODIFY BY HAND
+// ignore_for_file: unused_import, directives_ordering
+
+import 'package:flutter/material.dart';
+import 'package:flutter_test/flutter_test.dart';
+
+import './step/the_server_always_return_errors.dart';
+import './step/the_test_finishes.dart';
+import './step/the_app_is_running.dart';
+
+void main() {
+  setUp(() async {
+    await theServerAlwaysReturnErrors();
+  });
+  tearDown(() async {
+    await theTestFinishes();
+  });
+  group('Testing feature', () {
+    testWidgets('Testing scenario', (tester) async {
+      await theAppIsRunning(tester);
+    });
+  });
+}
+''';
+
+    final feature = FeatureFile(path: 'test.feature', input: featureFile);
+    expect(feature.dartContent, expectedFeatureDart);
+  });
+}