ansiblings/packages/nopy/tests/cubes.dependencies.test.ts

/**
 * Tests for cubes/dependencies module (BuildContext)
 */

import { describe, expect, it, vi } from 'vitest';
import { z } from 'zod';
import { BuildContext } from '../src/cubes/dependencies.js';
import { Cube, Manifest } from '../src/cubes/types.js';
import { Variables } from '../src/nopy.common.js';

// Mock VariableAssignment to avoid hanging on prompts
vi.mock('../src/nopy.prompts.js', async () => {
  const actual = await vi.importActual('../src/nopy.prompts.js');
  return {
    ...actual,
    VariableAssignment: vi.fn(),
  };
});

/**
 * Helper to create a minimal cube for testing
 */
function createTestCube(id: string, dependencies?: (vars: any) => any[]): Cube {
  const manifest = Manifest.create({
    id,
    name: `Test ${id}`,
    schema: z.object({}),
    dependencies,
  });
  return new Cube(manifest, `/test/${id}`, 'deploy.py');
}

describe('BuildContext.resolveCube', () => {
  it('resolves a single cube with no dependencies', async () => {
    const cubeA = createTestCube('cube-a');
    const cubes = { 'cube-a': cubeA };
    const vars = new Variables();
    const context = new BuildContext(cubes, vars, { cubes: [] } as any, { env: {} } as any, {
      method: 'ssh',
    });

    await context.resolveCube('cube-a', 'host1');

    expect(context.deployCalls).toHaveLength(1);
    expect(context.deployCalls[0].cube).toBe('cube-a');
    expect(context.deployCalls[0].host).toBe('host1');
  });

  it('resolves dependencies recursively', async () => {
    const cubeA = createTestCube('cube-a');
    const cubeB = createTestCube('cube-b', () => ['cube-a']);
    const cubes = { 'cube-a': cubeA, 'cube-b': cubeB };
    const vars = new Variables();
    const context = new BuildContext(cubes, vars, { cubes: [] } as any, { env: {} } as any, {
      method: 'ssh',
    });

    await context.resolveCube('cube-b', 'host1');

    expect(context.deployCalls).toHaveLength(2);
    // Dependencies should be resolved BEFORE the cube that depends on them
    expect(context.deployCalls[0].cube).toBe('cube-a');
    expect(context.deployCalls[1].cube).toBe('cube-b');
  });

  it('resolves dynamic dependencies based on variables', async () => {
    const cubeA = createTestCube('cube-a');
    const cubeB = createTestCube('cube-b');
    const cubeC = createTestCube('cube-c', (vars) => (vars.USE_A ? ['cube-a'] : ['cube-b']));

    cubeC.manifest.schema = z.object({ USE_A: z.boolean().default(true) });

    const cubes = { 'cube-a': cubeA, 'cube-b': cubeB, 'cube-c': cubeC };

    // Test with USE_A = true
    const vars1 = new Variables();
    const context1 = new BuildContext(cubes, vars1, { cubes: [] } as any, { env: {} } as any, {
      method: 'ssh',
    });
    await context1.resolveCube('cube-c', 'host1');
    expect(context1.deployCalls.map((c) => c.cube)).toEqual(['cube-a', 'cube-c']);

    // Test with USE_A = false
    const vars2 = new Variables();
    vars2.assign('cube-c', 'params', { USE_A: false });
    const context2 = new BuildContext(cubes, vars2, { cubes: [] } as any, { env: {} } as any, {
      method: 'ssh',
    });
    await context2.resolveCube('cube-c', 'host1');
    expect(context2.deployCalls.map((c) => c.cube)).toEqual(['cube-b', 'cube-c']);
  });

  it('passes variables to dependencies', async () => {
    const cubeA = createTestCube('cube-a');
    cubeA.manifest.schema = z.object({ VAR: z.string() });

    const cubeB = createTestCube('cube-b', () => [['cube-a', { VAR: 'from-b' }]]);

    const cubes = { 'cube-a': cubeA, 'cube-b': cubeB };
    const vars = new Variables();
    const context = new BuildContext(cubes, vars, { cubes: [] } as any, { env: {} } as any, {
      method: 'ssh',
    });

    await context.resolveCube('cube-b', 'host1');

    expect(context.deployCalls[0].cube).toBe('cube-a');
    expect(context.deployCalls[0].env.VAR).toBe('from-b');
  });

  it('avoids duplicate calls for the same cube on the same host', async () => {
    const cubeA = createTestCube('cube-a');
    const cubeB = createTestCube('cube-b', () => ['cube-a']);
    const cubeC = createTestCube('cube-c', () => ['cube-a', 'cube-b']);

    const cubes = { 'cube-a': cubeA, 'cube-b': cubeB, 'cube-c': cubeC };
    const vars = new Variables();
    const context = new BuildContext(cubes, vars, { cubes: [] } as any, { env: {} } as any, {
      method: 'ssh',
    });

    await context.resolveCube('cube-c', 'host1');

    // Execution order: cube-a, cube-b, cube-c
    expect(context.deployCalls.map((c) => c.cube)).toEqual(['cube-a', 'cube-b', 'cube-c']);
  });
});