Fix tests

Author: cezudas

packages/blocks/openops-tables/test/actions/delete-record-action.test.ts

@@ -233,9 +233,9 @@ function validateWrapperCall(context: any) {
   expect(cacheWrapperMock.getOrAdd).toHaveBeenCalledTimes(2);
   expect(cacheWrapperMock.getOrAdd).toHaveBeenNthCalledWith(
     1,
-    `${context.run.id}-table-${context.propsValue.tableName}`,
+    `${context.run.id}-table-${context.propsValue.tableName}-some databaseToken`,
     getTableIdByTableName,
-    [context.propsValue.tableName],
+    [context.propsValue.tableName, 'some databaseToken', false],
   );
   expect(cacheWrapperMock.getOrAdd).toHaveBeenNthCalledWith(
     2,

packages/blocks/openops-tables/test/actions/get-records-action.test.ts

@@ -96,6 +96,9 @@ describe('getRecordsAction test', () => {
       expect(openopsCommonMock.getTableFields).toHaveBeenCalledTimes(1);
       expect(openopsCommonMock.getTableFields).toHaveBeenCalledWith(
         'Opportunity',
+        undefined,
+        undefined,
+        false,
       );
     });
   });
@@ -313,9 +316,9 @@ function validateWrapperCall(context: any) {
   expect(cacheWrapperMock.getOrAdd).toHaveBeenCalledTimes(1);
   expect(cacheWrapperMock.getOrAdd).toHaveBeenNthCalledWith(
     1,
-    `${context.run.id}-table-${context.propsValue.tableName}`,
+    `${context.run.id}-table-${context.propsValue.tableName}-some databaseToken`,
     getTableIdByTableName,
-    [context.propsValue.tableName],
+    [context.propsValue.tableName, 'some databaseToken', false],
   );
 }
 

packages/blocks/openops-tables/test/actions/get-table-url-action.test.ts

@@ -15,6 +15,9 @@ jest.mock('@openops/server-shared', () => ({
 const openopsCommonMock = {
   ...jest.requireActual('@openops/common'),
   getTableIdByTableName: jest.fn().mockReturnValue(1),
+  getProjectDatabaseTokenForBlock: jest
+    .fn()
+    .mockResolvedValue('some databaseToken'),
   openopsTablesDropdownProperty: jest.fn().mockReturnValue({
     required: true,
     defaultValue: false,
@@ -47,6 +50,14 @@ describe('getTableUrlAction', () => {
 
     const context = {
       ...jest.requireActual('@openops/blocks-framework'),
+      server: {
+        apiUrl: 'https://api.example.com',
+        token: 'test-token',
+        publicUrl: 'https://app.example.com',
+      },
+      project: {
+        id: 'test-project-id',
+      },
       propsValue: {
         tableName: 'my table',
       },
@@ -58,9 +69,21 @@ describe('getTableUrlAction', () => {
       'https://some-url/tables?path=/database/1/table/123',
     );
 
+    expect(
+      openopsCommonMock.getProjectDatabaseTokenForBlock,
+    ).toHaveBeenCalledTimes(1);
+    expect(
+      openopsCommonMock.getProjectDatabaseTokenForBlock,
+    ).toHaveBeenCalledWith(
+      'https://api.example.com',
+      'test-token',
+      'test-project-id',
+    );
     expect(openopsCommonMock.getTableIdByTableName).toHaveBeenCalledTimes(1);
     expect(openopsCommonMock.getTableIdByTableName).toHaveBeenCalledWith(
       'my table',
+      'some databaseToken',
+      false,
     );
     expect(systemMock.getOrThrow).toHaveBeenCalledTimes(1);
     expect(systemMock.getOrThrow).toHaveBeenCalledWith('FRONTEND_URL');

packages/blocks/openops-tables/test/actions/update-record-action.test.ts

@@ -370,6 +370,9 @@ describe('fieldsProperties property', () => {
     expect(openopsCommonMock.getTableFields).toHaveBeenCalledTimes(1);
     expect(openopsCommonMock.getTableFields).toHaveBeenCalledWith(
       'Opportunity',
+      undefined,
+      'some databaseToken',
+      false,
     );
   });
 });
@@ -378,9 +381,9 @@ function validateWrapperCall(context: any) {
   expect(cacheWrapperMock.getOrAdd).toHaveBeenCalledTimes(2);
   expect(cacheWrapperMock.getOrAdd).toHaveBeenNthCalledWith(
     1,
-    `${context.run.id}-table-${context.propsValue.tableName}`,
+    `${context.run.id}-table-${context.propsValue.tableName}-some databaseToken`,
     getTableIdByTableName,
-    [context.propsValue.tableName],
+    [context.propsValue.tableName, 'some databaseToken', false],
   );
   expect(cacheWrapperMock.getOrAdd).toHaveBeenNthCalledWith(
     2,

packages/openops/test/openops-tables/openops-tables-common-properties.test.ts

@@ -17,6 +17,12 @@ jest.mock('../../src/lib/openops-tables/auth-user', () => {
       authenticateDefaultUserInOpenOpsTablesMock,
   };
 });
+const getProjectDatabaseTokenForBlockMock = jest.fn();
+jest.mock('../../src/lib/openops-tables/get-encrypted-token-for-block', () => {
+  return {
+    getProjectDatabaseTokenForBlock: getProjectDatabaseTokenForBlockMock,
+  };
+});
 
 import {
   DurationOpenOpsField,
@@ -33,13 +39,25 @@ describe('table property', () => {
     jest.clearAllMocks();
   });
 
+  const mockContext = {
+    server: {
+      apiUrl: 'https://api.example.com',
+      token: 'test-token',
+      publicUrl: 'https://app.example.com',
+    },
+    project: {
+      id: 'test-project-id',
+    },
+  };
+
   test('should return dropdown with all tables', async () => {
-    authenticateDefaultUserInOpenOpsTablesMock.mockResolvedValue({
-      token: 'token',
-    });
+    getProjectDatabaseTokenForBlockMock.mockResolvedValue('token');
     getTableNamesMock.mockResolvedValue(['table1', 'table2']);
 
-    const result = await openopsTablesDropdownProperty().options();
+    const result = await openopsTablesDropdownProperty().options(
+      {},
+      mockContext as any,
+    );
 
     expect(result).toMatchObject({
       disabled: false,
@@ -49,12 +67,17 @@ describe('table property', () => {
       ],
     });
     expect(getTableNamesMock).toHaveBeenCalledTimes(1);
+    expect(getTableNamesMock).toHaveBeenCalledWith('token', false);
   });
 
   test('should handle empty tables', async () => {
+    getProjectDatabaseTokenForBlockMock.mockResolvedValue('token');
     getTableNamesMock.mockResolvedValue([]);
 
-    const result = await openopsTablesDropdownProperty().options();
+    const result = await openopsTablesDropdownProperty().options(
+      {},
+      mockContext as any,
+    );
 
     expect(result).toMatchObject({
       disabled: false,
@@ -116,7 +139,13 @@ describe('getTableFields', () => {
     ]);
     expect(authenticateDefaultUserInOpenOpsTablesMock).toHaveBeenCalledTimes(1);
     expect(getFieldsMock).toHaveBeenCalledTimes(1);
-    expect(getFieldsMock).toHaveBeenCalledWith(1, 'token', false, undefined); // Adjust this line as needed.
+    expect(getFieldsMock).toHaveBeenCalledWith(
+      1,
+      'token',
+      false,
+      undefined,
+      true,
+    );
   });
 
   test('should handle authentication failure', async () => {
@@ -144,7 +173,13 @@ describe('getTableFields', () => {
     expect(result).toMatchObject([]);
     expect(authenticateDefaultUserInOpenOpsTablesMock).toHaveBeenCalledTimes(1);
     expect(getFieldsMock).toHaveBeenCalledTimes(1);
-    expect(getFieldsMock).toHaveBeenCalledWith(1, 'token', false, undefined);
+    expect(getFieldsMock).toHaveBeenCalledWith(
+      1,
+      'token',
+      false,
+      undefined,
+      true,
+    );
   });
 
   test('should return our supported properties for the field', async () => {