add a test

sreya · sreya · commit b4bf24f0ce28 · 2023-11-02T02:53:34.000Z
diff --git a/cli/start_test.go b/cli/start_test.go
@@ -219,3 +219,77 @@ func TestStartWithParameters(t *testing.T) {
 		})
 	})
 }
+
+// TestStartAutoUpdate also tests restart since the flows are virtually identical.
+func TestStartAutoUpdate(t *testing.T) {
+	t.Parallel()
+
+	const (
+		stringParameterName  = "myparam"
+		stringParameterValue = "abc"
+	)
+
+	stringRichParameters := []*proto.RichParameter{
+		{Name: stringParameterName, Type: "string", Mutable: true, Required: true},
+	}
+
+	type testcase struct {
+		Name string
+		Cmd  string
+	}
+
+	cases := []testcase{
+		{
+			Name: "StartOK",
+			Cmd:  "start",
+		},
+		{
+			Name: "RestartOK",
+			Cmd:  "restart",
+		},
+	}
+
+	for _, c := range cases {
+		c := c
+		t.Run(c.Name, func(t *testing.T) {
+			t.Parallel()
+
+			client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerDaemon: true})
+			owner := coderdtest.CreateFirstUser(t, client)
+			member, _ := coderdtest.CreateAnotherUser(t, client, owner.OrganizationID)
+			version1 := coderdtest.CreateTemplateVersion(t, client, owner.OrganizationID, nil)
+			coderdtest.AwaitTemplateVersionJobCompleted(t, client, version1.ID)
+			template := coderdtest.CreateTemplate(t, client, owner.OrganizationID, version1.ID)
+			workspace := coderdtest.CreateWorkspace(t, member, owner.OrganizationID, template.ID, func(cwr *codersdk.CreateWorkspaceRequest) {
+				cwr.AutomaticUpdates = codersdk.AutomaticUpdatesAlways
+			})
+			coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, workspace.LatestBuild.ID)
+
+			if c.Cmd == "start" {
+				coderdtest.MustTransitionWorkspace(t, member, workspace.ID, database.WorkspaceTransitionStart, database.WorkspaceTransitionStop)
+			}
+			version2 := coderdtest.CreateTemplateVersion(t, client, owner.OrganizationID, prepareEchoResponses(stringRichParameters), func(ctvr *codersdk.CreateTemplateVersionRequest) {
+				ctvr.TemplateID = template.ID
+			})
+			coderdtest.AwaitTemplateVersionJobCompleted(t, client, version2.ID)
+			coderdtest.UpdateActiveTemplateVersion(t, client, template.ID, version2.ID)
+
+			inv, root := clitest.New(t, c.Cmd, "-y", workspace.Name)
+			clitest.SetupConfig(t, member, root)
+			doneChan := make(chan struct{})
+			pty := ptytest.New(t).Attach(inv)
+			go func() {
+				defer close(doneChan)
+				err := inv.Run()
+				assert.NoError(t, err)
+			}()
+
+			pty.ExpectMatch(stringParameterName)
+			pty.WriteLine(stringParameterValue)
+			<-doneChan
+
+			workspace = coderdtest.MustWorkspace(t, member, workspace.ID)
+			require.Equal(t, version2.ID, workspace.LatestBuild.TemplateVersionID)
+		})
+	}
+}
