From d2fddcf3178e0b855eeea91bca756c75156824dc Mon Sep 17 00:00:00 2001
From: Justin Guckes <justin.guckes@gmail.com>
Date: Mon, 14 Sep 2020 23:23:19 +0200
Subject: [PATCH] feat(custom paths): Allow passing in a rootFolderName

https://github.com/bmw-tech/ozzie.flutter/issues/2
---
 lib/ozzie.dart       | 21 ++++++++++++---------
 test/ozzie_test.dart | 20 +++++++++++---------
 2 files changed, 23 insertions(+), 18 deletions(-)

diff --git a/lib/ozzie.dart b/lib/ozzie.dart
index 244162f..7f7fa9a 100644
--- a/lib/ozzie.dart
+++ b/lib/ozzie.dart
@@ -7,19 +7,21 @@ import 'package:flutter_driver/flutter_driver.dart';
 import 'reporter.dart';
 import 'zip_generator.dart';
 
-const rootFolderName = "ozzie";
-
 /// [Ozzie] is the class responsible for taking screenshots and generating
 /// an HTML report when running your integrationt tests on Flutter.
 class Ozzie {
   final FlutterDriver driver;
+  final String basePath;
   final String groupName;
+  final String rootFolderName;
   final bool shouldTakeScreenshots;
-  var _doesGroupFolderNeedToBeDeleted = true;
+  bool _doesGroupFolderNeedToBeDeleted = true;
 
   Ozzie._internal(
     this.driver, {
+    this.basePath,
     @required this.groupName,
+    @required this.rootFolderName,
     @required this.shouldTakeScreenshots,
   }) : assert(driver != null);
 
@@ -39,12 +41,12 @@ class Ozzie {
     FlutterDriver driver, {
     String groupName = "default",
     bool shouldTakeScreenshots = true,
+    String rootFolderName = "ozzie",
   }) =>
-      Ozzie._internal(
-        driver,
-        groupName: groupName,
-        shouldTakeScreenshots: shouldTakeScreenshots,
-      );
+      Ozzie._internal(driver,
+          groupName: groupName,
+          shouldTakeScreenshots: shouldTakeScreenshots,
+          rootFolderName: rootFolderName);
 
   /// It takes a an PNG screnshot of the given state of the application when
   /// being called. The name of the screenshot will be the given `screenshotName`
@@ -123,7 +125,8 @@ class Ozzie {
 
   String get _groupFolderName => '$rootFolderName/$groupName';
 
-  String get _timestamp => DateTime.now().toIso8601String();
+  String get _timestamp =>
+      DateTime.now().toIso8601String().replaceAll(':', '.');
 
   String _fileName(String screenshotName) => '$_timestamp-$screenshotName.png';
 
diff --git a/test/ozzie_test.dart b/test/ozzie_test.dart
index 7e471cc..949d4a7 100644
--- a/test/ozzie_test.dart
+++ b/test/ozzie_test.dart
@@ -28,8 +28,8 @@ void main() {
   });
 
   tearDown(() async {
-    if (await Directory(rootFolderName).exists()) {
-      await Directory(rootFolderName).delete(recursive: true);
+    if (await Directory(ozzie.rootFolderName).exists()) {
+      await Directory(ozzie.rootFolderName).delete(recursive: true);
     }
   });
 
@@ -51,7 +51,8 @@ void main() {
 
     test('not taking screenshots still generates an HTML report', () async {
       await ozzie.generateHtmlReport();
-      final fileExists = await File('$rootFolderName/index.html').exists();
+      final fileExists =
+          await File('${ozzie.rootFolderName}/index.html').exists();
       expect(fileExists, isTrue);
     });
 
@@ -66,7 +67,7 @@ void main() {
     group('with screenshots enabled', () {
       test('on first takeScreenshot call the the group folder is deleted',
           () async {
-        final filePath = '$rootFolderName/$testGroupName';
+        final filePath = '${ozzie.rootFolderName}/$testGroupName';
         var testFile = await File('$filePath/testFile').create(recursive: true);
         testFile.writeAsBytes([1, 2, 3]);
         await ozzie.takeScreenshot('alex');
@@ -83,8 +84,9 @@ void main() {
 
       test('takeScreenshot generates PNGs containing given name', () async {
         await ozzie.takeScreenshot('rim');
-        final files =
-            Directory('$rootFolderName/$testGroupName').listSync().toList();
+        final files = Directory('${ozzie.rootFolderName}/$testGroupName')
+            .listSync()
+            .toList();
         final resultFile =
             files.where((f) => f is File).map((f) => f as File).first;
         expect(true, resultFile.path.contains('rim.png'));
@@ -95,7 +97,7 @@ void main() {
             () async {
           await ozzie.takeScreenshot('alex');
           final isHtmlReportGenerated =
-              await File('$rootFolderName/index.html').exists();
+              await File('${ozzie.rootFolderName}/index.html').exists();
           expect(false, isHtmlReportGenerated);
         });
 
@@ -103,7 +105,7 @@ void main() {
           await ozzie.takeScreenshot('alex');
           await ozzie.generateHtmlReport();
           final isHtmlReportGenerated =
-              await File('$rootFolderName/index.html').exists();
+              await File('${ozzie.rootFolderName}/index.html').exists();
           expect(true, isHtmlReportGenerated);
         });
       });
@@ -126,7 +128,7 @@ void main() {
         );
         await noScreenshotsOzzie.takeScreenshot('alex');
         final isHtmlReportGenerated =
-            await File('$rootFolderName/index.html').exists();
+            await File('${ozzie.rootFolderName}/index.html').exists();
         expect(false, isHtmlReportGenerated);
       });
     });