- Notifications
You must be signed in to change notification settings - Fork1k
feat(coderd): add matched provisioner daemons information to more places#15688
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to ourterms of service andprivacy statement. We’ll occasionally send you account related emails.
Already on GitHub?Sign in to your account
Merged
Uh oh!
There was an error while loading.Please reload this page.
Merged
Changes from1 commit
Commits
Show all changes
16 commits Select commitHold shift + click to select a range
ba944ab feat(coderd/wsbuilder): return provisioners available at time of insert
johnstcn4e51f20 feat(coderd): add matched provisioner daemons information to addition…
johnstcn47036e8 fix linter complaint re nil ptr deref
johnstcn16be03b add test coverage for matched provisioners with workspace build creation
johnstcn4304a06 skip for non-postgres
johnstcn9ef68dd add tests for workspace creation
johnstcn38788d5 revert fe changes in this pr
johnstcn1c95ffe coderd/wsbuilder: use dbauthz.AsSystemReadProvisionerDaemons instead …
johnstcn98521be refactor: extract WarnMatchedProvisioners to cliutil
johnstcne1423f5 fixup! refactor: extract WarnMatchedProvisioners to cliutil
johnstcn517a505 feat(cli): add provisioner warning to create/start/stop commands
johnstcnc4295ef Apply suggestions from code review
johnstcnc5fb83b feat(coderd): add endpoint for matched provisioners of template versi…
johnstcn3bd62fd feat(cli): delete: warn on no matched provisioners
johnstcn2f625bc address linter complaint
johnstcn848338e add test for cli/delete
johnstcnFile filter
Filter by extension
Conversations
Failed to load comments.
Loading
Uh oh!
There was an error while loading.Please reload this page.
Jump to
Jump to file
Failed to load files.
Loading
Uh oh!
There was an error while loading.Please reload this page.
Diff view
Diff view
add test coverage for matched provisioners with workspace build creation
- Loading branch information
Uh oh!
There was an error while loading.Please reload this page.
commit16be03bc621fdc66d56dba67bbf2a8fc2583a0bb
There are no files selected for viewing
110 changes: 110 additions & 0 deletionscoderd/workspacebuilds_test.go
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1097,6 +1097,12 @@ func TestPostWorkspaceBuild(t *testing.T) { | ||
| Transition: codersdk.WorkspaceTransitionStart, | ||
| }) | ||
| require.NoError(t, err) | ||
| if assert.NotNil(t, build.MatchedProvisioners) { | ||
| require.Equal(t, 1, build.MatchedProvisioners.Count) | ||
| require.Equal(t, 1, build.MatchedProvisioners.Available) | ||
| require.NotZero(t, build.MatchedProvisioners.MostRecentlySeen.Time) | ||
| } | ||
| coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, build.ID) | ||
| require.Eventually(t, func() bool { | ||
| @@ -1124,6 +1130,12 @@ func TestPostWorkspaceBuild(t *testing.T) { | ||
| Transition: codersdk.WorkspaceTransitionStart, | ||
| }) | ||
| require.NoError(t, err) | ||
| if assert.NotNil(t, build.MatchedProvisioners) { | ||
| require.Equal(t, 1, build.MatchedProvisioners.Count) | ||
| require.Equal(t, 1, build.MatchedProvisioners.Available) | ||
| require.NotZero(t, build.MatchedProvisioners.MostRecentlySeen.Time) | ||
| } | ||
| require.Equal(t, workspace.LatestBuild.BuildNumber+1, build.BuildNumber) | ||
| }) | ||
| @@ -1150,6 +1162,12 @@ func TestPostWorkspaceBuild(t *testing.T) { | ||
| ProvisionerState: wantState, | ||
| }) | ||
| require.NoError(t, err) | ||
| if assert.NotNil(t, build.MatchedProvisioners) { | ||
| require.Equal(t, 1, build.MatchedProvisioners.Count) | ||
| require.Equal(t, 1, build.MatchedProvisioners.Available) | ||
| require.NotZero(t, build.MatchedProvisioners.MostRecentlySeen.Time) | ||
| } | ||
| gotState, err := client.WorkspaceBuildState(ctx, build.ID) | ||
| require.NoError(t, err) | ||
| require.Equal(t, wantState, gotState) | ||
| @@ -1173,6 +1191,12 @@ func TestPostWorkspaceBuild(t *testing.T) { | ||
| }) | ||
| require.NoError(t, err) | ||
| require.Equal(t, workspace.LatestBuild.BuildNumber+1, build.BuildNumber) | ||
| if assert.NotNil(t, build.MatchedProvisioners) { | ||
| require.Equal(t, 1, build.MatchedProvisioners.Count) | ||
| require.Equal(t, 1, build.MatchedProvisioners.Available) | ||
| require.NotZero(t, build.MatchedProvisioners.MostRecentlySeen.Time) | ||
| } | ||
| coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, build.ID) | ||
| res, err := client.Workspaces(ctx, codersdk.WorkspaceFilter{ | ||
| @@ -1181,6 +1205,92 @@ func TestPostWorkspaceBuild(t *testing.T) { | ||
| require.NoError(t, err) | ||
| require.Len(t, res.Workspaces, 0) | ||
| }) | ||
| t.Run("NoProvisionersAvailable", func(t *testing.T) { | ||
| t.Parallel() | ||
| // Given: a coderd instance with a provisioner daemon | ||
| store, ps, db := dbtestutil.NewDBWithSQLDB(t) | ||
| client, closeDaemon := coderdtest.NewWithProvisionerCloser(t, &coderdtest.Options{ | ||
| Database: store, | ||
| Pubsub: ps, | ||
| IncludeProvisionerDaemon: true, | ||
| }) | ||
| // Given: a user, template, and workspace | ||
| user := coderdtest.CreateFirstUser(t, client) | ||
| version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil) | ||
| coderdtest.AwaitTemplateVersionJobCompleted(t, client, version.ID) | ||
| template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID) | ||
| workspace := coderdtest.CreateWorkspace(t, client, template.ID) | ||
| coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, workspace.LatestBuild.ID) | ||
| // Stop the provisioner daemon. | ||
| require.NoError(t, closeDaemon.Close()) | ||
| ctx := testutil.Context(t, testutil.WaitLong) | ||
| // Given: no provisioner daemons exist. | ||
| _, err := db.ExecContext(ctx, `DELETE FROM provisioner_daemons;`) | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others.Learn more. self-review: this is kinda yucky but I don't agree with the idea of creating database queries for things that will only ever get used intests. | ||
| require.NoError(t, err) | ||
| // When: a new workspace build is created | ||
| build, err := client.CreateWorkspaceBuild(ctx, workspace.ID, codersdk.CreateWorkspaceBuildRequest{ | ||
| TemplateVersionID: template.ActiveVersionID, | ||
| Transition: codersdk.WorkspaceTransitionStart, | ||
| }) | ||
| // Then: the request should succeed. | ||
| require.NoError(t, err) | ||
| // Then: the provisioner job should remain pending. | ||
| require.Equal(t, codersdk.ProvisionerJobPending, build.Job.Status) | ||
| // Then: the response should indicate no provisioners are available. | ||
| if assert.NotNil(t, build.MatchedProvisioners) { | ||
| require.Zero(t, build.MatchedProvisioners.Count) | ||
| require.Zero(t, build.MatchedProvisioners.Available) | ||
| require.Zero(t, build.MatchedProvisioners.MostRecentlySeen.Time) | ||
| } | ||
| }) | ||
| t.Run("AllProvisionersStale", func(t *testing.T) { | ||
| t.Parallel() | ||
| // Given: a coderd instance with a provisioner daemon | ||
| store, ps, db := dbtestutil.NewDBWithSQLDB(t) | ||
| client, closeDaemon := coderdtest.NewWithProvisionerCloser(t, &coderdtest.Options{ | ||
| Database: store, | ||
| Pubsub: ps, | ||
| IncludeProvisionerDaemon: true, | ||
| }) | ||
| // Given: a user, template, and workspace | ||
| user := coderdtest.CreateFirstUser(t, client) | ||
| version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil) | ||
| coderdtest.AwaitTemplateVersionJobCompleted(t, client, version.ID) | ||
| template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID) | ||
| workspace := coderdtest.CreateWorkspace(t, client, template.ID) | ||
| coderdtest.AwaitWorkspaceBuildJobCompleted(t, client, workspace.LatestBuild.ID) | ||
| ctx := testutil.Context(t, testutil.WaitLong) | ||
| // Given: all provisioner daemons are stale | ||
| // First stop the provisioner | ||
| require.NoError(t, closeDaemon.Close()) | ||
| newLastSeenAt := dbtime.Now().Add(-time.Hour) | ||
| // Update the last seen at for all provisioner daemons. We have to use the | ||
| // SQL db directly because store.UpdateProvisionerDaemonLastSeenAt has a | ||
| // built-in check to prevent updating the last seen at to a time in the past. | ||
| _, err := db.ExecContext(ctx, `UPDATE provisioner_daemons SET last_seen_at = $1;`, newLastSeenAt) | ||
| require.NoError(t, err) | ||
| // When: a new workspace build is created | ||
| build, err := client.CreateWorkspaceBuild(ctx, workspace.ID, codersdk.CreateWorkspaceBuildRequest{ | ||
| TemplateVersionID: template.ActiveVersionID, | ||
| Transition: codersdk.WorkspaceTransitionStart, | ||
| }) | ||
| // Then: the request should succeed | ||
| require.NoError(t, err) | ||
| // Then: the provisioner job should remain pending | ||
| require.Equal(t, codersdk.ProvisionerJobPending, build.Job.Status) | ||
| // Then: the response should indicate no provisioners are available | ||
| if assert.NotNil(t, build.MatchedProvisioners) { | ||
| require.Zero(t, build.MatchedProvisioners.Available) | ||
| require.Equal(t, 1, build.MatchedProvisioners.Count) | ||
| require.Equal(t, newLastSeenAt.UTC(), build.MatchedProvisioners.MostRecentlySeen.Time.UTC()) | ||
| } | ||
| }) | ||
| } | ||
| func TestWorkspaceBuildTimings(t *testing.T) { | ||
Oops, something went wrong.
Uh oh!
There was an error while loading.Please reload this page.
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.