Fix bug when no build.yaml is provided. (#160)

bmw-tech · Apr 20, 2024 · 4d265cd · 4d265cd
1 parent c61c2f3
commit 4d265cd
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/widget_driver_generator/CHANGELOG.md b/widget_driver_generator/CHANGELOG.md
@@ -1,5 +1,9 @@
 # CHANGELOG
 
+## 1.2.1
+
+* Fixes bug when no build.yaml is provided.
+
 ## 1.2.0
 
 * Introduces the option to specify default values for types in your build.yaml.

diff --git a/widget_driver_generator/lib/src/widget_driver_generator.dart b/widget_driver_generator/lib/src/widget_driver_generator.dart
@@ -20,8 +20,11 @@ class WidgetDriverGenerator extends GeneratorForAnnotation<GenerateTestDriver> {
     required BuilderOptions options,
     PackageInfoProvider? packageInfoProvider,
   }) : _packageInfoProvider = packageInfoProvider ?? PackageInfoProvider() {
-    final config = (options.config["defaultTestValues"] as YamlMap).entries;
-    DefaultReturnValueHelper.setCustomDefaultValues(config);
+    final valuesFromConfig = options.config["defaultTestValues"];
+    if (valuesFromConfig is YamlMap) {
+      final customValues = valuesFromConfig.entries;
+      DefaultReturnValueHelper.setCustomDefaultValues(customValues);
+    }
   }
 
   @override

diff --git a/widget_driver_generator/pubspec.yaml b/widget_driver_generator/pubspec.yaml
@@ -1,6 +1,6 @@
 name: widget_driver_generator
 description: This package provides generators for WidgetDriver to automate the creation of your TestDrivers and WidgetDriverProviders
-version: 1.2.0
+version: 1.2.1
 repository: https://github.com/bmw-tech/widget_driver/tree/master/widget_driver_generator
 issue_tracker: https://github.com/bmw-tech/widget_driver/issues