Extract helper method to set up the mocks

hydephp · Sep 9, 2024 · 6f21293 · 6f21293
1 parent c5594a9
commit 6f21293
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/packages/framework/tests/Unit/BuildTaskServiceUnitTest.php b/packages/framework/tests/Unit/BuildTaskServiceUnitTest.php
@@ -207,47 +207,47 @@ public function testCanRunPostBuildTasksWithTasks()
 
     public function testRunPreBuildTasksCallsHandleMethods()
     {
-        $task = Mockery::mock(TestPreBuildTask::class)->makePartial()->shouldReceive('handle')->once()->getMock();
+        $task = $this->setupMock(TestPreBuildTask::class, 'handle')->getMock();
 
         $this->service->registerTask($task);
         $this->service->runPreBuildTasks();
     }
 
     public function testRunPostBuildTasksCallsHandleMethods()
     {
-        $task = Mockery::mock(TestPostBuildTask::class)->makePartial()->shouldReceive('handle')->once()->getMock();
+        $task = $this->setupMock(TestPostBuildTask::class, 'handle')->getMock();
 
         $this->service->registerTask($task);
         $this->service->runPostBuildTasks();
     }
 
     public function testRunPreBuildTasksCallsRunMethods()
     {
-        $task = Mockery::mock(TestPreBuildTask::class)->makePartial()->shouldReceive('run')->once()->getMock();
+        $task = $this->setupMock(TestPreBuildTask::class, 'run')->getMock();
 
         $this->service->registerTask($task);
         $this->service->runPreBuildTasks();
     }
 
     public function testRunPostBuildTasksCallsRunMethods()
     {
-        $task = Mockery::mock(TestPostBuildTask::class)->makePartial()->shouldReceive('run')->once()->getMock();
+        $task = $this->setupMock(TestPostBuildTask::class, 'run')->getMock();
 
         $this->service->registerTask($task);
         $this->service->runPostBuildTasks();
     }
 
     public function testRunPreBuildTasksCallsRunMethodsWithNullWhenServiceHasNoOutput()
     {
-        $task = Mockery::mock(TestPreBuildTask::class)->makePartial()->shouldReceive('run')->with(null)->once()->getMock();
+        $task = $this->setupMock(TestPreBuildTask::class, 'run')->with(null)->once()->getMock();
 
         $this->service->registerTask($task);
         $this->service->runPreBuildTasks();
     }
 
     public function testRunPostBuildTasksCallsRunMethodsWithNullWhenServiceHasNoOutput()
     {
-        $task = Mockery::mock(TestPostBuildTask::class)->makePartial()->shouldReceive('run')->with(null)->once()->getMock();
+        $task = $this->setupMock(TestPostBuildTask::class, 'run')->with(null)->once()->getMock();
 
         $this->service->registerTask($task);
         $this->service->runPostBuildTasks();
@@ -256,7 +256,7 @@ public function testRunPostBuildTasksCallsRunMethodsWithNullWhenServiceHasNoOutp
     public function testRunPreBuildTasksCallsRunMethodsWithOutputWhenServiceHasOutput()
     {
         $output = Mockery::mock(OutputStyle::class)->makePartial();
-        $task = Mockery::mock(TestPreBuildTask::class)->makePartial()->shouldReceive('run')->with($output)->once()->getMock();
+        $task = $this->setupMock(TestPreBuildTask::class, 'run')->with($output)->once()->getMock();
 
         $this->service->setOutput($output);
         $this->service->registerTask($task);
@@ -266,7 +266,7 @@ public function testRunPreBuildTasksCallsRunMethodsWithOutputWhenServiceHasOutpu
     public function testRunPostBuildTasksCallsRunMethodsWithOutputWhenServiceHasOutput()
     {
         $output = Mockery::mock(OutputStyle::class)->makePartial();
-        $task = Mockery::mock(TestPostBuildTask::class)->makePartial()->shouldReceive('run')->with($output)->once()->getMock();
+        $task = $this->setupMock(TestPostBuildTask::class, 'run')->with($output)->once()->getMock();
 
         $this->service->setOutput($output);
         $this->service->registerTask($task);
@@ -319,8 +319,7 @@ protected function createService(): BuildTaskService
 
     protected function mockKernelFilesystem(array $files = []): void
     {
-        $filesystem = Mockery::mock(Filesystem::class, [HydeKernel::getInstance()])
-            ->makePartial()->shouldReceive('smartGlob')->once()
+        $filesystem = $this->setupMock(Filesystem::class, 'smartGlob')
             ->with('app/Actions/*BuildTask.php', 0)
             ->andReturn(collect($files))->getMock();
 
@@ -332,6 +331,11 @@ protected function resetKernelInstance(): void
     {
         HydeKernel::setInstance(new HydeKernel());
     }
+
+    protected function setupMock(string $class, string $method): Mockery\ExpectationInterface|Mockery\Expectation|Mockery\HigherOrderMessage
+    {
+        return Mockery::mock($class)->makePartial()->shouldReceive($method)->once();
+    }
 }
 
 class InstantiableTestBuildTask extends BuildTask