Modified the geturl function to account for a connector being passed into the connectorConfig so that executeGraphql and executeGraphqlRead don't try to retrieve data from wrong endpoints. Included unit tests for the executeMutation function.

FOLUSO ONATEMOWO · FOLUSO ONATEMOWO · commit fe78b87af3ed · 2025-08-04T11:58:25.000-07:00
diff --git a/.gitignore b/.gitignore
@@ -21,3 +21,4 @@ firebase-admin-*.tgz
 
 docgen/markdown/
 service_account.json
+dataconnect/
diff --git a/src/data-connect/data-connect-api-client-internal.ts b/src/data-connect/data-connect-api-client-internal.ts
@@ -33,7 +33,7 @@ const FIREBASE_DATA_CONNECT_BASE_URL_FORMAT =
 
 /** The Firebase Data Connect backend base URL format including a connector. */
 const FIREBASE_DATA_CONNECT_BASE_URL_FORMAT_WITH_CONNECTOR =
-    'https://firebasedataconnect.googleapis.com/{version}/projects/{projectId}/locations/{locationId}/services/{serviceId}/connectors/${connector}:{endpointId}';
+    'https://firebasedataconnect.googleapis.com/{version}/projects/{projectId}/locations/{locationId}/services/{serviceId}/connectors/{connector}:{endpointId}';
 
 /** Firebase Data Connect base URl format when using the Data Connect emulator. */
 const FIREBASE_DATA_CONNECT_EMULATOR_BASE_URL_FORMAT =
@@ -153,6 +153,15 @@ export class DataConnectApiClient {
         DATA_CONNECT_ERROR_CODE_MAPPING.INVALID_ARGUMENT,
         'GraphqlOptions should be a non-null object');
     }
+    //Recent addition
+    if (typeof options !== 'undefined') {
+      if (!validator.isNonNullObject(options)) {
+        throw new FirebaseDataConnectError(
+          DATA_CONNECT_ERROR_CODE_MAPPING.INVALID_ARGUMENT,
+          'GraphqlOptions must be a non-null object');
+      }
+    }
+
     if (!("operationName" in options)) {
       throw new FirebaseDataConnectError(
         DATA_CONNECT_ERROR_CODE_MAPPING.INVALID_ARGUMENT,
@@ -215,7 +224,7 @@ export class DataConnectApiClient {
         };
         let urlFormat: string;
         if (useEmulator()) {
-          if ('connector' in urlParams){
+          if ('connector' in urlParams && (endpointId === EXECUTE_QUERY_ENDPOINT || endpointId === EXECUTE_MUTATION_ENDPOINT)){
             urlFormat = utils.formatString(FIREBASE_DATA_CONNECT_EMULATOR_BASE_URL_FORMAT_WITH_CONNECTOR, {
             host: emulatorHost()
           });
@@ -226,7 +235,7 @@ export class DataConnectApiClient {
             });
           }
         } else {
-            if ('connector' in urlParams){
+            if ('connector' in urlParams && (endpointId === EXECUTE_QUERY_ENDPOINT || endpointId === EXECUTE_MUTATION_ENDPOINT)){
               urlFormat = FIREBASE_DATA_CONNECT_BASE_URL_FORMAT_WITH_CONNECTOR}
             else{
               urlFormat = FIREBASE_DATA_CONNECT_BASE_URL_FORMAT;}
diff --git a/src/data-connect/data-connect.ts b/src/data-connect/data-connect.ts
@@ -84,9 +84,9 @@ export class DataConnect {
     query: string,
     options?: GraphqlOptions<Variables>,
   ): Promise<ExecuteGraphqlResponse<GraphqlResponse>> {
-    if ("connector" in this.connectorConfig){
-      throw new FirebaseDataConnectError(DATA_CONNECT_ERROR_CODE_MAPPING.INVALID_ARGUMENT,'executeGraphql does not require a connector');
-    }
+    // if ("connector" in this.connectorConfig){
+    //   throw new FirebaseDataConnectError(DATA_CONNECT_ERROR_CODE_MAPPING.INVALID_ARGUMENT,'executeGraphql does not require a connector');
+    // }
     return this.client.executeGraphql(query, options);
   }
 
@@ -102,9 +102,9 @@ export class DataConnect {
     query: string,
     options?: GraphqlOptions<Variables>,
   ): Promise<ExecuteGraphqlResponse<GraphqlResponse>> {
-    if ("connector" in this.connectorConfig){
-      throw new FirebaseDataConnectError(DATA_CONNECT_ERROR_CODE_MAPPING.INVALID_ARGUMENT,'executeGraphqlRead does not require a connector');
-    }
+    // if ("connector" in this.connectorConfig){
+    //   throw new FirebaseDataConnectError(DATA_CONNECT_ERROR_CODE_MAPPING.INVALID_ARGUMENT,'executeGraphqlRead does not require a connector');
+    // }
     return this.client.executeGraphqlRead(query, options);
   }
 
diff --git a/test/unit/data-connect/data-connect-api-client-internal.spec.ts b/test/unit/data-connect/data-connect-api-client-internal.spec.ts
@@ -81,6 +81,10 @@ describe('DataConnectApiClient', () => {
     connectorConfig,
     mocks.mockCredentialApp());
 
+  const clientWithoutProjectId_with_connector = new DataConnectApiClient(
+    connectorConfig_with_connector,
+    mocks.mockCredentialApp());
+
   const mockOptions = {
     credential: new mocks.MockCredential(),
     projectId: 'test-project',
@@ -89,12 +93,14 @@ describe('DataConnectApiClient', () => {
   let app: FirebaseApp;
 
   let apiClient: DataConnectApiClient;
+  let apiClient_with_connector: DataConnectApiClient;
   let sandbox: sinon.SinonSandbox;
 
   beforeEach(() => {
     sandbox = sinon.createSandbox();
     app = mocks.appWithOptions(mockOptions);
     apiClient = new DataConnectApiClient(connectorConfig, app);
+    apiClient_with_connector = new DataConnectApiClient(connectorConfig_with_connector, app);
   });
 
   afterEach(() => {
@@ -116,6 +122,17 @@ describe('DataConnectApiClient', () => {
     it('should initialize httpClient with the provided app', () => {
       expect((apiClient as any).httpClient).to.be.an.instanceOf(AuthorizedHttpClient);
     });
+    // Test for an app instance with a connector within the connector config
+    it('should throw an error if app is not a valid Firebase app instance', () => {
+      expect(() => new DataConnectApiClient(connectorConfig_with_connector, null as unknown as FirebaseApp)).to.throw(
+        FirebaseDataConnectError,
+        'First argument passed to getDataConnect() must be a valid Firebase app instance.'
+      );
+    });
+
+    it('should initialize httpClient with the provided app', () => {
+      expect((apiClient_with_connector as any).httpClient).to.be.an.instanceOf(AuthorizedHttpClient);
+    });
   });
 
   describe('executeGraphql', () => {
@@ -215,7 +232,7 @@ describe('DataConnectApiClient', () => {
             method: 'POST',
             url: `https://firebasedataconnect.googleapis.com/v1alpha/projects/test-project/locations/${connectorConfig.location}/services/${connectorConfig.serviceId}:executeGraphql`,
             headers: EXPECTED_HEADERS,
-            data: { query: 'query' }
+             data: { query: 'query' }
           });
         });
     });
@@ -235,6 +252,164 @@ describe('DataConnectApiClient', () => {
           });
         });
     });
+
+    it('should resolve with the GraphQL response on success when a connector is passed in', () => {
+      interface UsersResponse {
+        users: [
+          user: {
+            id: string;
+            name: string;
+            address: string;
+          }
+        ];
+      }
+      const stub = sandbox
+        .stub(HttpClient.prototype, 'send')
+        .resolves(utils.responseFrom(TEST_RESPONSE, 200));
+      return apiClient_with_connector.executeGraphql<UsersResponse, unknown>('query', {})
+        .then((resp) => {
+          expect(resp.data.users).to.be.not.empty;
+          expect(resp.data.users[0].name).to.be.not.undefined;
+          expect(resp.data.users[0].address).to.be.not.undefined;
+          expect(resp.data.users).to.deep.equal(TEST_RESPONSE.data.users);
+          expect(stub).to.have.been.calledOnce.and.calledWith({
+            method: 'POST',
+            url: `https://firebasedataconnect.googleapis.com/v1alpha/projects/test-project/locations/${connectorConfig.location}/services/${connectorConfig.serviceId}:executeGraphql`,
+            headers: EXPECTED_HEADERS,
+            data: { query: 'query' }
+          });
+        });
+    });
+
+    it('should use DATA_CONNECT_EMULATOR_HOST if set', () => {
+      process.env.DATA_CONNECT_EMULATOR_HOST = 'localhost:9399';
+      const stub = sandbox
+        .stub(HttpClient.prototype, 'send')
+        .resolves(utils.responseFrom(TEST_RESPONSE, 200));
+      return apiClient_with_connector.executeGraphql('query', {})
+        .then(() => {
+          expect(stub).to.have.been.calledOnce.and.calledWith({
+            method: 'POST',
+            url: `http://localhost:9399/v1alpha/projects/test-project/locations/${connectorConfig.location}/services/${connectorConfig.serviceId}:executeGraphql`,
+            headers: EMULATOR_EXPECTED_HEADERS,
+            data: { query: 'query' }
+          });
+        });
+    });
+
+  });
+
+  describe('executeMutation', () => {
+    //what if there's no project id and also there's no connector, what error would that be? -> should error the connector first because you won't even reach the endpoint to find out if there's a project id or not
+    it('should reject when project id is not available', () => {
+      return clientWithoutProjectId_with_connector.executeMutation({operationName: 'getById'})
+        .should.eventually.be.rejectedWith(noProjectId);
+    });
+
+    it('should throw an error if no arguments are passed in', async () => {
+      await expect(apiClient_with_connector.executeMutation(undefined as any)).to.be.rejectedWith(
+        FirebaseDataConnectError,
+        'GraphqlOptions should be a non-null object'
+      );
+    });
+
+    const invalidOptions = [null, NaN, 0, 1, true, false, [], _.noop];
+    invalidOptions.forEach((invalidOption) => {
+      it('should throw given an invalid options object: ' + JSON.stringify(invalidOption), async () => {
+        await expect(apiClient_with_connector.executeMutation(invalidOption as any)).to.be.rejectedWith(
+          FirebaseDataConnectError,
+          'GraphqlOptions must be a non-null object'
+        );
+      });
+    });
+    //could this pass as a null object, also what if the wrong operaton was passed in, would it be handled in another test- say integration?
+    it('should throw an error if there is no operationName', async () => {
+      await expect(apiClient_with_connector.executeMutation({})).to.be.rejectedWith(
+        FirebaseDataConnectError,
+        'GraphqlOptions must contain `operationName`.'
+      );
+    });
+
+    it('should reject when a full platform error response is received', () => {
+      sandbox
+        .stub(HttpClient.prototype, 'send')
+        .rejects(utils.errorFrom(ERROR_RESPONSE, 404));
+      const expected = new FirebaseDataConnectError('not-found', 'Requested entity not found');
+      return apiClient_with_connector.executeMutation({operationName: 'getById'})
+        .should.eventually.be.rejected.and.deep.include(expected);
+    });
+
+    it('should reject with unknown-error when error code is not present', () => {
+      sandbox
+        .stub(HttpClient.prototype, 'send')
+        .rejects(utils.errorFrom({}, 404));
+      const expected = new FirebaseDataConnectError('unknown-error', 'Unknown server error: {}');
+      return apiClient_with_connector.executeMutation({operationName: 'getById'})
+        .should.eventually.be.rejected.and.deep.include(expected);
+    });
+
+    it('should reject with unknown-error for non-json response', () => {
+      sandbox
+        .stub(HttpClient.prototype, 'send')
+        .rejects(utils.errorFrom('not json', 404));
+      const expected = new FirebaseDataConnectError(
+        'unknown-error', 'Unexpected response with status: 404 and body: not json');
+      return apiClient_with_connector.executeMutation({operationName: 'getById'})
+        .should.eventually.be.rejected.and.deep.include(expected);
+    });
+
+    it('should reject when rejected with a FirebaseDataConnectError', () => {
+      const expected = new FirebaseDataConnectError('internal-error', 'socket hang up');
+      sandbox
+        .stub(HttpClient.prototype, 'send')
+        .rejects(expected);
+      return apiClient_with_connector.executeMutation({operationName: 'getById'})
+        .should.eventually.be.rejected.and.deep.include(expected);
+    });
+
+    it('should resolve with the Mutation response on success', () => {
+      interface UsersResponse {
+        users: [
+          user: {
+            id: string;
+            name: string;
+            address: string;
+          }
+        ];
+      }
+      const stub = sandbox
+        .stub(HttpClient.prototype, 'send')
+        .resolves(utils.responseFrom(TEST_RESPONSE, 200));
+      return apiClient_with_connector.executeMutation<UsersResponse, unknown>({operationName: 'getById'})
+        .then((resp) => {
+          expect(resp.data.users).to.be.not.empty;
+          expect(resp.data.users[0].name).to.be.not.undefined;
+          expect(resp.data.users[0].address).to.be.not.undefined;
+          expect(resp.data.users).to.deep.equal(TEST_RESPONSE.data.users);
+          expect(stub).to.have.been.calledOnce.and.calledWith({
+            method: 'POST',
+            url: `https://firebasedataconnect.googleapis.com/v1alpha/projects/test-project/locations/${connectorConfig_with_connector.location}/services/${connectorConfig_with_connector.serviceId}/connectors/${connectorConfig_with_connector.connector}:executeMutation`,
+            headers: EXPECTED_HEADERS,
+             data: { query: undefined, name: 'getById', operationName: 'getById' }
+          });
+        });
+    });
+
+    it('should use DATA_CONNECT_EMULATOR_HOST if set', () => {
+      process.env.DATA_CONNECT_EMULATOR_HOST = 'localhost:9399';
+      const stub = sandbox
+        .stub(HttpClient.prototype, 'send')
+        .resolves(utils.responseFrom(TEST_RESPONSE, 200));
+      return apiClient_with_connector.executeMutation({operationName: 'getById'})
+        .then(() => {
+          expect(stub).to.have.been.calledOnce.and.calledWith({
+            method: 'POST',
+            url: `http://localhost:9399/v1alpha/projects/test-project/locations/${connectorConfig_with_connector.location}/services/${connectorConfig_with_connector.serviceId}/connectors/${connectorConfig_with_connector.connector}:executeMutation`,
+            headers: EMULATOR_EXPECTED_HEADERS,
+             data: { query: undefined, name: 'getById', operationName: 'getById' }
+          });
+        });
+    });
   });
 });
 

Original file line number	Diff line number	Diff line change
`@@ -21,3 +21,4 @@ firebase-admin-*.tgz`
`21`	`21`
`22`	`22`	`docgen/markdown/`
`23`	`23`	`service_account.json`
	`24`	`+dataconnect/`