mirror of
https://github.com/snapotter-hq/SnapOtter.git
synced 2026-08-03 07:46:42 +02:00
test: reorganize flat test files into purpose-based subdirectories (phase 6)
Group 245 flat integration tests and 25 loose unit tests into
discoverable subdirectories per spec section 6:
integration/tools/{image,video,audio,document,data}/ (156 files)
integration/platform/ (64 files)
integration/generated/ (14 files)
integration/security/ (10 files)
unit/security/ (8 files, new subdir)
unit/api/ (7 files moved in)
unit/web/ (3 files moved in)
unit/shared/ (6 files moved in)
unit/image-engine/ (1 file moved in)
All moves via git mv (history preserved). Relative imports repaired
for both depth levels (platform/generated/security = +1, tools/ = +2):
static from-imports, dynamic import() calls, vi.mock() paths,
import.meta.dirname joins, and __dirname joins.
Vitest discovery unchanged (no test.include in config, recursive glob
matches subdirs, shard-by-hash unaffected). test-server.ts and
tool-route-drift.test.ts stay at integration root. fixtures/ untouched.
Parity gate: 13189 passing test names before = 13189 after (0 dropped).
This commit is contained in:
@@ -0,0 +1,210 @@
|
||||
import { execFileSync } from "node:child_process";
|
||||
import { existsSync, mkdirSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from "node:fs";
|
||||
import { tmpdir } from "node:os";
|
||||
import { join } from "node:path";
|
||||
import { afterEach, beforeEach, describe, expect, it } from "vitest";
|
||||
|
||||
/**
|
||||
* Unit tests for the venv upgrade-stamp mechanism in docker/entrypoint.sh.
|
||||
*
|
||||
* The entrypoint compares /opt/venv/.venv-version (baked into the image at
|
||||
* build time) against /data/ai/venv/.venv-version (persisted on the volume).
|
||||
* When missing or mismatched, the venv is nuked and re-copied so upgraded
|
||||
* base packages take effect immediately.
|
||||
*
|
||||
* These tests exercise the bootstrap logic in isolation using temp dirs.
|
||||
*/
|
||||
|
||||
let root: string;
|
||||
let optVenv: string;
|
||||
let dataAi: string;
|
||||
let aiVenv: string;
|
||||
let aiVenvTmp: string;
|
||||
let installedJson: string;
|
||||
|
||||
// Self-contained shell script mirroring the entrypoint bootstrap block.
|
||||
// Uses env vars for paths so we can point at temp directories.
|
||||
const BOOTSTRAP_SCRIPT = `
|
||||
#!/bin/sh
|
||||
set -e
|
||||
|
||||
AI_VENV="$TEST_AI_VENV"
|
||||
AI_VENV_TMP="$TEST_AI_VENV_TMP"
|
||||
OPT_VENV="$TEST_OPT_VENV"
|
||||
|
||||
if [ -d "$AI_VENV_TMP" ]; then
|
||||
echo "CLEANUP_INTERRUPTED"
|
||||
rm -rf "$AI_VENV_TMP"
|
||||
fi
|
||||
|
||||
if [ -d "$OPT_VENV" ]; then
|
||||
NEED_BOOTSTRAP=false
|
||||
|
||||
if [ ! -d "$AI_VENV" ]; then
|
||||
NEED_BOOTSTRAP=true
|
||||
echo "FIRST_RUN"
|
||||
elif [ -f "$OPT_VENV/.venv-version" ]; then
|
||||
IMAGE_STAMP=$(cat "$OPT_VENV/.venv-version")
|
||||
CURRENT_STAMP=""
|
||||
if [ -f "$AI_VENV/.venv-version" ]; then
|
||||
CURRENT_STAMP=$(cat "$AI_VENV/.venv-version")
|
||||
fi
|
||||
if [ "$CURRENT_STAMP" != "$IMAGE_STAMP" ]; then
|
||||
NEED_BOOTSTRAP=true
|
||||
echo "STAMP_MISMATCH"
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ "$NEED_BOOTSTRAP" = true ]; then
|
||||
rm -rf "$AI_VENV"
|
||||
cp -r "$OPT_VENV" "$AI_VENV_TMP"
|
||||
mv "$AI_VENV_TMP" "$AI_VENV"
|
||||
if [ -f "$TEST_INSTALLED_JSON" ]; then
|
||||
echo '{"bundles":{}}' > "$TEST_INSTALLED_JSON"
|
||||
echo "BUNDLES_RESET"
|
||||
fi
|
||||
echo "VENV_READY"
|
||||
else
|
||||
echo "SKIP"
|
||||
fi
|
||||
else
|
||||
echo "NO_OPT_VENV"
|
||||
fi
|
||||
`;
|
||||
|
||||
function runBootstrap(): string {
|
||||
return execFileSync("/bin/sh", ["-c", BOOTSTRAP_SCRIPT], {
|
||||
env: {
|
||||
TEST_AI_VENV: aiVenv,
|
||||
TEST_AI_VENV_TMP: aiVenvTmp,
|
||||
TEST_OPT_VENV: optVenv,
|
||||
TEST_INSTALLED_JSON: installedJson,
|
||||
PATH: process.env.PATH,
|
||||
},
|
||||
encoding: "utf-8",
|
||||
}).trim();
|
||||
}
|
||||
|
||||
beforeEach(() => {
|
||||
root = mkdtempSync(join(tmpdir(), "venv-stamp-"));
|
||||
optVenv = join(root, "opt-venv");
|
||||
dataAi = join(root, "data-ai");
|
||||
aiVenv = join(dataAi, "venv");
|
||||
aiVenvTmp = join(dataAi, "venv.bootstrapping");
|
||||
installedJson = join(dataAi, "installed.json");
|
||||
|
||||
// Simulate /opt/venv with a stamp and a marker file
|
||||
mkdirSync(optVenv, { recursive: true });
|
||||
writeFileSync(join(optVenv, ".venv-version"), "abc123\n");
|
||||
writeFileSync(join(optVenv, "marker.txt"), "base-package-content");
|
||||
|
||||
mkdirSync(dataAi, { recursive: true });
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
rmSync(root, { recursive: true, force: true });
|
||||
});
|
||||
|
||||
describe("venv upgrade stamp", () => {
|
||||
it("bootstraps on first run when no venv exists", () => {
|
||||
const output = runBootstrap();
|
||||
expect(output).toContain("FIRST_RUN");
|
||||
expect(output).toContain("VENV_READY");
|
||||
expect(existsSync(join(aiVenv, ".venv-version"))).toBe(true);
|
||||
expect(readFileSync(join(aiVenv, "marker.txt"), "utf-8")).toBe("base-package-content");
|
||||
});
|
||||
|
||||
it("skips when stamps match", () => {
|
||||
// Simulate an existing venv with matching stamp
|
||||
mkdirSync(aiVenv, { recursive: true });
|
||||
writeFileSync(join(aiVenv, ".venv-version"), "abc123\n");
|
||||
|
||||
const output = runBootstrap();
|
||||
expect(output).toBe("SKIP");
|
||||
});
|
||||
|
||||
it("refreshes when stamps differ (upgrade)", () => {
|
||||
// Simulate stale venv with old stamp
|
||||
mkdirSync(aiVenv, { recursive: true });
|
||||
writeFileSync(join(aiVenv, ".venv-version"), "old-hash\n");
|
||||
writeFileSync(join(aiVenv, "stale-file.txt"), "should-be-removed");
|
||||
|
||||
const output = runBootstrap();
|
||||
expect(output).toContain("STAMP_MISMATCH");
|
||||
expect(output).toContain("VENV_READY");
|
||||
// Old content replaced with fresh copy
|
||||
expect(existsSync(join(aiVenv, "stale-file.txt"))).toBe(false);
|
||||
expect(readFileSync(join(aiVenv, "marker.txt"), "utf-8")).toBe("base-package-content");
|
||||
expect(readFileSync(join(aiVenv, ".venv-version"), "utf-8")).toBe("abc123\n");
|
||||
});
|
||||
|
||||
it("refreshes when venv has no stamp (pre-stamp image upgrade)", () => {
|
||||
// Simulate old venv without any stamp file
|
||||
mkdirSync(aiVenv, { recursive: true });
|
||||
writeFileSync(join(aiVenv, "old-pkg.txt"), "legacy");
|
||||
|
||||
const output = runBootstrap();
|
||||
expect(output).toContain("STAMP_MISMATCH");
|
||||
expect(output).toContain("VENV_READY");
|
||||
expect(existsSync(join(aiVenv, "old-pkg.txt"))).toBe(false);
|
||||
});
|
||||
|
||||
it("resets installed.json when refreshing", () => {
|
||||
// Simulate existing venv + installed bundles
|
||||
mkdirSync(aiVenv, { recursive: true });
|
||||
writeFileSync(join(aiVenv, ".venv-version"), "old-hash\n");
|
||||
writeFileSync(
|
||||
installedJson,
|
||||
JSON.stringify({
|
||||
bundles: {
|
||||
"background-removal": {
|
||||
version: "1.0.0",
|
||||
installedAt: "2025-01-01T00:00:00Z",
|
||||
},
|
||||
},
|
||||
}),
|
||||
);
|
||||
|
||||
const output = runBootstrap();
|
||||
expect(output).toContain("BUNDLES_RESET");
|
||||
const data = JSON.parse(readFileSync(installedJson, "utf-8"));
|
||||
expect(data).toEqual({ bundles: {} });
|
||||
});
|
||||
|
||||
it("does not reset installed.json when no bundles were installed", () => {
|
||||
// No installed.json exists
|
||||
mkdirSync(aiVenv, { recursive: true });
|
||||
writeFileSync(join(aiVenv, ".venv-version"), "old-hash\n");
|
||||
|
||||
const output = runBootstrap();
|
||||
expect(output).toContain("VENV_READY");
|
||||
expect(output).not.toContain("BUNDLES_RESET");
|
||||
});
|
||||
|
||||
it("does nothing when /opt/venv has no stamp (old image)", () => {
|
||||
// Remove the stamp from the image venv
|
||||
rmSync(join(optVenv, ".venv-version"));
|
||||
mkdirSync(aiVenv, { recursive: true });
|
||||
writeFileSync(join(aiVenv, "existing.txt"), "keep");
|
||||
|
||||
const output = runBootstrap();
|
||||
expect(output).toBe("SKIP");
|
||||
// Existing venv untouched
|
||||
expect(existsSync(join(aiVenv, "existing.txt"))).toBe(true);
|
||||
});
|
||||
|
||||
it("cleans up interrupted bootstrap from previous start", () => {
|
||||
mkdirSync(aiVenvTmp, { recursive: true });
|
||||
writeFileSync(join(aiVenvTmp, "partial.txt"), "incomplete");
|
||||
|
||||
const output = runBootstrap();
|
||||
expect(output).toContain("CLEANUP_INTERRUPTED");
|
||||
expect(existsSync(aiVenvTmp)).toBe(false);
|
||||
});
|
||||
|
||||
it("does nothing when /opt/venv does not exist", () => {
|
||||
rmSync(optVenv, { recursive: true });
|
||||
const output = runBootstrap();
|
||||
expect(output).toBe("NO_OPT_VENV");
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user