use custom stub file, if provided

Author: TimoKoerber

src/OneTimeOperationCreator.php

@@ -57,6 +57,11 @@ protected function getPath(): string
 
     protected function getStubFilepath(): string
     {
+        // check for custom stub file
+        if (File::exists(base_path('stubs/one-time-operation.stub'))) {
+            return File::get(base_path('stubs/one-time-operation.stub'));
+        }
+
         if ($this->essential) {
             return File::get(__DIR__.'/../stubs/one-time-operation-essential.stub');
         }

tests/Feature/OneTimeOperationCreatorTest.php

@@ -25,4 +25,22 @@ public function it_creates_an_operation_file_instance()
 
         File::delete($filepath);
     }
+
+    /** @test */
+    public function it_creates_an_operation_file_with_custom_stub()
+    {
+        $mockFile = File::partialMock();
+        $mockFile->allows('exists')->with(base_path('stubs/one-time-operation.stub'))->andReturnTrue();
+        $mockFile->allows('get')->with(base_path('stubs/one-time-operation.stub'))->andReturns('This is a custom stub');
+
+        $directory = Config::get('one-time-operations.directory');
+        $filepath = base_path($directory).DIRECTORY_SEPARATOR.'2015_10_21_072800_test_operation.php';
+
+        OneTimeOperationCreator::createOperationFile('TestOperation');
+
+        $this->assertFileExists($filepath);
+        $this->assertStringContainsString('This is a custom stub', File::get($filepath));
+
+        File::delete($filepath);
+    }
 }