replace usage of deprecated getMockForTrait

michaeldyrynda · Sep 13, 2023 · f71a674 · f71a674
1 parent 25ed8d9
commit f71a674
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/tests/Unit/UuidResolversTest.php b/tests/Unit/UuidResolversTest.php
@@ -3,6 +3,7 @@
 namespace Tests\Unit;
 
 use Dyrynda\Database\Support\GeneratesUuid;
+use Mockery;
 use PHPUnit\Framework\TestCase;
 
 class UuidResolversTest extends TestCase
@@ -29,12 +30,14 @@ public static function provider_for_it_handles_uuid_versions(): array
      */
     public function it_handles_uuid_versions($version, $resolved)
     {
-        /* @var \Dyrynda\Database\Support\GeneratesUuid $generator */
-        $generator = $this->getMockForTrait(GeneratesUuid::class, mockedMethods: [
-            'uuidVersion',
-        ]);
-        $generator->method('uuidVersion')->willReturn($version);
+        $generator = Mockery::mock(UuidTestClass::class)->makePartial();
+        $generator->shouldReceive('uuidVersion')->once()->andReturn($version);
 
         $this->assertSame($resolved, $generator->resolveUuidVersion());
     }
 }
+
+class UuidTestClass
+{
+    use GeneratesUuid;
+}