Function TestDetectorHungWorkspaceBuild

coderd/jobreaper/detector_test.go:152–202

(t *testing.T)

}

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

	ctx := testutil.Context(t, testutil.WaitLong)
	env := newDetectorTestEnv(ctx, t)
	defer env.close()

	var (
		now                         = time.Now()
		twentyMinAgo                = now.Add(-time.Minute * 20)
		tenMinAgo                   = now.Add(-time.Minute * 10)
		sixMinAgo                   = now.Add(-time.Minute * 6)
		org                         = dbgen.Organization(t, env.DB, database.Organization{})
		user                        = dbgen.User(t, env.DB, database.User{})
		expectedWorkspaceBuildState = []byte(`{"dean":"cool","colin":"also cool"}`)
	)

	// Previous build (completed successfully).
	previousBuild := dbfake.WorkspaceBuild(t, env.DB, database.WorkspaceTable{
		OrganizationID: org.ID,
		OwnerID:        user.ID,
	}).Pubsub(env.Pubsub).Seed(database.WorkspaceBuild{}).
		ProvisionerState(expectedWorkspaceBuildState).
		Succeeded(dbfake.WithJobCompletedAt(twentyMinAgo)).
		Do()

	// Current build (hung - running job with UpdatedAt > 5 min ago).
	currentBuild := dbfake.WorkspaceBuild(t, env.DB, previousBuild.Workspace).
		Pubsub(env.Pubsub).
		Seed(database.WorkspaceBuild{BuildNumber: 2}).
		Starting(dbfake.WithJobStartedAt(tenMinAgo), dbfake.WithJobUpdatedAt(sixMinAgo)).
		Do()

	t.Log("previous job ID: ", previousBuild.Build.JobID)
	t.Log("current job ID: ", currentBuild.Build.JobID)

	stats := env.tick(ctx, now)
	require.NoError(t, stats.Error)
	require.Len(t, stats.TerminatedJobIDs, 1)
	require.Equal(t, currentBuild.Build.JobID, stats.TerminatedJobIDs[0])

	// Check that the current provisioner job was updated.
	requireTerminatedJob(ctx, t, env.DB, currentBuild.Build.JobID, now, jobreaper.Hung)

	// Check that the provisioner state was copied.
	build, err := env.DB.GetWorkspaceBuildByID(ctx, currentBuild.Build.ID)
	require.NoError(t, err)
	provisionerStateRow, err := env.DB.GetWorkspaceBuildProvisionerStateByID(ctx, build.ID)
	require.NoError(t, err)
	require.Equal(t, expectedWorkspaceBuildState, provisionerStateRow.ProvisionerState)
}

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