update tests

Author: paula-stacho

packages/compass-global-writes/src/services/atlas-global-writes-service.ts

@@ -37,7 +37,7 @@ type ReplicationItem = {
   zoneId: string;
   zoneName: string;
 };
-type ClusterDetailsApiResponse = {
+export type ClusterDetailsApiResponse = {
   geoSharding: GeoShardingData;
   replicationSpecList: ReplicationItem[];
 };

packages/compass-global-writes/src/store/index.spec.ts

@@ -7,12 +7,22 @@ import {
   type CreateShardKeyData,
 } from './reducer';
 import sinon from 'sinon';
+import type { ClusterDetailsApiResponse } from '../services/atlas-global-writes-service';
 
 const DB = 'test';
 const COLL = 'coll';
 const NS = `${DB}.${COLL}`;
 
-function createJsonResponse(data: any) {
+const clusterDetails: ClusterDetailsApiResponse = {
+  geoSharding: {
+    customZoneMapping: {},
+    managedNamespaces: [],
+    selfManagedSharding: false,
+  },
+  replicationSpecList: [],
+};
+
+function createClusterDetailsResponse(data: ClusterDetailsApiResponse) {
   return {
     json: () => Promise.resolve(data),
   };
@@ -41,13 +51,11 @@ describe('GlobalWritesStore Store', function () {
       it('when the namespace is not managed', async function () {
         const store = createStore({
           authenticatedFetch: () =>
-            createJsonResponse({
-              geoSharding: { customZoneMapping: {}, managedNamespaces: [] },
-            }),
+            createClusterDetailsResponse(clusterDetails),
         });
         await store.dispatch(fetchClusterShardingData());
         expect(store.getState().status).to.equal('UNSHARDED');
-        expect(store.getState().isNamespaceSharded).to.equal(false);
+        expect(store.getState().managedNamespace).to.equal(undefined);
       });
 
       // TODO (COMPASS-8277): Add more test for fetching shard key and process errors
@@ -65,9 +73,7 @@ describe('GlobalWritesStore Store', function () {
       it('sets SUBMITTING_FOR_SHARDING state when starting to create shard key and sets to SHARDING on success', async function () {
         const store = createStore({
           authenticatedFetch: () =>
-            createJsonResponse({
-              geoSharding: { customZoneMapping: {}, managedNamespaces: [] },
-            }),
+            createClusterDetailsResponse(clusterDetails),
         });
 
         const promise = store.dispatch(createShardKey(shardKeyData));
@@ -102,9 +108,10 @@ describe('GlobalWritesStore Store', function () {
           },
         ];
 
-        const getClusterInfoApiResponse = createJsonResponse({
+        const getClusterInfoApiResponse = createClusterDetailsResponse({
+          ...clusterDetails,
           geoSharding: {
-            customZoneMapping: {},
+            ...clusterDetails.geoSharding,
             managedNamespaces: alreadyManagedNamespaces,
           },
         });
@@ -137,6 +144,7 @@ describe('GlobalWritesStore Store', function () {
             ...alreadyManagedNamespaces,
             { ...shardKeyData, db: DB, collection: COLL },
           ],
+          selfManagedSharding: false,
         });
       });
     });

packages/compass-global-writes/tests/create-store.tsx

@@ -7,16 +7,25 @@ import { activateGlobalWritesPlugin } from '../src/store';
 import { createActivateHelpers } from 'hadron-app-registry';
 import { createNoopLogger } from '@mongodb-js/compass-logging/provider';
 import { createNoopTrack } from '@mongodb-js/compass-telemetry/provider';
-import type {
-  ConnectionInfo,
-  ConnectionInfoRef,
-} from '@mongodb-js/compass-connections/provider';
+import type { ConnectionInfo } from '@mongodb-js/compass-connections/provider';
 import type { AtlasService } from '@mongodb-js/atlas-service/provider';
 import { Provider } from 'react-redux';
 import { renderWithActiveConnection } from '@mongodb-js/testing-library-compass';
 
 import clusterApiResponse from './cluster-api-response.json';
 
+const TEST_CONNECTION_INFO = {
+  id: 'TEST',
+  connectionOptions: {
+    connectionString: 'mongodb://localhost',
+  },
+  atlasMetadata: {
+    clusterName: 'Cluster0',
+    clusterType: 'UNSHARDED',
+    projectId: 'Project0',
+  } as unknown as ConnectionInfo['atlasMetadata'],
+};
+
 const atlasService = {
   cloudEndpoint: (p: string) => {
     return `https://example.com/${p}`;
@@ -39,19 +48,9 @@ const atlasService = {
 
 export const setupStore = (
   options: Partial<GlobalWritesPluginOptions> = {},
-  services: Partial<GlobalWritesPluginServices> = {}
+  services: Partial<GlobalWritesPluginServices> = {},
+  connectionInfo: ConnectionInfo = TEST_CONNECTION_INFO
 ) => {
-  const connectionInfoRef = {
-    current: {
-      id: 'TEST',
-      atlasMetadata: {
-        clusterName: 'Cluster0',
-        clusterType: 'GEOSHARDED',
-        projectId: 'Project0',
-      },
-    },
-  } as ConnectionInfoRef;
-
   return activateGlobalWritesPlugin(
     {
       namespace: 'airbnb.listings',
@@ -60,7 +59,12 @@ export const setupStore = (
     {
       logger: createNoopLogger('TEST'),
       track: createNoopTrack(),
-      connectionInfoRef,
+      connectionInfoRef: {
+        current: {
+          ...connectionInfo,
+          title: 'My connection',
+        },
+      },
       ...services,
       atlasService: {
         ...atlasService,
@@ -76,19 +80,14 @@ export const renderWithStore = (
   {
     services = {},
     options = {},
-    connectionInfo = {
-      id: 'testConnection',
-      connectionOptions: {
-        connectionString: 'mongodb://localhost',
-      },
-    },
+    connectionInfo = TEST_CONNECTION_INFO,
   }: {
     services?: Partial<GlobalWritesPluginServices>;
     options?: Partial<GlobalWritesPluginOptions>;
     connectionInfo?: ConnectionInfo;
   } = {}
 ) => {
-  const store = setupStore(options, services);
+  const store = setupStore(options, services, connectionInfo);
   return renderWithActiveConnection(
     <Provider store={store}>{component}</Provider>,
     connectionInfo