2024-09-01 15:54:16 -07:00
|
|
|
import { SerializedContinueConfig } from "core";
|
|
|
|
|
// import Mock from "core/llm/llms/Mock.js";
|
2024-06-21 11:09:03 -07:00
|
|
|
import { FromIdeProtocol, ToIdeProtocol } from "core/protocol/index.js";
|
2024-12-05 21:48:06 -08:00
|
|
|
import { IMessenger } from "core/protocol/messenger";
|
2024-12-24 14:03:23 -05:00
|
|
|
import FileSystemIde from "core/util/filesystem";
|
2024-06-21 11:09:03 -07:00
|
|
|
import fs from "fs";
|
2024-09-21 10:55:40 -07:00
|
|
|
import {
|
|
|
|
|
ChildProcessWithoutNullStreams,
|
|
|
|
|
execSync,
|
|
|
|
|
spawn,
|
|
|
|
|
} from "node:child_process";
|
2024-06-21 11:09:03 -07:00
|
|
|
import path from "path";
|
|
|
|
|
import {
|
|
|
|
|
CoreBinaryMessenger,
|
|
|
|
|
CoreBinaryTcpMessenger,
|
|
|
|
|
} from "../src/IpcMessenger";
|
|
|
|
|
|
|
|
|
|
// jest.setTimeout(100_000);
|
|
|
|
|
|
|
|
|
|
const USE_TCP = false;
|
|
|
|
|
|
|
|
|
|
function autodetectPlatformAndArch() {
|
|
|
|
|
const platform = {
|
|
|
|
|
aix: "linux",
|
|
|
|
|
darwin: "darwin",
|
|
|
|
|
freebsd: "linux",
|
|
|
|
|
linux: "linux",
|
|
|
|
|
openbsd: "linux",
|
|
|
|
|
sunos: "linux",
|
|
|
|
|
win32: "win32",
|
|
|
|
|
android: "linux",
|
|
|
|
|
cygwin: "win32",
|
|
|
|
|
netbsd: "linux",
|
|
|
|
|
haiku: "linux",
|
|
|
|
|
}[process.platform];
|
|
|
|
|
const arch = {
|
|
|
|
|
arm: "arm64",
|
|
|
|
|
arm64: "arm64",
|
|
|
|
|
ia32: "x64",
|
|
|
|
|
loong64: "arm64",
|
|
|
|
|
mips: "arm64",
|
|
|
|
|
mipsel: "arm64",
|
|
|
|
|
ppc: "x64",
|
|
|
|
|
ppc64: "x64",
|
|
|
|
|
riscv64: "arm64",
|
|
|
|
|
s390: "x64",
|
|
|
|
|
s390x: "x64",
|
|
|
|
|
x64: "x64",
|
|
|
|
|
}[process.arch];
|
|
|
|
|
return [platform, arch];
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
const CONTINUE_GLOBAL_DIR = path.join(__dirname, "..", ".continue");
|
2024-09-01 15:54:16 -07:00
|
|
|
if (fs.existsSync(CONTINUE_GLOBAL_DIR)) {
|
|
|
|
|
fs.rmSync(CONTINUE_GLOBAL_DIR, { recursive: true, force: true });
|
|
|
|
|
}
|
|
|
|
|
fs.mkdirSync(CONTINUE_GLOBAL_DIR);
|
2024-06-21 11:09:03 -07:00
|
|
|
|
|
|
|
|
describe("Test Suite", () => {
|
|
|
|
|
let messenger: IMessenger<ToIdeProtocol, FromIdeProtocol>;
|
|
|
|
|
let subprocess: ChildProcessWithoutNullStreams;
|
|
|
|
|
|
|
|
|
|
beforeAll(async () => {
|
|
|
|
|
const [platform, arch] = autodetectPlatformAndArch();
|
2024-09-21 10:55:40 -07:00
|
|
|
const binaryDir = path.join(__dirname, "..", "bin", `${platform}-${arch}`);
|
|
|
|
|
const exe = platform === "win32" ? ".exe" : "";
|
|
|
|
|
const binaryPath = path.join(binaryDir, `continue-binary${exe}`);
|
|
|
|
|
const expectedItems = [
|
|
|
|
|
`continue-binary${exe}`,
|
|
|
|
|
`esbuild${exe}`,
|
|
|
|
|
"index.node",
|
|
|
|
|
"package.json",
|
|
|
|
|
"build/Release/node_sqlite3.node",
|
|
|
|
|
];
|
|
|
|
|
expectedItems.forEach((item) => {
|
|
|
|
|
expect(fs.existsSync(path.join(binaryDir, item))).toBe(true);
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
// Set execute permissions and remove quarantine attribute if on macOS
|
|
|
|
|
if (platform !== "win32") {
|
|
|
|
|
try {
|
|
|
|
|
fs.chmodSync(binaryPath, 0o755);
|
|
|
|
|
console.log("Execute permissions set for the binary");
|
|
|
|
|
|
2024-09-21 15:30:28 -07:00
|
|
|
if (platform === "darwin") {
|
2024-09-21 10:55:40 -07:00
|
|
|
const indexNodePath = path.join(binaryDir, "index.node");
|
|
|
|
|
const filesToUnquarantine = [binaryPath, indexNodePath];
|
|
|
|
|
|
|
|
|
|
for (const file of filesToUnquarantine) {
|
|
|
|
|
try {
|
|
|
|
|
execSync(`xattr -d com.apple.quarantine "${file}"`, {
|
|
|
|
|
stdio: "ignore",
|
|
|
|
|
});
|
|
|
|
|
console.log(
|
|
|
|
|
`Quarantine attribute removed from ${path.basename(file)}`,
|
|
|
|
|
);
|
|
|
|
|
} catch (error) {
|
|
|
|
|
console.warn(
|
|
|
|
|
`Failed to remove quarantine attribute from ${path.basename(file)}:`,
|
|
|
|
|
error,
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
} catch (error) {
|
|
|
|
|
console.error(
|
|
|
|
|
"Error setting permissions or removing quarantine:",
|
|
|
|
|
error,
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
}
|
2024-06-21 11:09:03 -07:00
|
|
|
|
|
|
|
|
if (USE_TCP) {
|
|
|
|
|
messenger = new CoreBinaryTcpMessenger<ToIdeProtocol, FromIdeProtocol>();
|
|
|
|
|
} else {
|
2024-09-21 10:55:40 -07:00
|
|
|
try {
|
|
|
|
|
subprocess = spawn(binaryPath, {
|
|
|
|
|
env: { ...process.env, CONTINUE_GLOBAL_DIR },
|
|
|
|
|
});
|
2024-09-21 15:30:28 -07:00
|
|
|
console.log("Successfully spawned subprocess");
|
2024-09-21 10:55:40 -07:00
|
|
|
} catch (error) {
|
|
|
|
|
console.error("Error spawning subprocess:", error);
|
|
|
|
|
throw error;
|
|
|
|
|
}
|
2024-06-21 11:09:03 -07:00
|
|
|
messenger = new CoreBinaryMessenger<ToIdeProtocol, FromIdeProtocol>(
|
|
|
|
|
subprocess,
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
|
2024-07-05 22:40:19 -07:00
|
|
|
const testDir = path.join(__dirname, "..", ".test");
|
|
|
|
|
if (!fs.existsSync(testDir)) {
|
|
|
|
|
fs.mkdirSync(testDir);
|
|
|
|
|
}
|
|
|
|
|
const ide = new FileSystemIde(testDir);
|
2024-12-05 21:48:06 -08:00
|
|
|
// const reverseIde = new ReverseMessageIde(messenger.on.bind(messenger), ide);
|
2024-06-21 11:09:03 -07:00
|
|
|
|
|
|
|
|
// Wait for core to set itself up
|
|
|
|
|
await new Promise((resolve) => setTimeout(resolve, 1000));
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
afterAll(async () => {
|
|
|
|
|
// Wait for the subprocess to exit
|
|
|
|
|
if (USE_TCP) {
|
|
|
|
|
(
|
|
|
|
|
messenger as CoreBinaryTcpMessenger<ToIdeProtocol, FromIdeProtocol>
|
|
|
|
|
).close();
|
|
|
|
|
} else {
|
|
|
|
|
subprocess.kill();
|
|
|
|
|
await new Promise((resolve) => subprocess.on("close", resolve));
|
|
|
|
|
await new Promise((resolve) => setTimeout(resolve, 1000));
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
it("should respond to ping with pong", async () => {
|
|
|
|
|
const resp = await messenger.request("ping", "ping");
|
|
|
|
|
expect(resp).toBe("pong");
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
it("should create .continue directory at the specified location with expected files", async () => {
|
|
|
|
|
expect(fs.existsSync(CONTINUE_GLOBAL_DIR)).toBe(true);
|
|
|
|
|
|
|
|
|
|
// Many of the files are only created when trying to load the config
|
|
|
|
|
const config = await messenger.request(
|
2024-07-08 21:51:53 -07:00
|
|
|
"config/getSerializedProfileInfo",
|
2024-06-21 11:09:03 -07:00
|
|
|
undefined,
|
|
|
|
|
);
|
|
|
|
|
|
|
|
|
|
const expectedFiles = [
|
|
|
|
|
"config.json",
|
|
|
|
|
"config.ts",
|
|
|
|
|
"package.json",
|
|
|
|
|
"logs/core.log",
|
|
|
|
|
"index/autocompleteCache.sqlite",
|
|
|
|
|
"types/core/index.d.ts",
|
|
|
|
|
];
|
|
|
|
|
|
2024-09-21 11:02:52 -07:00
|
|
|
const missingFiles = expectedFiles.filter((file) => {
|
2024-06-21 11:09:03 -07:00
|
|
|
const filePath = path.join(CONTINUE_GLOBAL_DIR, file);
|
2024-09-21 11:02:52 -07:00
|
|
|
return !fs.existsSync(filePath);
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
expect(missingFiles).toEqual([]);
|
|
|
|
|
if (missingFiles.length > 0) {
|
|
|
|
|
console.log("Missing files:", missingFiles);
|
2024-06-21 11:09:03 -07:00
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
2024-09-01 15:54:16 -07:00
|
|
|
it("should return valid config object", async () => {
|
2024-12-24 14:03:23 -05:00
|
|
|
const { result } = await messenger.request(
|
2024-07-08 21:51:53 -07:00
|
|
|
"config/getSerializedProfileInfo",
|
2024-06-21 11:09:03 -07:00
|
|
|
undefined,
|
|
|
|
|
);
|
2024-12-24 14:03:23 -05:00
|
|
|
const { config } = result;
|
2024-06-21 11:09:03 -07:00
|
|
|
expect(config).toHaveProperty("models");
|
|
|
|
|
expect(config).toHaveProperty("embeddingsProvider");
|
|
|
|
|
expect(config).toHaveProperty("contextProviders");
|
|
|
|
|
expect(config).toHaveProperty("slashCommands");
|
|
|
|
|
});
|
2024-09-01 15:54:16 -07:00
|
|
|
|
|
|
|
|
it("should properly handle history requests", async () => {
|
|
|
|
|
const sessionId = "test-session-id";
|
|
|
|
|
await messenger.request("history/save", {
|
|
|
|
|
history: [],
|
|
|
|
|
sessionId,
|
|
|
|
|
title: "test-title",
|
|
|
|
|
|
|
|
|
|
workspaceDirectory: "test-workspace-directory",
|
|
|
|
|
});
|
|
|
|
|
const sessions = await messenger.request("history/list", {});
|
|
|
|
|
expect(sessions.length).toBeGreaterThan(0);
|
|
|
|
|
|
|
|
|
|
const session = await messenger.request("history/load", {
|
|
|
|
|
id: sessionId,
|
|
|
|
|
});
|
|
|
|
|
expect(session).toHaveProperty("history");
|
|
|
|
|
|
|
|
|
|
await messenger.request("history/delete", {
|
|
|
|
|
id: sessionId,
|
|
|
|
|
});
|
|
|
|
|
const sessionsAfterDelete = await messenger.request("history/list", {});
|
|
|
|
|
expect(sessionsAfterDelete.length).toBe(sessions.length - 1);
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
it("should add and delete a model from config.json", async () => {
|
|
|
|
|
const model: SerializedContinueConfig["models"][number] = {
|
|
|
|
|
title: "Test Model",
|
|
|
|
|
provider: "openai",
|
|
|
|
|
model: "gpt-3.5-turbo",
|
2025-05-25 20:27:23 -07:00
|
|
|
underlyingProviderName: "openai",
|
2024-09-01 15:54:16 -07:00
|
|
|
};
|
|
|
|
|
await messenger.request("config/addModel", {
|
|
|
|
|
model,
|
|
|
|
|
});
|
2024-12-24 14:03:23 -05:00
|
|
|
const {
|
|
|
|
|
result: { config },
|
|
|
|
|
} = await messenger.request("config/getSerializedProfileInfo", undefined);
|
|
|
|
|
|
2025-04-08 19:28:48 -07:00
|
|
|
expect(config!.modelsByRole.chat.some((m) => m.title === model.title)).toBe(
|
|
|
|
|
true,
|
|
|
|
|
);
|
2024-09-01 15:54:16 -07:00
|
|
|
|
|
|
|
|
await messenger.request("config/deleteModel", { title: model.title });
|
2024-12-24 14:03:23 -05:00
|
|
|
const {
|
|
|
|
|
result: { config: configAfterDelete },
|
|
|
|
|
} = await messenger.request("config/getSerializedProfileInfo", undefined);
|
2025-04-08 19:28:48 -07:00
|
|
|
expect(
|
|
|
|
|
configAfterDelete!.modelsByRole.chat.some((m) => m.title === model.title),
|
|
|
|
|
).toBe(false);
|
2024-09-01 15:54:16 -07:00
|
|
|
});
|
|
|
|
|
|
|
|
|
|
it("should make an LLM completion", async () => {
|
|
|
|
|
const model: SerializedContinueConfig["models"][number] = {
|
|
|
|
|
title: "Test Model",
|
|
|
|
|
provider: "mock",
|
|
|
|
|
model: "gpt-3.5-turbo",
|
2025-05-25 20:27:23 -07:00
|
|
|
underlyingProviderName: "mock",
|
2024-09-01 15:54:16 -07:00
|
|
|
};
|
|
|
|
|
await messenger.request("config/addModel", {
|
|
|
|
|
model,
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
const resp = await messenger.request("llm/complete", {
|
|
|
|
|
prompt: "Say 'Hello' and nothing else",
|
|
|
|
|
completionOptions: {},
|
|
|
|
|
title: "Test Model",
|
|
|
|
|
});
|
|
|
|
|
expect(resp).toBe("Test Completion");
|
|
|
|
|
});
|
2024-06-21 11:09:03 -07:00
|
|
|
});
|
