test: Fix test timeouts due to contexts created too early (#7242)

This commit is contained in:
Mathias Fredriksson
2023-04-21 17:32:25 +03:00
committed by GitHub
parent ea78ca5dff
commit 6f06f8dadb
2 changed files with 7 additions and 3 deletions

View File

@ -572,8 +572,7 @@ func TestWorkspaceBuildState(t *testing.T) {
func TestWorkspaceBuildStatus(t *testing.T) {
t.Parallel()
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
defer cancel()
auditor := audit.NewMock()
numLogs := len(auditor.AuditLogs())
client, closeDaemon, api := coderdtest.NewWithAPI(t, &coderdtest.Options{IncludeProvisionerDaemon: true, Auditor: auditor})
@ -597,6 +596,10 @@ func TestWorkspaceBuildStatus(t *testing.T) {
closeDaemon = coderdtest.NewProvisionerDaemon(t, api)
// after successful build is "running"
_ = coderdtest.AwaitWorkspaceBuildJob(t, client, workspace.LatestBuild.ID)
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
defer cancel()
workspace, err := client.Workspace(ctx, workspace.ID)
require.NoError(t, err)
require.EqualValues(t, codersdk.WorkspaceStatusRunning, workspace.LatestBuild.Status)