Use model connection for load data query

Author: ellgreen

src/Builder/Builder.php

@@ -44,7 +44,7 @@ public function load(): bool
     {
         $query = $this->compile();
 
-        $connection = $this->databaseManager->connection($this->connection);
+        $connection = $this->databaseManager->connection($this->getConnectionName());
         return $connection->statement($query->getSql(), $query->getBindings());
     }
 
@@ -53,4 +53,9 @@ public function connection(?string $name = null): self
         $this->connection = $name;
         return $this;
     }
+
+    public function getConnectionName(): ?string
+    {
+        return $this->connection;
+    }
 }

src/Laravel/Traits/LoadsFiles.php

@@ -21,7 +21,10 @@ public static function loadFileBuilder(string $file, ?bool $local = null): Build
         $builder = app(Builder::class);
 
         $model = app(static::class);
-        $builder->file($file, $local)->into($model->getTable());
+
+        $builder->connection($model->getConnectionName())
+            ->file($file, $local)
+            ->into($model->getTable());
 
         $model->loadFileOptions($builder);
 
@@ -33,5 +36,7 @@ public function loadFileOptions(Builder $builder): void
         //
     }
 
+    abstract public function getConnectionName();
+
     abstract public function getTable();
 }

tests/Feature/LoadsFilesTraitTest.php

@@ -22,13 +22,27 @@ public function testGetBuilderWithOptions(): void
         $this->assertTrue($builder->isReplace());
     }
 
-    public function testDefaultLoadFileOptions()
+    public function testDefaultLoadFileOptions(): void
     {
         $builder = TestUserNoOptions::loadFileBuilder('/path/to/test_users.csv', true);
 
         $this->assertFalse($builder->isReplace());
     }
 
+    public function testLoadFileIsUsingModelConnection(): void
+    {
+        $builder = TestUser::loadFileBuilder('/path/to/test_users.csv', true);
+
+        $this->assertSame('test-connection', $builder->getConnectionName());
+    }
+
+    public function testLoadFileIsUsingModelConnectionNull(): void
+    {
+        $builder = TestUserNoOptions::loadFileBuilder('/path/to/test_users.csv', true);
+
+        $this->assertNull($builder->getConnectionName());
+    }
+
     public function testLoadFile(): void
     {
         $this->instance(Builder::class, $builder = $this->createPartialMock(Builder::class, [

tests/Feature/Models/TestUser.php

@@ -10,6 +10,8 @@ class TestUser extends Model
 {
     use LoadsFiles;
 
+    protected $connection = 'test-connection';
+
     public function loadFileOptions(Builder $builder): void
     {
         $builder->replace();