fix(data-connect): Include connector in DataConnect cache key

src/data-connect/data-connect.ts

@@ -36,7 +36,7 @@ export class DataConnectService {
   }
 
   getDataConnect(connectorConfig: ConnectorConfig): DataConnect {
-    const id = `${connectorConfig.location}-${connectorConfig.serviceId}`;
+    const id = `${connectorConfig.location}-${connectorConfig.serviceId}-${connectorConfig.connector ?? ''}`;
     const dc = this.dataConnectInstances.get(id);
     if (typeof dc !== 'undefined') {
       return dc;

test/unit/data-connect/data-connect.spec.ts

@@ -224,3 +224,112 @@ describe('DataConnect', () => {
     });
   });
 });
+
+describe('getDataConnect() caching', () => {
+  const mockOptions = {
+    credential: new mocks.MockCredential(),
+    projectId: 'test-project',
+  };
+  let mockApp: FirebaseApp;
+  let getAppStub: sinon.SinonStub;
+
+  beforeEach(() => {
+    mockApp = mocks.appWithOptions(mockOptions);
+    getAppStub = sinon.stub(appIndex, 'getApp').returns(mockApp);
+  });
+
+  afterEach(() => {
+    getAppStub.restore();
+    return mockApp.delete();
+  });
+
+  describe('should cache DataConnect instances correctly', () => {
+    it('should return the same instance for identical connector configs', () => {
+      const config: ConnectorConfig = {
+        location: 'us-west2',
+        serviceId: 'my-service',
+        connector: 'my-connector',
+      };
+
+      const dc1 = getDataConnect(config);
+      const dc2 = getDataConnect(config);
+
+      expect(dc1).to.equal(dc2);
+    });
+
+    it('should return different instances for different connectors with same location and serviceId', () => {
+      const config1: ConnectorConfig = {
+        location: 'us-west2',
+        serviceId: 'my-service',
+        connector: 'connector-a',
+      };
+      const config2: ConnectorConfig = {
+        location: 'us-west2',
+        serviceId: 'my-service',
+        connector: 'connector-b',
+      };
+
+      const dc1 = getDataConnect(config1);
+      const dc2 = getDataConnect(config2);
+
+      expect(dc1).to.not.equal(dc2);
+      expect(dc1.connectorConfig.connector).to.equal('connector-a');
+      expect(dc2.connectorConfig.connector).to.equal('connector-b');
+    });
+
+    it('should return different instances for different locations', () => {
+      const config1: ConnectorConfig = {
+        location: 'us-west2',
+        serviceId: 'my-service',
+        connector: 'my-connector',
+      };
+      const config2: ConnectorConfig = {
+        location: 'us-east1',
+        serviceId: 'my-service',
+        connector: 'my-connector',
+      };
+
+      const dc1 = getDataConnect(config1);
+      const dc2 = getDataConnect(config2);
+
+      expect(dc1).to.not.equal(dc2);
+    });
+
+    it('should return different instances for different serviceIds', () => {
+      const config1: ConnectorConfig = {
+        location: 'us-west2',
+        serviceId: 'service-a',
+        connector: 'my-connector',
+      };
+      const config2: ConnectorConfig = {
+        location: 'us-west2',
+        serviceId: 'service-b',
+        connector: 'my-connector',
+      };
+
+      const dc1 = getDataConnect(config1);
+      const dc2 = getDataConnect(config2);
+
+      expect(dc1).to.not.equal(dc2);
+    });
+
+    it('should handle connector being undefined vs defined', () => {
+      const configWithConnector: ConnectorConfig = {
+        location: 'us-west2',
+        serviceId: 'my-service',
+        connector: 'my-connector',
+      };
+      const configWithoutConnector: ConnectorConfig = {
+        location: 'us-west2',
+        serviceId: 'my-service',
+      };
+
+      const dc1 = getDataConnect(configWithConnector);
+      const dc2 = getDataConnect(configWithoutConnector);
+
+      expect(dc1).to.not.equal(dc2);
+      expect(dc1.connectorConfig.connector).to.equal('my-connector');
+      expect(dc2.connectorConfig.connector).to.be.undefined;
+    });
+  });
+});