Thanks to visit codestin.com
Credit goes to github.com

Skip to content

feat: enforce template-level constraints for TTL and autostart #2018

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.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 20 commits into from
Jun 7, 2022
Merged

feat: enforce template-level constraints for TTL and autostart #2018

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
1713431
RED: add failing unit tests
johnstcn Jun 2, 2022
f11f942
RED: add migrations and make gen
johnstcn Jun 2, 2022
c5c9a7a
ORANGE: fix audit/diff tests
johnstcn Jun 2, 2022
74f4dee
fixup! RED: add migrations and make gen
johnstcn Jun 2, 2022
7da27a0
fixup! RED: add migrations and make gen
johnstcn Jun 2, 2022
5ad048b
feat: schedule: add schedule.Min function to determine minimum interval
johnstcn Jun 2, 2022
65b980e
ORANGE: coderd: fix a whole bunch of tests
johnstcn Jun 2, 2022
aa298a6
RED: fast-fail some slow-failing tests
johnstcn Jun 2, 2022
a5ec9bf
GREEN: fix even more tests
johnstcn Jun 3, 2022
a6255d6
GREEN: fix final unit test
johnstcn Jun 3, 2022
245795e
cli: templatecreate: add CLI flags --max-ttl --min-autostart-interval
johnstcn Jun 3, 2022
709db5b
fixup! cli: templatecreate: add CLI flags --max-ttl --min-autostart-i…
johnstcn Jun 3, 2022
c4a0927
make gen
johnstcn Jun 3, 2022
244e479
update test entities
johnstcn Jun 3, 2022
60ef479
Merge remote-tracking branch 'origin/main' into 1433-template-constra…
johnstcn Jun 7, 2022
c9aab8b
fixup! Merge remote-tracking branch 'origin/main' into 1433-template-…
johnstcn Jun 7, 2022
f997af2
coderd: default newly-minted workspaces to template max_ttl
johnstcn Jun 7, 2022
d7f697b
address PR comment
johnstcn Jun 7, 2022
ebe4c15
fix broken unit test
johnstcn Jun 7, 2022
08ef2eb
fix some more tests
johnstcn Jun 7, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
coderd: default newly-minted workspaces to template max_ttl
  • Loading branch information
@johnstcn
johnstcn committed Jun 7, 2022
commit f997af26194ede8424a93ea6781930d6db8590f4
5 changes: 5 additions & 0 deletions coderd/workspaces.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -287,6 +287,11 @@ func (api *API) postWorkspacesByOrganization(rw http.ResponseWriter, r *http.Req
return
}

if !dbTTL.Valid {
// Default to template maximum when creating a new workspace
dbTTL = sql.NullInt64{Valid: true, Int64: template.MaxTtl}
}

workspace, err := api.Database.GetWorkspaceByOwnerIDAndName(r.Context(), database.GetWorkspaceByOwnerIDAndNameParams{
OwnerID: apiKey.UserID,
Name: createWorkspace.Name,
Expand Down
21 changes: 18 additions & 3 deletions coderd/workspaces_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -165,6 +165,24 @@ func TestPostWorkspacesByOrganization(t *testing.T) {
_ = coderdtest.CreateWorkspace(t, client, user.OrganizationID, template.ID)
})

t.Run("TemplateCustomTTL", func(t *testing.T) {
t.Parallel()
client := coderdtest.New(t, &coderdtest.Options{IncludeProvisionerD: true})
user := coderdtest.CreateFirstUser(t, client)
version := coderdtest.CreateTemplateVersion(t, client, user.OrganizationID, nil)
templateTTL := 24 * time.Hour.Milliseconds()
template := coderdtest.CreateTemplate(t, client, user.OrganizationID, version.ID, func(ctr *codersdk.CreateTemplateRequest) {
ctr.MaxTTLMillis = ptr.Ref(templateTTL)
})
coderdtest.AwaitTemplateVersionJob(t, client, version.ID)
workspace := coderdtest.CreateWorkspace(t, client, user.OrganizationID, template.ID, func(cwr *codersdk.CreateWorkspaceRequest) {
cwr.TTLMillis = nil // ensure that no default TTL is set
})
// TTL should be set by the template
require.Equal(t, template.MaxTTLMillis, templateTTL)
require.Equal(t, template.MaxTTLMillis, template.MaxTTLMillis, workspace.TTLMillis)
})

t.Run("InvalidTTL", func(t *testing.T) {
t.Parallel()
t.Run("BelowMin", func(t *testing.T) {
Expand Down Expand Up @@ -653,9 +671,6 @@ func TestWorkspaceUpdateTTL(t *testing.T) {
})
)

// ensure test invariant: new workspaces have no autostop schedule.
require.Nil(t, workspace.TTLMillis, "expected newly-minted workspace to have no TTL")

err := client.UpdateWorkspaceTTL(ctx, workspace.ID, codersdk.UpdateWorkspaceTTLRequest{
TTLMillis: testCase.ttlMillis,
})
Expand Down