BUG: Fix general test

miroslaws8 · Mar 18, 2023 · db25778 · db25778
1 parent b3da0e0
commit db25778
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/Unit/AIModels/GeneralTest.php b/tests/Unit/AIModels/GeneralTest.php
@@ -3,6 +3,7 @@
 namespace Unit\AIModels;
 
 use Illuminate\Contracts\Container\BindingResolutionException;
+use Illuminate\Support\Collection;
 use Itsimiro\OpenAI\Services\AIModels\General;
 use Itsimiro\OpenAI\Services\API\ApiService;
 use Itsimiro\OpenAI\Services\API\Results\ModelResult;
@@ -24,9 +25,8 @@ public function testGetModels(): void
 
         $expectedModel = $this->getMockModel();
 
-        $dataCollection = $this->getMockBuilder(DataCollection::class)
+        $dataCollection = $this->getMockBuilder(Collection::class)
             ->setConstructorArgs([
-                'dataClass' => ModelResult::class,
                 'items' => []
             ])->getMock();