Test

presleyp · presleyp · commit bd3a026ee9df · 2022-05-10T17:29:44.000Z
diff --git a/site/src/components/Workspace/Workspace.test.tsx b/site/src/components/Workspace/Workspace.test.tsx
diff --git a/site/src/components/WorkspaceStatusBar/WorkspaceStatusBar.tsx b/site/src/components/WorkspaceStatusBar/WorkspaceStatusBar.tsx
@@ -73,7 +73,7 @@ export const WorkspaceStatusBar: React.FC<WorkspaceStatusBarProps> = ({
         <div className={styles.horizontal}>
           <div className={styles.horizontal}>
             <Typography variant="h4">{workspace.name}</Typography>
-            <Box className={styles.statusChip}>
+            <Box className={styles.statusChip} role="status">
               {workspaceStatus === "started" && Language.started}
               {workspaceStatus === "starting" && Language.starting}
               {workspaceStatus === "stopped" && Language.stopped}
diff --git a/site/src/pages/WorkspacePage/WorkspacePage.test.tsx b/site/src/pages/WorkspacePage/WorkspacePage.test.tsx
@@ -1,6 +1,8 @@
 import { screen } from "@testing-library/react"
+import { rest } from "msw"
 import React from "react"
-import { MockTemplate, MockWorkspace, renderWithAuth } from "../../testHelpers"
+import { MockFailedWorkspace, MockStoppedWorkspace, MockTemplate, MockWorkspace, renderWithAuth } from "../../testHelpers"
+import { server } from "../../testHelpers/server"
 import { WorkspacePage } from "./WorkspacePage"
 
 describe("Workspace Page", () => {
@@ -11,4 +13,47 @@ describe("Workspace Page", () => {
     expect(workspaceName).toBeDefined()
     expect(templateName).toBeDefined()
   })
+  it("shows the status of the workspace", async () => {
+    renderWithAuth(<WorkspacePage />, { route: `/workspaces/${MockWorkspace.id}`, path: "/workspaces/:workspace" })
+    const status = await screen.findByRole("status")
+    expect(status).toHaveTextContent("Running")
+  })
+  it("stops the workspace when the user presses Stop", async () => {
+    renderWithAuth(<WorkspacePage />, { route: `/workspaces/${MockWorkspace.id}`, path: "/workspaces/:workspace" })
+    const status = await screen.findByText("Running")
+    expect(status).toBeDefined()
+    const stopButton = await screen.findByText("Stop")
+    stopButton.click()
+    const laterStatus = await screen.findByText("Stopping")
+    expect(laterStatus).toBeDefined()
+  })
+  it("starts the workspace when the user presses Start", async () => {
+    server.use(
+      rest.get(`/api/v2/workspaces/${MockWorkspace.id}`, (req, res, ctx) => {
+        return res(ctx.status(200), ctx.json(MockStoppedWorkspace))
+      }),
+    )
+    renderWithAuth(<WorkspacePage />, { route: `/workspaces/${MockWorkspace.id}`, path: "/workspaces/:workspace" })
+    const startButton = await screen.findByText("Start")
+    const status = await screen.findByText("Stopped")
+    expect(status).toBeDefined()
+    startButton.click()
+    const laterStatus = await screen.findByText("Building")
+    expect(laterStatus).toBeDefined()
+  })
+  it("retries starting the workspace when the user presses Retry", async () => {
+    // MockFailedWorkspace.latest_build.transition is start so Retry will attempt to start
+    renderWithAuth(<WorkspacePage />, { route: `/workspaces/${MockWorkspace.id}`, path: "/workspaces/:workspace" })
+    server.use(
+      rest.get(`/api/v2/workspaces/${MockWorkspace.id}`, (req, res, ctx) => {
+        return res(ctx.status(200), ctx.json(MockFailedWorkspace))
+      }),
+    )
+    const status = await screen.findByText("Build Failed")
+    expect(status).toBeDefined()
+    const retryButton = await screen.findByText("Retry")
+    retryButton.click()
+    const laterStatus = await screen.findByText("Building")
+    expect(laterStatus).toBeDefined()
+  })
 })
diff --git a/site/src/testHelpers/entities.ts b/site/src/testHelpers/entities.ts
@@ -93,12 +93,34 @@ export const MockProvisionerJob = {
   worker_id: "test-worker-id",
 }
 
+export const MockFailedProvisionerJob = {
+  id: "test-provisioner-job",
+  created_at: "",
+  started_at: "",
+  completed_at: "",
+  error: "",
+  status: "failed" as ProvisionerJobStatus,
+  worker_id: "test-worker-id",
+}
+
 export const MockWorkspaceBuild = {
   id: "test-workspace-build",
   transition: "start" as WorkspaceBuildTransition,
   job: MockProvisionerJob,
 }
 
+export const MockWorkspaceBuildStop = {
+  id: "test-workspace-build",
+  transition: "stop" as WorkspaceBuildTransition,
+  job: MockProvisionerJob,
+}
+
+export const MockFailedWorkspaceBuild = {
+  id: "test-workspace-build",
+  transition: "start" as WorkspaceBuildTransition,
+  job: MockFailedProvisionerJob,
+}
+
 // These are special cases of MockWorkspaceBuild for more precise testing
 export const MockWorkspaceStart = {
   id: "test-workspace-build-start",
@@ -128,6 +150,34 @@ export const MockWorkspace: Workspace = {
   latest_build: MockWorkspaceBuild,
 }
 
+export const MockStoppedWorkspace: Workspace = {
+  id: "test-workspace",
+  name: "Test-Workspace",
+  created_at: "",
+  updated_at: "",
+  template_id: MockTemplate.id,
+  outdated: false,
+  owner_id: MockUser.id,
+  autostart_schedule: MockWorkspaceAutostartEnabled.schedule,
+  autostop_schedule: MockWorkspaceAutostopEnabled.schedule,
+  latest_build: MockWorkspaceBuildStop,
+}
+
+export const MockFailedWorkspace: Workspace = {
+  id: "test-workspace",
+  name: "Test-Workspace",
+  created_at: "",
+  updated_at: "",
+  template_id: MockTemplate.id,
+  outdated: false,
+  owner_id: MockUser.id,
+  autostart_schedule: MockWorkspaceAutostartEnabled.schedule,
+  autostop_schedule: MockWorkspaceAutostopEnabled.schedule,
+  latest_build: MockFailedWorkspaceBuild,
+}
+
+
+
 export const MockWorkspaceAgent: WorkspaceAgent = {
   id: "test-workspace-agent",
   name: "a-workspace-agent",
