initial transfer
This commit is contained in:
@@ -0,0 +1,115 @@
|
||||
/**
|
||||
* 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']);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user