Add GrpcFetch and InProcess Unit Tests

Signed-off-by: marcozabel <[email protected]>

Author: marcozabel

libs/providers/flagd/src/lib/service/in-process/grpc/grpc-fetch.spec.ts

@@ -18,6 +18,7 @@ const disconnectCallback = jest.fn();
 const removeAllListeners = jest.fn();
 const cancel = jest.fn();
 const destroy = jest.fn();
+const setSyncContext = jest.fn();
 
 let onDataCallback: (data: SyncFlagsResponse) => void = () => ({});
 let onErrorCallback: (err: Error) => void = () => ({});
@@ -59,11 +60,12 @@ describe('grpc fetch', () => {
 
   it('should handle data sync and emit callbacks', (done) => {
     const flagConfiguration = '{"flags":{}}';
-    const fetch = new GrpcFetch(cfg, serviceMock);
+    const fetch = new GrpcFetch(cfg, setSyncContext, serviceMock);
     fetch
       .connect(dataCallback, reconnectCallback, jest.fn(), disconnectCallback)
       .then(() => {
         try {
+          expect(setSyncContext).toHaveBeenCalledTimes(0);
           expect(dataCallback).toHaveBeenCalledTimes(1);
           expect(dataCallback).toHaveBeenCalledWith(flagConfiguration);
           expect(changedCallback).toHaveBeenCalledTimes(0);
@@ -80,14 +82,40 @@ describe('grpc fetch', () => {
     onDataCallback({ flagConfiguration });
   });
 
+  it('should handle SyncContext from SyncFlagsResponse', (done) => {
+    const initFlagConfig = '{"flags":{}}';
+    const syncContext = {"test": "example"};
+
+    const fetch = new GrpcFetch(cfg, setSyncContext, serviceMock);
+    fetch
+      .connect(dataCallback, reconnectCallback, changedCallback, disconnectCallback)
+      .then(() => {
+        try {
+          // Callback assertions
+          expect(setSyncContext).toHaveBeenCalledTimes(1);
+          expect(setSyncContext).toHaveBeenCalledWith(syncContext);
+
+          done();
+        } catch (err) {
+          done(err);
+        }
+      })
+      .catch((err) => {
+        done(err);
+      });
+
+    // First connection
+    onDataCallback({ flagConfiguration: initFlagConfig, syncContext: syncContext });
+  });
+
   it('should handle data sync reconnection', (done) => {
     const initFlagConfig = '{"flags":{}}';
     const updatedFlagConfig =
       '{"flags":{"test":{"state":"ENABLED","variants":{"on":true,"off":false},"defaultVariant":"off"}}}';
     const reconnectFlagConfig =
       '{"flags":{"test":{"state":"ENABLED","variants":{"on":true,"off":false},"defaultVariant":"on"}}}';
 
-    const fetch = new GrpcFetch(cfg, serviceMock);
+    const fetch = new GrpcFetch(cfg, jest.fn(), serviceMock);
     fetch
       .connect(dataCallback, reconnectCallback, changedCallback, disconnectCallback)
       .then(() => {
@@ -128,7 +156,7 @@ describe('grpc fetch', () => {
   });
 
   it('should handle error and watch channel for reconnect', (done) => {
-    const fetch = new GrpcFetch(cfg, serviceMock);
+    const fetch = new GrpcFetch(cfg, jest.fn(), serviceMock);
     fetch.connect(jest.fn(), jest.fn(), jest.fn(), disconnectCallback).catch((err) => {
       try {
         expect(err).toBeInstanceOf(Error);

libs/providers/flagd/src/lib/service/in-process/in-process-service.spec.ts

@@ -13,7 +13,7 @@ describe('In-process-service', () => {
 
   it('should sync and allow to resolve flags', async () => {
     // given
-    const service = new InProcessService({ deadlineMs: 500, host: '', port: 0, tls: false }, dataFetcher);
+    const service = new InProcessService({ deadlineMs: 500, host: '', port: 0, tls: false }, jest.fn(), dataFetcher);
 
     // when
     await service.connect(jest.fn, jest.fn, jest.fn);
@@ -31,7 +31,7 @@ describe('In-process-service', () => {
     it('should include scope as flag metadata', async () => {
       // given
       const selector = 'devFlags';
-      const service = new InProcessService({ deadlineMs: 500, host: '', port: 0, tls: false, selector }, dataFetcher);
+      const service = new InProcessService({ deadlineMs: 500, host: '', port: 0, tls: false, selector }, jest.fn(), dataFetcher);
 
       // when
       await service.connect(jest.fn, jest.fn, jest.fn);
@@ -44,7 +44,7 @@ describe('In-process-service', () => {
     it('should not override existing scope in flag metadata', async () => {
       // given
       const selector = 'devFlags';
-      const service = new InProcessService({ deadlineMs: 500, host: '', port: 0, tls: false, selector }, dataFetcher);
+      const service = new InProcessService({ deadlineMs: 500, host: '', port: 0, tls: false, selector }, jest.fn(), dataFetcher);
 
       // when
       await service.connect(jest.fn, jest.fn, jest.fn);