@elevasis/core 0.1.0 → 0.2.1

package/src/execution/engine/tools/platform/resource-invocation/__tests__/resource-invocation-service.test.ts

@@ -1,500 +0,0 @@
-import { describe, it, expect, vi, beforeEach } from 'vitest'
-import { z } from 'zod'
-import { ResourceInvocationService, createResourceInvocationService } from '../resource-invocation-service'
-import type { ResourceRegistry } from '../../../../../../platform/registry/resource-registry'
-import type { WorkflowDefinition } from '../../../../workflow/types'
-import type { AgentDefinition } from '../../../../agent/core/types'
-import type { ResourceInvocationResult, ExecuteResourceCallback } from '../types'
-import { createMockExecutionContext } from '../../../../test-utils/mocks'
-
-describe('ResourceInvocationService', () => {
-  // Mock data
-  const mockContext = createMockExecutionContext({
-    organizationId: 'org-123',
-    organizationName: 'test-org',
-    executionId: 'exec-456',
-    resourceId: 'parent-agent',
-    executionDepth: 0
-  })
-
-  const mockWorkflowDefinition: WorkflowDefinition = {
-    config: {
-      resourceId: 'validation-workflow',
-      name: 'Validation Workflow',
-      description: 'Validates data',
-      version: '1.0.0',
-      type: 'workflow',
-      status: 'dev'
-    },
-    contract: {
-      inputSchema: z.object({
-        email: z.string().email(),
-        name: z.string()
-      }),
-      outputSchema: z.object({
-        valid: z.boolean(),
-        errors: z.array(z.string())
-      })
-    },
-    entryPoint: 'validate',
-    steps: {}
-  }
-
-  const mockAgentDefinition: AgentDefinition = {
-    config: {
-      resourceId: 'research-agent',
-      name: 'Research Agent',
-      description: 'Researches topics',
-      version: '1.0.0',
-      type: 'agent',
-      status: 'dev'
-    },
-    contract: {
-      inputSchema: z.object({
-        query: z.string()
-      }),
-      outputSchema: z.object({
-        results: z.string()
-      })
-    },
-    modelConfig: {
-      provider: 'openai',
-      model: 'gpt-4o',
-      temperature: 0.7
-    },
-    systemPrompt: 'You are a research assistant',
-    tools: []
-  }
-
-  let mockRegistry: ResourceRegistry
-  let mockExecuteResource: ReturnType<typeof vi.fn>
-  let service: ResourceInvocationService
-
-  beforeEach(() => {
-    // Create mock registry
-    mockRegistry = {
-      getResourceDefinition: vi.fn(),
-      getRelationships: vi.fn().mockReturnValue(undefined)
-    } as unknown as ResourceRegistry
-
-    // Create mock execute function
-    mockExecuteResource = vi.fn()
-
-    // Create service instance
-    service = new ResourceInvocationService(mockRegistry, mockExecuteResource)
-  })
-
-  describe('executeSync', () => {
-    it('validates resource exists', async () => {
-      // Setup: Mock registry returns undefined for resourceId
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(null)
-
-      const result = await service.executeSync(
-        'non-existent-resource',
-        { email: 'test@test.com', name: 'Test' },
-        'test-org',
-        mockContext
-      )
-
-      expect(result.success).toBe(false)
-      expect(result.error).toContain('not found')
-      expect(result.error).toContain('non-existent-resource')
-      expect(mockRegistry.getResourceDefinition).toHaveBeenCalledWith('test-org', 'non-existent-resource')
-    })
-
-    it('invokes workflow and returns result', async () => {
-      // Setup: Mock registry returns workflow definition
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-
-      // Mock successful execution
-      const mockResult: ResourceInvocationResult = {
-        success: true,
-        output: { valid: true, errors: [] },
-        executionId: 'exec-789',
-        resourceType: 'workflow'
-      }
-      mockExecuteResource.mockResolvedValue(mockResult)
-
-      const result = await service.executeSync(
-        'validation-workflow',
-        { email: 'test@test.com', name: 'Test User' },
-        'test-org',
-        mockContext
-      )
-
-      expect(result.success).toBe(true)
-      expect(result.resourceType).toBe('workflow')
-      expect(result.output).toEqual({ valid: true, errors: [] })
-      expect(result.executionId).toBe('exec-789')
-
-      // Verify executeResource called with correct parameters
-      expect(mockExecuteResource).toHaveBeenCalledWith(
-        {
-          resourceId: 'validation-workflow',
-          input: { email: 'test@test.com', name: 'Test User' },
-          organizationId: 'org-123',
-          organizationName: 'test-org'
-        },
-        mockContext
-      )
-    })
-
-    it('invokes agent and returns result', async () => {
-      // Setup: Mock registry returns agent definition
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockAgentDefinition)
-
-      // Mock successful execution
-      const mockResult: ResourceInvocationResult = {
-        success: true,
-        output: { results: 'Research findings...' },
-        executionId: 'exec-999',
-        resourceType: 'agent'
-      }
-      mockExecuteResource.mockResolvedValue(mockResult)
-
-      const result = await service.executeSync(
-        'research-agent',
-        { query: 'AI trends' },
-        'test-org',
-        mockContext
-      )
-
-      expect(result.success).toBe(true)
-      expect(result.resourceType).toBe('agent')
-      expect(result.output).toEqual({ results: 'Research findings...' })
-      expect(result.executionId).toBe('exec-999')
-    })
-
-    it('validates input against resource contract', async () => {
-      // Setup: Mock registry returns workflow with Zod schema
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-
-      // Invalid input (missing required fields)
-      const result = await service.executeSync(
-        'validation-workflow',
-        { email: 'invalid-email' }, // Missing 'name', invalid email
-        'test-org',
-        mockContext
-      )
-
-      expect(result.success).toBe(false)
-      expect(result.error).toContain('Input validation failed')
-      expect(mockExecuteResource).not.toHaveBeenCalled()
-    })
-
-    it('passes parent context to nested execution', async () => {
-      // Setup: Mock registry returns workflow
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-
-      // Mock executor that captures context
-      const capturedContext = vi.fn()
-      mockExecuteResource.mockImplementation(async (request, context) => {
-        capturedContext(context)
-        return {
-          success: true,
-          output: { valid: true, errors: [] },
-          executionId: 'exec-nested',
-          resourceType: 'workflow'
-        }
-      })
-
-      await service.executeSync(
-        'validation-workflow',
-        { email: 'test@test.com', name: 'Test' },
-        'test-org',
-        mockContext
-      )
-
-      // Verify parent context was passed
-      expect(capturedContext).toHaveBeenCalledWith(mockContext)
-    })
-
-    it('handles execution errors from executor', async () => {
-      // Setup: Mock registry returns workflow
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-
-      // Mock failed execution
-      const mockResult: ResourceInvocationResult = {
-        success: false,
-        output: null,
-        executionId: 'exec-fail',
-        resourceType: 'workflow',
-        error: 'Execution timeout'
-      }
-      mockExecuteResource.mockResolvedValue(mockResult)
-
-      const result = await service.executeSync(
-        'validation-workflow',
-        { email: 'test@test.com', name: 'Test' },
-        'test-org',
-        mockContext
-      )
-
-      expect(result.success).toBe(false)
-      expect(result.error).toBe('Execution timeout')
-      expect(result.resourceType).toBe('workflow')
-    })
-
-    it('preserves organization context through invocation', async () => {
-      // Setup: Mock registry returns workflow
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-
-      mockExecuteResource.mockResolvedValue({
-        success: true,
-        output: { valid: true, errors: [] },
-        executionId: 'exec-org',
-        resourceType: 'workflow'
-      })
-
-      await service.executeSync(
-        'validation-workflow',
-        { email: 'test@test.com', name: 'Test' },
-        'test-org',
-        mockContext
-      )
-
-      // Verify organizationId and organizationName passed correctly
-      expect(mockExecuteResource).toHaveBeenCalledWith(
-        expect.objectContaining({
-          organizationId: 'org-123',
-          organizationName: 'test-org'
-        }),
-        mockContext
-      )
-    })
-
-    it('handles workflow with null outputSchema', async () => {
-      // Create workflow with no outputSchema (fire-and-forget)
-      const fireAndForgetWorkflow: WorkflowDefinition = {
-        ...mockWorkflowDefinition,
-        contract: {
-          inputSchema: z.object({ data: z.string() }),
-          outputSchema: null
-        }
-      }
-
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(fireAndForgetWorkflow)
-
-      mockExecuteResource.mockResolvedValue({
-        success: true,
-        output: null,
-        executionId: 'exec-fire',
-        resourceType: 'workflow'
-      })
-
-      const result = await service.executeSync(
-        'fire-and-forget-workflow',
-        { data: 'test' },
-        'test-org',
-        mockContext
-      )
-
-      expect(result.success).toBe(true)
-      expect(result.output).toBeNull()
-    })
-  })
-
-  describe('createResourceInvocationService', () => {
-    it('creates service with valid dependencies', () => {
-      const service = createResourceInvocationService(mockRegistry, mockExecuteResource)
-
-      expect(service).toBeInstanceOf(ResourceInvocationService)
-    })
-
-    it('throws if registry not provided', () => {
-      expect(() => {
-        createResourceInvocationService(null as unknown as ResourceRegistry, mockExecuteResource)
-      }).toThrow('ResourceRegistry required')
-    })
-
-    it('throws if executeResource callback not provided', () => {
-      expect(() => {
-        createResourceInvocationService(mockRegistry, null as unknown as ExecuteResourceCallback)
-      }).toThrow('executeResource callback required')
-    })
-  })
-
-  describe('multi-tenancy safety', () => {
-    it('scopes resource lookup by organization name', async () => {
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-
-      mockExecuteResource.mockResolvedValue({
-        success: true,
-        output: { valid: true, errors: [] },
-        executionId: 'exec-tenant',
-        resourceType: 'workflow'
-      })
-
-      await service.executeSync(
-        'validation-workflow',
-        { email: 'test@test.com', name: 'Test' },
-        'acme-corp',
-        mockContext
-      )
-
-      // Verify registry called with correct organization name
-      expect(mockRegistry.getResourceDefinition).toHaveBeenCalledWith('acme-corp', 'validation-workflow')
-    })
-
-    it('passes organization context to nested execution', async () => {
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-
-      const contextWithOrg = createMockExecutionContext({
-        organizationId: 'org-different',
-        organizationName: 'different-org'
-      })
-
-      mockExecuteResource.mockResolvedValue({
-        success: true,
-        output: { valid: true, errors: [] },
-        executionId: 'exec-ctx',
-        resourceType: 'workflow'
-      })
-
-      await service.executeSync(
-        'validation-workflow',
-        { email: 'test@test.com', name: 'Test' },
-        'different-org',
-        contextWithOrg
-      )
-
-      // Verify correct organization context passed
-      expect(mockExecuteResource).toHaveBeenCalledWith(
-        expect.objectContaining({
-          organizationId: 'org-different',
-          organizationName: 'different-org'
-        }),
-        contextWithOrg
-      )
-    })
-  })
-
-  describe('relationship enforcement', () => {
-    it('allows invocation when relationship is declared (workflow target)', async () => {
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-      vi.mocked(mockRegistry.getRelationships).mockReturnValue({
-        'parent-agent': {
-          triggers: { workflows: ['validation-workflow'] }
-        }
-      })
-
-      mockExecuteResource.mockResolvedValue({
-        success: true,
-        output: { valid: true, errors: [] },
-        executionId: 'exec-rel',
-        resourceType: 'workflow'
-      })
-
-      const result = await service.executeSync(
-        'validation-workflow',
-        { email: 'test@test.com', name: 'Test' },
-        'test-org',
-        mockContext
-      )
-
-      expect(result.success).toBe(true)
-    })
-
-    it('allows invocation when relationship is declared (agent target)', async () => {
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockAgentDefinition)
-      vi.mocked(mockRegistry.getRelationships).mockReturnValue({
-        'parent-agent': {
-          triggers: { agents: ['research-agent'] }
-        }
-      })
-
-      mockExecuteResource.mockResolvedValue({
-        success: true,
-        output: { results: 'findings' },
-        executionId: 'exec-agent-rel',
-        resourceType: 'agent'
-      })
-
-      const result = await service.executeSync(
-        'research-agent',
-        { query: 'test' },
-        'test-org',
-        mockContext
-      )
-
-      expect(result.success).toBe(true)
-    })
-
-    it('throws when invocation is not declared', async () => {
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-      vi.mocked(mockRegistry.getRelationships).mockReturnValue({
-        'parent-agent': {
-          triggers: { workflows: ['other-workflow'] }
-        }
-      })
-
-      await expect(
-        service.executeSync(
-          'validation-workflow',
-          { email: 'test@test.com', name: 'Test' },
-          'test-org',
-          mockContext
-        )
-      ).rejects.toThrow('Undeclared resource invocation')
-    })
-
-    it('throws when caller has no entry in relationships', async () => {
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-      vi.mocked(mockRegistry.getRelationships).mockReturnValue({
-        'some-other-resource': {
-          triggers: { workflows: ['validation-workflow'] }
-        }
-      })
-
-      await expect(
-        service.executeSync(
-          'validation-workflow',
-          { email: 'test@test.com', name: 'Test' },
-          'test-org',
-          mockContext
-        )
-      ).rejects.toThrow('Undeclared resource invocation')
-    })
-
-    it('skips enforcement when org has no relationships (undefined)', async () => {
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-      vi.mocked(mockRegistry.getRelationships).mockReturnValue(undefined)
-
-      mockExecuteResource.mockResolvedValue({
-        success: true,
-        output: { valid: true, errors: [] },
-        executionId: 'exec-no-rel',
-        resourceType: 'workflow'
-      })
-
-      const result = await service.executeSync(
-        'validation-workflow',
-        { email: 'test@test.com', name: 'Test' },
-        'test-org',
-        mockContext
-      )
-
-      expect(result.success).toBe(true)
-    })
-
-    it('error message includes actionable fix instructions', async () => {
-      vi.mocked(mockRegistry.getResourceDefinition).mockReturnValue(mockWorkflowDefinition)
-      vi.mocked(mockRegistry.getRelationships).mockReturnValue({
-        'parent-agent': {
-          triggers: { workflows: ['other-workflow'] }
-        }
-      })
-
-      await expect(
-        service.executeSync(
-          'validation-workflow',
-          { email: 'test@test.com', name: 'Test' },
-          'test-org',
-          mockContext
-        )
-      ).rejects.toThrow(
-        `Add to relationships: { 'parent-agent': { triggers: { workflows: ['validation-workflow'] } } }`
-      )
-    })
-  })
-})