Extract helper method create mocks

hydephp · Sep 9, 2024 · e423d8a · e423d8a
1 parent 6f21293
commit e423d8a
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/packages/framework/tests/Unit/BuildTaskServiceUnitTest.php b/packages/framework/tests/Unit/BuildTaskServiceUnitTest.php
@@ -156,7 +156,7 @@ public function testCanSetOutputWithNull()
 
     public function testCanSetOutputWithOutputStyle()
     {
-        $this->can(fn () => $this->service->setOutput(Mockery::mock(OutputStyle::class)));
+        $this->can(fn () => $this->service->setOutput($this->mockOutput()));
     }
 
     public function testGenerateBuildManifestExtendsPostBuildTask()
@@ -255,7 +255,7 @@ public function testRunPostBuildTasksCallsRunMethodsWithNullWhenServiceHasNoOutp
 
     public function testRunPreBuildTasksCallsRunMethodsWithOutputWhenServiceHasOutput()
     {
-        $output = Mockery::mock(OutputStyle::class)->makePartial();
+        $output = $this->mockOutput();
         $task = $this->setupMock(TestPreBuildTask::class, 'run')->with($output)->once()->getMock();
 
         $this->service->setOutput($output);
@@ -265,7 +265,7 @@ public function testRunPreBuildTasksCallsRunMethodsWithOutputWhenServiceHasOutpu
 
     public function testRunPostBuildTasksCallsRunMethodsWithOutputWhenServiceHasOutput()
     {
-        $output = Mockery::mock(OutputStyle::class)->makePartial();
+        $output = $this->mockOutput();
         $task = $this->setupMock(TestPostBuildTask::class, 'run')->with($output)->once()->getMock();
 
         $this->service->setOutput($output);
@@ -336,6 +336,11 @@ protected function setupMock(string $class, string $method): Mockery\Expectation
     {
         return Mockery::mock($class)->makePartial()->shouldReceive($method)->once();
     }
+
+    protected function mockOutput(): Mockery\LegacyMockInterface|Mockery\MockInterface|OutputStyle
+    {
+        return Mockery::mock(OutputStyle::class)->makePartial();
+    }
 }
 
 class InstantiableTestBuildTask extends BuildTask