diff --git a/tests/Feature/DeepSeekClient.php b/tests/Feature/DeepSeekClient.php
new file mode 100644
index 0000000..a0866e0
--- /dev/null
+++ b/tests/Feature/DeepSeekClient.php
@@ -0,0 +1,52 @@
+<?php
+
+use DeepSeek\DeepSeekClient;
+use DeepSeek\Enums\Requests\HTTPState;
+
+test('Run query with valid API Key should return 200', function () {
+    // Arrange
+    $apiKey = "valid-api-key";
+    $client = DeepSeekClient::build($apiKey)
+        ->query('Hello DeepSeek, how are you today?')
+        ->setTemperature(1.5);
+
+    // Act
+    $response = $client->run();
+    $result = $client->getResult();
+
+    // Assert
+    expect($response)->not->toBeEmpty($response)
+        ->and($result->getStatusCode())->toEqual(HTTPState::OK->value);
+});
+
+test('Run query with valid API Key & insufficient balance should return 402', function () {
+    // Arrange
+    $apiKey = "insufficient-balance-api-key";
+    $client = DeepSeekClient::build($apiKey)
+        ->query('Hello DeepSeek, how are you today?')
+        ->setTemperature(1.5);
+
+    // Act
+    $response = $client->run();
+    $result = $client->getResult();
+
+    // Assert
+    expect($response)->not->toBeEmpty($response)
+        ->and($result->getStatusCode())->toEqual(HTTPState::PAYMENT_REQUIRED->value);
+});
+
+test('Run query with invalid API key should return 401', function () {
+    // Arrange
+    $apiKey = "insufficient-balance-api-key";
+    $client = DeepSeekClient::build($apiKey)
+        ->query('Hello DeepSeek, how are you today?')
+        ->setTemperature(1.5);
+
+    // Act
+    $response = $client->run();
+    $result = $client->getResult();
+
+    // Assert
+    expect($response)->not->toBeEmpty($response)
+        ->and($result->getStatusCode())->toEqual(HTTPState::UNAUTHORIZED->value);
+});
diff --git a/tests/Feature/HandelResultDeepseekTest.php b/tests/Feature/HandelResultDeepseekTest.php
deleted file mode 100644
index 9b2ab44..0000000
--- a/tests/Feature/HandelResultDeepseekTest.php
+++ /dev/null
@@ -1,54 +0,0 @@
-<?php
-namespace Tests\Feature;
-
-
-use DeepSeek\DeepSeekClient;
-use DeepSeek\Enums\Requests\HTTPState;
-use PHPUnit\Framework\TestCase;
-
-class HandelResultDeepseekTest extends TestCase
-{
-    protected $apiKey;
-    protected $expiredApiKey;
-    protected function setUp():void
-    {
-        $this->apiKey = "valid-api-key";
-        $this->expiredApiKey = "expired-api-key";
-    }
-    public function test_ok_response()
-    {
-        $deepseek = DeepSeekClient::build($this->apiKey)
-            ->query('Hello Deepseek, how are you today?')
-            ->setTemperature(1.5);
-        $response = $deepseek->run();
-        $result = $deepseek->getResult();
-
-        $this->assertNotEmpty($response);
-        $this->assertEquals(HTTPState::OK->value, $result->getStatusCode());
-    }
-    public function test_can_not_access_with_api_expired_payment()
-    {
-        $deepseek = DeepSeekClient::build($this->expiredApiKey)
-            ->query('Hello Deepseek, how are you today?')
-            ->setTemperature(1.5);
-        $response = $deepseek->run();
-        $result = $deepseek->getResult();
-
-        $this->assertNotEmpty($response);
-        if(!$result->isSuccess())
-        {
-            $this->assertEquals(HTTPState::PAYMENT_REQUIRED->value, $result->getStatusCode());
-        }
-    }
-    public function test_access_with_wrong_api_key()
-    {
-        $deepseek = DeepSeekClient::build($this->apiKey."wrong-api-key")
-            ->query('Hello Deepseek, how are you today?')
-            ->setTemperature(1.5);
-        $response = $deepseek->run();
-        $result = $deepseek->getResult();
-
-        $this->assertNotEmpty($response);
-        $this->assertEquals(HTTPState::UNAUTHORIZED->value, $result->getStatusCode());
-    }
-}