ktx/packages/cli/src/runtime-requirements.test.ts

import { MANAGED_SENTENCE_TRANSFORMERS_BASE_URL } from '@ktx/context';
import { buildDefaultKtxProjectConfig, type KtxProjectConfig } from '@ktx/context/project';
import { describe, expect, it } from 'vitest';
import {
  resolveProjectRuntimeRequirements,
  resolvePublicIngestRuntimeRequirements,
} from './runtime-requirements.js';

describe('runtime requirement detection', () => {
  it('does not require runtime for agent/MCP setup alone', () => {
    const config = buildDefaultKtxProjectConfig();

    expect(resolveProjectRuntimeRequirements(config).features).toEqual([]);
  });

  it('requires core for Looker source ingest unless an external daemon is configured', () => {
    const config: KtxProjectConfig = {
      ...buildDefaultKtxProjectConfig(),
      connections: {
        looker: { driver: 'looker', base_url: 'https://looker.example.com', client_id: 'client-id' },
      },
    };

    expect(resolveProjectRuntimeRequirements(config).features).toEqual(['core']);
    expect(resolveProjectRuntimeRequirements(config, { env: { KTX_DAEMON_URL: 'http://127.0.0.1:8765' } }).features).toEqual(
      [],
    );
  });

  it('requires core for query-history ingest unless SQL analysis is externally configured', () => {
    const config: KtxProjectConfig = {
      ...buildDefaultKtxProjectConfig(),
      connections: {
        warehouse: { driver: 'postgres', context: { queryHistory: { enabled: true } } },
      },
    };

    expect(resolveProjectRuntimeRequirements(config).features).toEqual(['core']);
    expect(
      resolveProjectRuntimeRequirements(config, { env: { KTX_SQL_ANALYSIS_URL: 'http://127.0.0.1:8765' } }).features,
    ).toEqual([]);
  });

  it('requires local-embeddings for managed sentence-transformers embeddings', () => {
    const config: KtxProjectConfig = {
      ...buildDefaultKtxProjectConfig(),
      ingest: {
        ...buildDefaultKtxProjectConfig().ingest,
        embeddings: {
          backend: 'sentence-transformers' as const,
          model: 'all-MiniLM-L6-v2',
          dimensions: 384,
          sentenceTransformers: {
            base_url: MANAGED_SENTENCE_TRANSFORMERS_BASE_URL,
          },
        },
      },
    };

    expect(resolveProjectRuntimeRequirements(config).features).toEqual(['local-embeddings']);
  });

  it('detects foreground ingest runtime needs from selected query-history targets', () => {
    expect(
      resolvePublicIngestRuntimeRequirements({
        projectDir: '/tmp/project',
        warnings: [],
        targets: [
          {
            connectionId: 'warehouse',
            driver: 'postgres',
            operation: 'database-ingest',
            debugCommand: 'ktx ingest warehouse --debug',
            steps: ['database-schema', 'query-history'],
            queryHistory: { enabled: true },
          },
        ],
      }).features,
    ).toEqual(['core']);
  });
});