openlattice-ssh 0.0.3

package/tests/integration.test.ts

@@ -0,0 +1,152 @@
+import { describe, it, expect, beforeAll, afterEach } from "vitest";
+import { SSHProvider } from "../src/ssh-provider";
+
+const HAS_SSH = process.env.TEST_SSH === "1";
+
+describe.skipIf(!HAS_SSH)("SSHProvider integration", () => {
+  let provider: SSHProvider;
+  const toCleanup: string[] = [];
+
+  beforeAll(() => {
+    provider = new SSHProvider({
+      hosts: [
+        {
+          host: process.env.SSH_HOST ?? "localhost",
+          port: parseInt(process.env.SSH_PORT ?? "22", 10),
+          username: process.env.SSH_USER ?? "root",
+          privateKey: process.env.SSH_KEY,
+        },
+      ],
+    });
+  });
+
+  afterEach(async () => {
+    for (const id of toCleanup) {
+      try {
+        await provider.destroy(id);
+      } catch {
+        /* best-effort */
+      }
+    }
+    toCleanup.length = 0;
+  });
+
+  it(
+    "provisions, execs, and destroys",
+    async () => {
+      const node = await provider.provision({
+        runtime: { image: "any" },
+      });
+      toCleanup.push(node.externalId);
+
+      expect(node.externalId).toBeTruthy();
+
+      const result = await provider.exec(node.externalId, [
+        "echo",
+        "hello from ssh",
+      ]);
+      expect(result.exitCode).toBe(0);
+      expect(result.stdout).toContain("hello from ssh");
+
+      await provider.destroy(node.externalId);
+      toCleanup.pop();
+
+      const status = await provider.inspect(node.externalId);
+      expect(status.status).toBe("terminated");
+    },
+    30_000
+  );
+
+  it(
+    "exec captures stderr and exit codes",
+    async () => {
+      const node = await provider.provision({
+        runtime: { image: "any" },
+      });
+      toCleanup.push(node.externalId);
+
+      const result = await provider.exec(node.externalId, [
+        "sh",
+        "-c",
+        "echo err >&2; exit 42",
+      ]);
+      expect(result.exitCode).toBe(42);
+      expect(result.stderr).toContain("err");
+    },
+    30_000
+  );
+
+  it(
+    "exec supports cwd and env",
+    async () => {
+      const node = await provider.provision({
+        runtime: { image: "any" },
+      });
+      toCleanup.push(node.externalId);
+
+      const result = await provider.exec(node.externalId, [
+        "sh",
+        "-c",
+        "echo $MY_VAR",
+      ], { env: { MY_VAR: "test123" } });
+      expect(result.stdout).toContain("test123");
+    },
+    30_000
+  );
+
+  it(
+    "inspect returns running status",
+    async () => {
+      const node = await provider.provision({
+        runtime: { image: "any" },
+      });
+      toCleanup.push(node.externalId);
+
+      const status = await provider.inspect(node.externalId);
+      expect(status.status).toBe("running");
+    },
+    30_000
+  );
+
+  it(
+    "healthCheck returns healthy",
+    async () => {
+      const health = await provider.healthCheck();
+      expect(health.healthy).toBe(true);
+    },
+    30_000
+  );
+
+  it(
+    "file extension reads and writes via SFTP",
+    async () => {
+      const node = await provider.provision({
+        runtime: { image: "any" },
+      });
+      toCleanup.push(node.externalId);
+
+      const files = provider.getExtension(node.externalId, "files")!;
+      expect(files).toBeDefined();
+
+      await files.write("/tmp/openlattice-test.txt", "hello ssh");
+      const content = await files.read("/tmp/openlattice-test.txt");
+      expect(content).toContain("hello ssh");
+    },
+    30_000
+  );
+
+  it(
+    "destroy is idempotent",
+    async () => {
+      const node = await provider.provision({
+        runtime: { image: "any" },
+      });
+
+      await provider.destroy(node.externalId);
+      await expect(
+        provider.destroy(node.externalId)
+      ).resolves.toBeUndefined();
+    },
+    30_000
+  );
+});