Merge pull request #84 from coder/missing-tests

Author: stirby

code-server/main.test.ts

@@ -0,0 +1,12 @@
+import { describe, expect, it } from "bun:test";
+import { runTerraformInit, testRequiredVariables } from "../test";
+
+describe("code-server", async () => {
+  await runTerraformInit(import.meta.dir);
+
+  testRequiredVariables(import.meta.dir, {
+    agent_id: "foo",
+  });
+
+  // More tests depend on shebang refactors
+});

coder-login/main.tf

@@ -23,7 +23,7 @@ resource "coder_script" "coder-login" {
     CODER_DEPLOYMENT_URL : data.coder_workspace.me.access_url
   })
   display_name       = "Coder Login"
-  icon               = "http://svgur.com/i/y5G.svg"
+  icon               = "/icon/coder.svg"
   run_on_start       = true
   start_blocks_login = true
 }

dotfiles/main.test.ts

@@ -0,0 +1,21 @@
+import { describe, expect, it } from "bun:test";
+import {
+  runTerraformApply,
+  runTerraformInit,
+  testRequiredVariables,
+} from "../test";
+
+describe("dotfiles", async () => {
+  await runTerraformInit(import.meta.dir);
+
+  testRequiredVariables(import.meta.dir, {
+    agent_id: "foo",
+  });
+
+  it("default output", async () => {
+    const state = await runTerraformApply(import.meta.dir, {
+      agent_id: "foo",
+    });
+    expect(state.outputs.dotfiles_uri.value).toBe("");
+  });
+});

fly-region/main.test.ts

@@ -1,6 +1,5 @@
 import { describe, expect, it } from "bun:test";
 import {
-  executeScriptInContainer,
   runTerraformApply,
   runTerraformInit,
   testRequiredVariables,
@@ -22,4 +21,12 @@ describe("fly-region", async () => {
     });
     expect(state.outputs.value.value).toBe("atl");
   });
+
+  it("region filter", async () => {
+    const state = await runTerraformApply(import.meta.dir, {
+      default: "atl",
+      regions: '["arn", "ams", "bos"]',
+    });
+    expect(state.outputs.value.value).toBe("");
+  });
 });

gcp-region/README.md

@@ -28,6 +28,8 @@ resource "google_compute_instance" "example" {
 
 ### Add only GPU zones in the US West 1 region
 
+Note: setting `gpu_only = true` and using a default region without GPU support, the default will be set to `null`.
+
 ```hcl
 module "gcp_region" {
     source   = "https://registry.coder.com/modules/gcp-region"

gcp-region/main.test.ts

@@ -0,0 +1,43 @@
+import { describe, expect, it } from "bun:test";
+import {
+  runTerraformApply,
+  runTerraformInit,
+  testRequiredVariables,
+} from "../test";
+
+describe("gcp-region", async () => {
+  await runTerraformInit(import.meta.dir);
+
+  testRequiredVariables(import.meta.dir, {});
+
+  it("default output", async () => {
+    const state = await runTerraformApply(import.meta.dir, {});
+    expect(state.outputs.value.value).toBe("");
+  });
+
+  it("customized default", async () => {
+    const state = await runTerraformApply(import.meta.dir, {
+      regions: '["asia"]',
+      default: "asia-east1-a",
+    });
+    expect(state.outputs.value.value).toBe("asia-east1-a");
+  });
+
+  it("gpu only invalid default", async () => {
+    const state = await runTerraformApply(import.meta.dir, {
+      regions: '["us-west2"]',
+      default: "us-west2-a",
+      gpu_only: "true",
+    });
+    expect(state.outputs.value.value).toBe("");
+  });
+
+  it("gpu only valid default", async () => {
+    const state = await runTerraformApply(import.meta.dir, {
+      regions: '["us-west2"]',
+      default: "us-west2-b",
+      gpu_only: "true",
+    });
+    expect(state.outputs.value.value).toBe("us-west2-b");
+  });
+});

gcp-region/main.tf

@@ -714,7 +714,7 @@ data "coder_parameter" "region" {
   description  = var.description
   icon         = "/icon/gcp.png"
   mutable      = var.mutable
-  default      = var.default != null && var.default != "" ? var.default : null
+  default      = var.default != null && var.default != "" && (!var.gpu_only || try(local.zones[var.default].gpu, false)) ? var.default : null
   dynamic "option" {
     for_each = {
       for k, v in local.zones : k => v

jetbrains-gateway/main.test.ts

@@ -1,7 +1,12 @@
-import { describe } from "bun:test";
-import { runTerraformInit, testRequiredVariables } from "../test";
+import { it, expect, describe } from "bun:test";
+import {
+  runTerraformInit,
+  testRequiredVariables,
+  executeScriptInContainer,
+  runTerraformApply,
+} from "../test";
 
-describe("jetbrains-gateway`", async () => {
+describe("jetbrains-gateway", async () => {
   await runTerraformInit(import.meta.dir);
 
   await testRequiredVariables(import.meta.dir, {
@@ -10,4 +15,16 @@ describe("jetbrains-gateway`", async () => {
     folder: "/baz/",
     jetbrains_ides: '["IU", "IC", "PY"]',
   });
+
+  it("default to first ide", async () => {
+    const state = await runTerraformApply(import.meta.dir, {
+      agent_id: "foo",
+      agent_name: "bar",
+      folder: "/baz/",
+      jetbrains_ides: '["IU", "IC", "PY"]',
+    });
+    expect(state.outputs.jetbrains_ides.value).toBe(
+      '["IU","232.9921.47","https://download.jetbrains.com/idea/ideaIU-2023.2.2.tar.gz"]',
+    );
+  });
 });

jupyterlab/main.test.ts

@@ -0,0 +1,63 @@
+import { describe, expect, it } from "bun:test";
+import {
+  executeScriptInContainer,
+  runTerraformApply,
+  runTerraformInit,
+  testRequiredVariables,
+  findResourceInstance,
+  runContainer,
+  TerraformState,
+  execContainer,
+} from "../test";
+
+// executes the coder script after installing pip
+const executeScriptInContainerWithPip = async (
+  state: TerraformState,
+  image: string,
+  shell: string = "sh",
+): Promise<{
+  exitCode: number;
+  stdout: string[];
+  stderr: string[];
+}> => {
+  const instance = findResourceInstance(state, "coder_script");
+  const id = await runContainer(image);
+  const respPip = await execContainer(id, [shell, "-c", "apk add py3-pip"]);
+  const resp = await execContainer(id, [shell, "-c", instance.script]);
+  const stdout = resp.stdout.trim().split("\n");
+  const stderr = resp.stderr.trim().split("\n");
+  return {
+    exitCode: resp.exitCode,
+    stdout,
+    stderr,
+  };
+};
+
+describe("jupyterlab", async () => {
+  await runTerraformInit(import.meta.dir);
+
+  testRequiredVariables(import.meta.dir, {
+    agent_id: "foo",
+  });
+
+  it("fails without pip3", async () => {
+    const state = await runTerraformApply(import.meta.dir, {
+      agent_id: "foo",
+    });
+    const output = await executeScriptInContainer(state, "alpine");
+    expect(output.exitCode).toBe(1);
+    expect(output.stdout).toEqual([
+      "\u001B[0;1mInstalling jupyterlab!",
+      "pip3 is not installed",
+      "Please install pip3 in your Dockerfile/VM image before running this script",
+    ]);
+  });
+
+  // TODO: Add faster test to run with pip3.
+  // currently times out.
+  // it("runs with pip3", async () => {
+  //   ...
+  //   const output = await executeScriptInContainerWithPip(state, "alpine");
+  //   ...
+  // });
+});

personalize/main.test.ts

@@ -0,0 +1,33 @@
+import { readableStreamToText, spawn } from "bun";
+import { describe, expect, it } from "bun:test";
+import {
+  executeScriptInContainer,
+  runTerraformApply,
+  runTerraformInit,
+  testRequiredVariables,
+  runContainer,
+  execContainer,
+  findResourceInstance,
+} from "../test";
+
+describe("personalize", async () => {
+  await runTerraformInit(import.meta.dir);
+
+  testRequiredVariables(import.meta.dir, {
+    agent_id: "foo",
+  });
+
+  it("warns without personalize script", async () => {
+    const state = await runTerraformApply(import.meta.dir, {
+      agent_id: "foo",
+    });
+    const output = await executeScriptInContainer(state, "alpine");
+    expect(output.exitCode).toBe(0);
+    expect(output.stdout).toEqual([
+      "✨ \u001b[0;1mYou don't have a personalize script!",
+      "",
+      "Run \u001b[36;40;1mtouch ~/personalize && chmod +x ~/personalize\u001b[0m to create one.",
+      "It will run every time your workspace starts. Use it to install personal packages!",
+    ]);
+  });
+});