Add tests for the static API authentication features

Author: CodeWithKyrian

.github/workflows/test.yml

@@ -15,11 +15,20 @@ jobs:
     name: Tests on PHP ${{ matrix.php }} - ${{ matrix.dependency-version }}
 
     services:
-      chroma:
+      chroma-wo-auth:
         image: chromadb/chroma
         ports:
           - 8000:8000
 
+      chroma-w-auth:
+        image: chromadb/chroma
+        ports:
+          - 8001:8000
+        env:
+          CHROMA_SERVER_AUTH_CREDENTIALS: 'test-token'
+          CHROMA_SERVER_AUTH_CREDENTIALS_PROVIDER: 'chromadb.auth.token.TokenConfigServerAuthCredentialsProvider'
+          CHROMA_SERVER_AUTH_PROVIDER: 'chromadb.auth.token.TokenAuthServerProvider'
+
     steps:
       - name: Checkout
         uses: actions/checkout@v3

docker-compose.yml

@@ -1,13 +1,16 @@
 version: '3.9'
 
 services:
-  server:
+  chroma_wo_auth:
     image: 'chromadb/chroma'
     ports:
       - '8000:8000'
-    volumes:
-      - chroma-data:/chroma/chroma
 
-volumes:
-  chroma-data:
-    driver: local
+  chroma_w_auth:
+    image: 'chromadb/chroma'
+    ports:
+      - '8001:8000'
+    environment:
+      CHROMA_SERVER_AUTH_CREDENTIALS: 'test-token'
+      CHROMA_SERVER_AUTH_CREDENTIALS_PROVIDER: 'chromadb.auth.token.TokenConfigServerAuthCredentialsProvider'
+      CHROMA_SERVER_AUTH_PROVIDER: 'chromadb.auth.token.TokenAuthServerProvider'

src/Generated/ChromaApiClient.php

@@ -357,9 +357,14 @@ private function handleChromaApiException(\Exception|ClientExceptionInterface $e
                     ChromaException::throwSpecific($message, $error_type, $e->getCode());
                 }
 
+                // If the structure is {'error': 'Error Type', 'message' : 'Error message'}
+                if (isset($error['error']) && isset($error['message'])) {
+                    ChromaException::throwSpecific($error['message'], $error['error'], $e->getCode());
+                }
+
                 // If the structure is 'error' => 'Collection not found'
                 if (isset($error['error'])) {
-                    $message = $error['message'] ?? $error['error'];
+                    $message = $error['error'];
                     $error_type = ChromaException::inferTypeFromMessage($message);
 
                     ChromaException::throwSpecific($message, $error_type, $e->getCode());

src/Generated/Exceptions/ChromaException.php

@@ -26,6 +26,7 @@ public static function inferTypeFromMessage(string $message): string
     {
         return match (true) {
             str_contains($message, 'NotFoundError') => 'NotFoundError',
+            str_contains($message, 'AuthorizationError') => 'AuthorizationError',
             str_contains($message, 'UniqueConstraintError') => 'UniqueConstraintError',
             str_contains($message, 'ValueError') => 'ValueError',
             str_contains($message, 'dimensionality') => 'DimensionalityError',

tests/ChromaDB.php

@@ -4,18 +4,41 @@
 
 use Codewithkyrian\ChromaDB\Client;
 use Codewithkyrian\ChromaDB\ChromaDB;
+use Codewithkyrian\ChromaDB\Generated\Exceptions\ChromaAuthorizationException;
 
-it('can create a client instance', function () {
+it('can connect to a normal chroma server', function () {
     $client = ChromaDB::client();
 
     expect($client)->toBeInstanceOf(Client::class);
 });
 
-it('can create a client instance via factory', function () {
+it('can connect to a chroma server using factory', function () {
     $client = ChromaDB::factory()
         ->withHost('http://localhost')
         ->withPort(8000)
         ->connect();
 
     expect($client)->toBeInstanceOf(Client::class);
 });
+
+test('can connect to an API token authenticated chroma server', function () {
+    $client = ChromaDB::factory()
+        ->withPort(8001)
+        ->withBearerToken('test-token')
+        ->connect();
+
+    expect($client)->toBeInstanceOf(Client::class);
+});
+
+it('cannot connect to an API token authenticated chroma server with wrong token', function () {
+    ChromaDB::factory()
+        ->withPort(8001)
+        ->withBearerToken('wrong-token')
+        ->connect();
+})->throws(ChromaAuthorizationException::class);
+
+it('throws exception when connecting to API token authenticated chroma server with no token', function () {
+    ChromaDB::factory()
+        ->withPort(8001)
+        ->connect();
+})->throws(ChromaAuthorizationException::class);