chore: import upstream snapshot with attribution

apps/sim/executor/handlers/generic/generic-handler.test.ts

@@ -0,0 +1,147 @@
+import '@sim/testing/mocks/executor'
+
+import { beforeEach, describe, expect, it, type Mock, vi } from 'vitest'
+import { BlockType } from '@/executor/constants'
+import { GenericBlockHandler } from '@/executor/handlers/generic/generic-handler'
+import type { ExecutionContext } from '@/executor/types'
+import type { SerializedBlock } from '@/serializer/types'
+import { executeTool } from '@/tools'
+import type { ToolConfig } from '@/tools/types'
+import { getTool } from '@/tools/utils'
+
+const mockGetTool = vi.mocked(getTool)
+const mockExecuteTool = executeTool as Mock
+
+describe('GenericBlockHandler', () => {
+  let handler: GenericBlockHandler
+  let mockBlock: SerializedBlock
+  let mockContext: ExecutionContext
+  let mockTool: ToolConfig
+
+  beforeEach(() => {
+    handler = new GenericBlockHandler()
+
+    mockBlock = {
+      id: 'generic-block-1',
+      metadata: { id: 'custom-type', name: 'Test Generic Block' },
+      position: { x: 40, y: 40 },
+      config: { tool: 'some_custom_tool', params: { param1: 'value1' } },
+      inputs: { param1: 'string' }, // Using ParamType strings
+      outputs: {},
+      enabled: true,
+    }
+
+    mockContext = {
+      workflowId: 'test-workflow-id',
+      blockStates: new Map(),
+      blockLogs: [],
+      metadata: { duration: 0 },
+      environmentVariables: {},
+      decisions: { router: new Map(), condition: new Map() },
+      loopExecutions: new Map(),
+      executedBlocks: new Set(),
+      activeExecutionPath: new Set(),
+      completedLoops: new Set(),
+    }
+
+    mockTool = {
+      id: 'some_custom_tool',
+      name: 'Some Custom Tool',
+      description: 'Does something custom',
+      version: '1.0',
+      params: { param1: { type: 'string' } },
+      request: {
+        url: 'https://example.com/api',
+        method: 'POST',
+        headers: () => ({ 'Content-Type': 'application/json' }),
+        body: (params) => params,
+      },
+    }
+
+    // Reset mocks using vi
+    vi.clearAllMocks()
+
+    // Set up mockGetTool to return mockTool
+    mockGetTool.mockImplementation((toolId) => {
+      if (toolId === 'some_custom_tool') {
+        return mockTool
+      }
+      return undefined
+    })
+
+    // Default mock implementations
+    mockExecuteTool.mockResolvedValue({ success: true, output: { customResult: 'OK' } })
+  })
+
+  it.concurrent('should always handle any block type', () => {
+    const agentBlock: SerializedBlock = { ...mockBlock, metadata: { id: BlockType.AGENT } }
+    expect(handler.canHandle(agentBlock)).toBe(true)
+    expect(handler.canHandle(mockBlock)).toBe(true)
+    const noMetaIdBlock: SerializedBlock = { ...mockBlock, metadata: undefined }
+    expect(handler.canHandle(noMetaIdBlock)).toBe(true)
+  })
+
+  it.concurrent('should execute generic block by calling its associated tool', async () => {
+    const inputs = { param1: 'resolvedValue1' }
+    const expectedToolParams = {
+      ...inputs,
+      _context: { workflowId: mockContext.workflowId },
+    }
+    const expectedOutput: any = { customResult: 'OK' }
+
+    const result = await handler.execute(mockContext, mockBlock, inputs)
+
+    expect(mockGetTool).toHaveBeenCalledWith('some_custom_tool')
+    expect(mockExecuteTool).toHaveBeenCalledWith('some_custom_tool', expectedToolParams, {
+      executionContext: mockContext,
+    })
+    expect(result).toEqual(expectedOutput)
+  })
+
+  it('should throw error if the associated tool is not found', async () => {
+    const inputs = { param1: 'value' }
+
+    // Override mock to return undefined for this test
+    mockGetTool.mockImplementation(() => undefined)
+
+    await expect(handler.execute(mockContext, mockBlock, inputs)).rejects.toThrow(
+      'Tool not found: some_custom_tool'
+    )
+    expect(mockExecuteTool).not.toHaveBeenCalled()
+  })
+
+  it('should handle tool execution errors correctly', async () => {
+    const inputs = { param1: 'value' }
+    const errorResult = {
+      success: false,
+      error: 'Custom tool failed',
+      output: { detail: 'error detail' },
+    }
+    mockExecuteTool.mockResolvedValue(errorResult)
+
+    await expect(handler.execute(mockContext, mockBlock, inputs)).rejects.toThrow(
+      'Custom tool failed'
+    )
+
+    // Re-execute to check error properties after catching
+    try {
+      await handler.execute(mockContext, mockBlock, inputs)
+    } catch (e: any) {
+      expect(e.toolId).toBe('some_custom_tool')
+      expect(e.blockName).toBe('Test Generic Block')
+      expect(e.output).toEqual({ detail: 'error detail' })
+    }
+
+    expect(mockExecuteTool).toHaveBeenCalledTimes(2) // Called twice now
+  })
+
+  it.concurrent('should handle tool execution errors with no specific message', async () => {
+    const inputs = { param1: 'value' }
+    const errorResult = { success: false, output: {} }
+    mockExecuteTool.mockResolvedValue(errorResult)
+
+    await expect(handler.execute(mockContext, mockBlock, inputs)).rejects.toThrow(
+      'Block execution of Some Custom Tool failed with no error message'
+    )
+  })
+})