test: fix tests

supabase · Aug 12, 2024 · 1914280 · 1914280
1 parent 20a53c1
commit 1914280
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 4 deletions.
diff --git a/packages/postgrest/test/basic_test.dart b/packages/postgrest/test/basic_test.dart
@@ -91,11 +91,15 @@ void main() {
       final httpClient = CustomHttpClient();
       final postgrest = PostgrestClient(rootUrl, httpClient: httpClient);
 
-      await postgrest.rpc('func').setHeader("myKey", "myValue").select();
+      await postgrest
+          .rpc('empty-succ')
+          .setHeader("myKey", "myValue")
+          .select()
+          .head();
       expect(httpClient.lastRequest!.headers, containsPair("myKey", "myValue"));
 
       // Other following requests should not have the header
-      await postgrest.rpc('func').select();
+      await postgrest.rpc('empty-succ').select().head();
       expect(httpClient.lastRequest!.headers,
           isNot(containsPair("myKey", "myValue")));
     });
@@ -104,11 +108,15 @@ void main() {
       final httpClient = CustomHttpClient();
       final postgrest = PostgrestClient(rootUrl, httpClient: httpClient);
 
-      await postgrest.from('users').setHeader("myKey", "myValue").select();
+      await postgrest
+          .from('empty-succ')
+          .setHeader("myKey", "myValue")
+          .select()
+          .head();
       expect(httpClient.lastRequest!.headers, containsPair("myKey", "myValue"));
 
       // Other following requests should not have the header
-      await postgrest.from('users').select();
+      await postgrest.from('empty-succ').select().head();
       expect(httpClient.lastRequest!.headers,
           isNot(containsPair("myKey", "myValue")));
     });

diff --git a/packages/postgrest/test/custom_http_client.dart b/packages/postgrest/test/custom_http_client.dart
@@ -6,6 +6,13 @@ class CustomHttpClient extends BaseClient {
   Future<StreamedResponse> send(BaseRequest request) async {
     lastRequest = request;
 
+    if (request.url.path.endsWith("empty-succ")) {
+      return StreamedResponse(
+        Stream.empty(),
+        200,
+        request: request,
+      );
+    }
     //Return custom status code to check for usage of this client.
     return StreamedResponse(
       request.finalize(),