refactored

Author: bayareawebpro

src/SimpleCsvFacade.php

@@ -1,11 +1,12 @@
 <?php namespace BayAreaWebPro\SimpleCsv;
 
-use Illuminate\Support\Facades\Facade as LaravelFacade;
+use Illuminate\Support\Collection;
 use Illuminate\Support\LazyCollection;
+use Illuminate\Support\Facades\Facade as LaravelFacade;
 /**
  * The SimpleCsv Service Facade
- * @method static \Symfony\Component\HttpFoundation\StreamedResponse download(LazyCollection $collection, string $filename)
- * @method static void export(LazyCollection $collection, string $path)
+ * @method static \Symfony\Component\HttpFoundation\StreamedResponse download(Collection|LazyCollection|\Iterator|\Generator|array $collection, string $filename)
+ * @method static void export(Collection|LazyCollection|\Iterator|\Generator|array $collection, string $path)
  * @method static LazyCollection import(string $path)
  */
 class SimpleCsvFacade extends LaravelFacade

tests/Unit/DefaultTest.php

@@ -4,6 +4,7 @@
 
 
 use Illuminate\Database\Eloquent\Model;
+use Illuminate\Support\Collection;
 use Illuminate\Support\Str;
 use Illuminate\Support\Facades\File;
 use Illuminate\Support\LazyCollection;
@@ -57,6 +58,31 @@ private function getRandomStoragePath()
         return storage_path(Str::random(16) . '.csv');
     }
 
+    public function test_export_from_iterables()
+    {
+        $items = $this->getCollectionData(10)->toArray();
+
+        // Array
+        $pathA = $this->getRandomStoragePath();
+        SimpleCsv::export($items, $pathA);
+
+        $this->assertFileExists($pathA);
+        $fileData = File::get($pathA);
+        foreach ($items as $item) {
+            $this->assertStringContainsString($item['email'], $fileData);
+        }
+
+        // Collection
+        $pathB = $this->getRandomStoragePath();
+        SimpleCsv::export(Collection::make($items), $pathB);
+
+        $this->assertFileExists($pathB);
+        $fileData = File::get($pathB);
+        foreach ($items as $item) {
+            $this->assertStringContainsString($item['email'], $fileData);
+        }
+    }
+
     public function test_export_files_and_restore()
     {
         $items = $this->getCollectionData()->toArray();