Function TestWorkspaceAppUpsertRestart

coderd/workspaces_test.go:5526–5608

(t *testing.T)

}

func TestWorkspaceAppUpsertRestart(t *testing.T) {
	t.Parallel()

	client := coderdtest.New(t, &coderdtest.Options{
		IncludeProvisionerDaemon: true,
	})
	user := coderdtest.CreateFirstUser(t, client)

	// Define an app to be created with the workspace
	apps := []*proto.App{
		{
			Id:          uuid.NewString(),
			Slug:        "test-app",
			DisplayName: "Test App",
			Command:     "test-command",
			Url:         "http://localhost:8080",
			Icon:        "/test.svg",
		},
	}

	// Create template version with workspace app
	version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, &echo.Responses{
		Parse: echo.ParseComplete,
		ProvisionGraph: []*proto.Response{{
			Type: &proto.Response_Graph{
				Graph: &proto.GraphComplete{
					Resources: []*proto.Resource{{
						Name: "test-resource",
						Type: "example",
						Agents: []*proto.Agent{{
							Id:   uuid.NewString(),
							Name: "dev",
							Auth: &proto.Agent_Token{},
							Apps: apps,
						}},
					}},
				},
			},
		}},
	})
	coderdtest.AwaitTemplateVersionJobCompleted(t, client, version.ID)

	// Create template and workspace
	template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID)
	workspace := coderdtest.CreateWorkspace(t, client, template.ID)
	coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, workspace.LatestBuild.ID)

	ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
	defer cancel()

	// Verify initial workspace has the app
	workspace, err := client.Workspace(ctx, workspace.ID)
	require.NoError(t, err)
	require.Len(t, workspace.LatestBuild.Resources[0].Agents, 1)
	agent := workspace.LatestBuild.Resources[0].Agents[0]
	require.Len(t, agent.Apps, 1)
	require.Equal(t, "test-app", agent.Apps[0].Slug)
	require.Equal(t, "Test App", agent.Apps[0].DisplayName)