Add helper to execute steps

iamoat · Nov 5, 2018 · d5b86f4 · d5b86f4
1 parent 4869a79
commit d5b86f4
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 19 deletions.
diff --git a/tests/TestCase.php b/tests/TestCase.php
@@ -2,9 +2,28 @@
 
 namespace Tests;
 
+use Marquine\Etl\Loaders\Loader;
+use Marquine\Etl\Transformers\Transformer;
 use PHPUnit\Framework\TestCase as BaseTestCase;
 
 abstract class TestCase extends BaseTestCase
 {
-    //
+    protected function execute($step, $data)
+    {
+        if ($step instanceof Transformer) {
+            $method = 'transform';
+        }
+
+        if ($step instanceof Loader) {
+            $method = 'load';
+        }
+
+        $step->initialize();
+
+        foreach ($data as $row) {
+            $step->$method($row);
+        }
+
+        $step->finalize();
+    }
 }
diff --git a/tests/Transformers/ConvertCaseTest.php b/tests/Transformers/ConvertCaseTest.php
@@ -30,9 +30,7 @@ public function lowercase()
 
         $transformer->options(['mode' => 'lower']);
 
-        $transformer->initialize();
-
-        array_map([$transformer, 'transform'], $this->data);
+        $this->execute($transformer, $this->data);
 
         $this->assertEquals($expected, $this->data);
     }
@@ -49,9 +47,7 @@ public function uppercase()
 
         $transformer->options(['mode' => 'upper']);
 
-        $transformer->initialize();
-
-        array_map([$transformer, 'transform'], $this->data);
+        $this->execute($transformer, $this->data);
 
         $this->assertEquals($expected, $this->data);
     }
@@ -68,9 +64,7 @@ public function titlecase()
 
         $transformer->options(['mode' => 'title']);
 
-        $transformer->initialize();
-
-        array_map([$transformer, 'transform'], $this->data);
+        $this->execute($transformer, $this->data);
 
         $this->assertEquals($expected, $this->data);
     }
@@ -87,9 +81,7 @@ public function custom_columns()
 
         $transformer->options(['columns' => ['name']]);
 
-        $transformer->initialize();
-
-        array_map([$transformer, 'transform'], $this->data);
+        $this->execute($transformer, $this->data);
 
         $this->assertEquals($expected, $this->data);
     }

diff --git a/tests/Transformers/JsonDecodeTest.php b/tests/Transformers/JsonDecodeTest.php
@@ -28,7 +28,7 @@ public function default_options()
 
         $transformer = new JsonDecode;
 
-        array_map([$transformer, 'transform'], $this->data);
+        $this->execute($transformer, $this->data);
 
         $this->assertEquals($expected, $this->data);
     }
@@ -45,7 +45,7 @@ public function converting_objects_to_associative_arrays()
 
         $transformer->options(['assoc' => true]);
 
-        array_map([$transformer, 'transform'], $this->data);
+        $this->execute($transformer, $this->data);
 
         $this->assertEquals($expected, $this->data);
     }
@@ -62,7 +62,7 @@ public function custom_columns()
 
         $transformer->options(['columns' => ['data']]);
 
-        array_map([$transformer, 'transform'], $this->data);
+        $this->execute($transformer, $this->data);
 
         $this->assertEquals($expected, $this->data);
     }

diff --git a/tests/Transformers/JsonEncodeTest.php b/tests/Transformers/JsonEncodeTest.php
@@ -28,7 +28,7 @@ public function default_options()
 
         $transformer = new JsonEncode;
 
-        array_map([$transformer, 'transform'], $this->data);
+        $this->execute($transformer, $this->data);
 
         $this->assertEquals($expected, $this->data);
     }
@@ -45,7 +45,7 @@ public function custom_columns()
 
         $transformer->options(['columns' => ['data']]);
 
-        array_map([$transformer, 'transform'], $this->data);
+        $this->execute($transformer, $this->data);
 
         $this->assertEquals($expected, $this->data);
     }

diff --git a/tests/Transformers/RenameColumnsTest.php b/tests/Transformers/RenameColumnsTest.php
@@ -25,7 +25,7 @@ public function rename_column()
 
         $transformer->options(['columns' => ['email_address' => 'email']]);
 
-        array_map([$transformer, 'transform'], $data);
+        $this->execute($transformer, $data);
 
         $this->assertEquals($expected, $data);
     }