Skip to content

Commit 420ef9d

Browse files
waleedlatif1waleedlatif1
committed
fix(tests): update audit mock and route tests for new audit actions
1 parent 6c40c14 commit 420ef9d

File tree

4 files changed

+76
-10
lines changed

4 files changed

+76
-10
lines changed

‎apps/sim/app/api/knowledge/[id]/connectors/[connectorId]/documents/route.test.ts‎

Lines changed: 25 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,11 @@ vi.mock('@/lib/auth/hybrid', () => ({
5353
vi.mock('@/lib/core/utils/request', () => ({
5454
generateRequestId: vi.fn().mockReturnValue('test-req-id'),
5555
}))
56+
vi.mock('@/lib/audit/log', () => ({
57+
recordAudit: vi.fn(),
58+
AuditAction: {},
59+
AuditResourceType: {},
60+
}))
5661

5762
import { GET, PATCH } from '@/app/api/knowledge/[id]/connectors/[connectorId]/documents/route'
5863

@@ -168,8 +173,16 @@ describe('Connector Documents API Route', () => {
168173
})
169174

170175
it('returns success for restore operation', async () => {
171-
mockCheckSession.mockResolvedValue({ success: true, userId: 'user-1' })
172-
mockCheckWriteAccess.mockResolvedValue({ hasAccess: true })
176+
mockCheckSession.mockResolvedValue({
177+
success: true,
178+
userId: 'user-1',
179+
userName: 'Test',
180+
userEmail: 'test@test.com',
181+
})
182+
mockCheckWriteAccess.mockResolvedValue({
183+
hasAccess: true,
184+
knowledgeBase: { workspaceId: 'ws-1', name: 'Test KB' },
185+
})
173186
mockDbChain.limit.mockResolvedValueOnce([{ id: 'conn-456' }])
174187
mockDbChain.returning.mockResolvedValueOnce([{ id: 'doc-1' }])
175188

@@ -182,8 +195,16 @@ describe('Connector Documents API Route', () => {
182195
})
183196

184197
it('returns success for exclude operation', async () => {
185-
mockCheckSession.mockResolvedValue({ success: true, userId: 'user-1' })
186-
mockCheckWriteAccess.mockResolvedValue({ hasAccess: true })
198+
mockCheckSession.mockResolvedValue({
199+
success: true,
200+
userId: 'user-1',
201+
userName: 'Test',
202+
userEmail: 'test@test.com',
203+
})
204+
mockCheckWriteAccess.mockResolvedValue({
205+
hasAccess: true,
206+
knowledgeBase: { workspaceId: 'ws-1', name: 'Test KB' },
207+
})
187208
mockDbChain.limit.mockResolvedValueOnce([{ id: 'conn-456' }])
188209
mockDbChain.returning.mockResolvedValueOnce([{ id: 'doc-2' }, { id: 'doc-3' }])
189210

‎apps/sim/app/api/knowledge/[id]/connectors/[connectorId]/route.test.ts‎

Lines changed: 25 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,11 @@ vi.mock('@/lib/knowledge/tags/service', () => ({
7575
vi.mock('@/lib/knowledge/documents/service', () => ({
7676
deleteDocumentStorageFiles: vi.fn().mockResolvedValue(undefined),
7777
}))
78+
vi.mock('@/lib/audit/log', () => ({
79+
recordAudit: vi.fn(),
80+
AuditAction: {},
81+
AuditResourceType: {},
82+
}))
7883

7984
import { DELETE, GET, PATCH } from '@/app/api/knowledge/[id]/connectors/[connectorId]/route'
8085

@@ -183,8 +188,16 @@ describe('Knowledge Connector By ID API Route', () => {
183188
})
184189

185190
it('returns 200 and updates status', async () => {
186-
mockCheckSession.mockResolvedValue({ success: true, userId: 'user-1' })
187-
mockCheckWriteAccess.mockResolvedValue({ hasAccess: true })
191+
mockCheckSession.mockResolvedValue({
192+
success: true,
193+
userId: 'user-1',
194+
userName: 'Test',
195+
userEmail: 'test@test.com',
196+
})
197+
mockCheckWriteAccess.mockResolvedValue({
198+
hasAccess: true,
199+
knowledgeBase: { workspaceId: 'ws-1', name: 'Test KB' },
200+
})
188201

189202
const updatedConnector = { id: 'conn-456', status: 'paused', syncIntervalMinutes: 120 }
190203
mockDbChain.limit.mockResolvedValueOnce([updatedConnector])
@@ -210,8 +223,16 @@ describe('Knowledge Connector By ID API Route', () => {
210223
})
211224

212225
it('returns 200 on successful hard-delete', async () => {
213-
mockCheckSession.mockResolvedValue({ success: true, userId: 'user-1' })
214-
mockCheckWriteAccess.mockResolvedValue({ hasAccess: true })
226+
mockCheckSession.mockResolvedValue({
227+
success: true,
228+
userId: 'user-1',
229+
userName: 'Test',
230+
userEmail: 'test@test.com',
231+
})
232+
mockCheckWriteAccess.mockResolvedValue({
233+
hasAccess: true,
234+
knowledgeBase: { workspaceId: 'ws-1', name: 'Test KB' },
235+
})
215236
mockDbChain.where
216237
.mockReturnValueOnce(mockDbChain)
217238
.mockResolvedValueOnce([{ id: 'doc-1', fileUrl: '/api/uploads/test.txt' }])

‎apps/sim/app/api/knowledge/[id]/connectors/[connectorId]/sync/route.test.ts‎

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,11 @@ vi.mock('@/lib/core/utils/request', () => ({
4343
vi.mock('@/lib/knowledge/connectors/sync-engine', () => ({
4444
dispatchSync: mockDispatchSync,
4545
}))
46+
vi.mock('@/lib/audit/log', () => ({
47+
recordAudit: vi.fn(),
48+
AuditAction: {},
49+
AuditResourceType: {},
50+
}))
4651

4752
import { POST } from '@/app/api/knowledge/[id]/connectors/[connectorId]/sync/route'
4853

@@ -92,8 +97,16 @@ describe('Connector Manual Sync API Route', () => {
9297
})
9398

9499
it('dispatches sync on valid request', async () => {
95-
mockCheckSession.mockResolvedValue({ success: true, userId: 'user-1' })
96-
mockCheckWriteAccess.mockResolvedValue({ hasAccess: true })
100+
mockCheckSession.mockResolvedValue({
101+
success: true,
102+
userId: 'user-1',
103+
userName: 'Test',
104+
userEmail: 'test@test.com',
105+
})
106+
mockCheckWriteAccess.mockResolvedValue({
107+
hasAccess: true,
108+
knowledgeBase: { workspaceId: 'ws-1', name: 'Test KB' },
109+
})
97110
mockDbChain.limit.mockResolvedValueOnce([{ id: 'conn-456', status: 'active' }])
98111

99112
const req = createMockRequest('POST')

‎packages/testing/src/mocks/audit.mock.ts‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,13 +32,16 @@ export const auditMock = {
3232
CREDENTIAL_SET_INVITATION_ACCEPTED: 'credential_set_invitation.accepted',
3333
CREDENTIAL_SET_INVITATION_RESENT: 'credential_set_invitation.resent',
3434
CREDENTIAL_SET_INVITATION_REVOKED: 'credential_set_invitation.revoked',
35+
CONNECTOR_DOCUMENT_RESTORED: 'connector_document.restored',
36+
CONNECTOR_DOCUMENT_EXCLUDED: 'connector_document.excluded',
3537
DOCUMENT_UPLOADED: 'document.uploaded',
3638
DOCUMENT_UPDATED: 'document.updated',
3739
DOCUMENT_DELETED: 'document.deleted',
3840
ENVIRONMENT_UPDATED: 'environment.updated',
3941
FILE_UPLOADED: 'file.uploaded',
4042
FILE_UPDATED: 'file.updated',
4143
FILE_DELETED: 'file.deleted',
44+
FILE_RESTORED: 'file.restored',
4245
FOLDER_CREATED: 'folder.created',
4346
FOLDER_DELETED: 'folder.deleted',
4447
FOLDER_DUPLICATED: 'folder.duplicated',
@@ -47,9 +50,14 @@ export const auditMock = {
4750
FORM_DELETED: 'form.deleted',
4851
INVITATION_ACCEPTED: 'invitation.accepted',
4952
INVITATION_REVOKED: 'invitation.revoked',
53+
CONNECTOR_CREATED: 'connector.created',
54+
CONNECTOR_UPDATED: 'connector.updated',
55+
CONNECTOR_DELETED: 'connector.deleted',
56+
CONNECTOR_SYNCED: 'connector.synced',
5057
KNOWLEDGE_BASE_CREATED: 'knowledge_base.created',
5158
KNOWLEDGE_BASE_UPDATED: 'knowledge_base.updated',
5259
KNOWLEDGE_BASE_DELETED: 'knowledge_base.deleted',
60+
KNOWLEDGE_BASE_RESTORED: 'knowledge_base.restored',
5361
MCP_SERVER_ADDED: 'mcp_server.added',
5462
MCP_SERVER_UPDATED: 'mcp_server.updated',
5563
MCP_SERVER_REMOVED: 'mcp_server.removed',
@@ -80,13 +88,15 @@ export const auditMock = {
8088
TABLE_CREATED: 'table.created',
8189
TABLE_UPDATED: 'table.updated',
8290
TABLE_DELETED: 'table.deleted',
91+
TABLE_RESTORED: 'table.restored',
8392
TEMPLATE_CREATED: 'template.created',
8493
TEMPLATE_UPDATED: 'template.updated',
8594
TEMPLATE_DELETED: 'template.deleted',
8695
WEBHOOK_CREATED: 'webhook.created',
8796
WEBHOOK_DELETED: 'webhook.deleted',
8897
WORKFLOW_CREATED: 'workflow.created',
8998
WORKFLOW_DELETED: 'workflow.deleted',
99+
WORKFLOW_RESTORED: 'workflow.restored',
90100
WORKFLOW_DEPLOYED: 'workflow.deployed',
91101
WORKFLOW_UNDEPLOYED: 'workflow.undeployed',
92102
WORKFLOW_DUPLICATED: 'workflow.duplicated',
@@ -104,6 +114,7 @@ export const auditMock = {
104114
BILLING: 'billing',
105115
BYOK_KEY: 'byok_key',
106116
CHAT: 'chat',
117+
CONNECTOR: 'connector',
107118
CREDENTIAL_SET: 'credential_set',
108119
DOCUMENT: 'document',
109120
ENVIRONMENT: 'environment',

0 commit comments

Comments
 (0)