Function TestNotifications

enterprise/coderd/schedule/template_test.go:617–898

(t *testing.T)

}

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

	t.Run("Dormancy", func(t *testing.T) {
		t.Parallel()

		var (
			db, _ = dbtestutil.NewDB(t)
			ctx   = testutil.Context(t, testutil.WaitLong)
			user  = dbgen.User(t, db, database.User{})
			file  = dbgen.File(t, db, database.File{
				CreatedBy: user.ID,
			})
			templateJob = dbgen.ProvisionerJob(t, db, nil, database.ProvisionerJob{
				FileID:      file.ID,
				InitiatorID: user.ID,
				Tags: database.StringMap{
					"foo": "bar",
				},
			})
			timeTilDormant  = time.Minute * 2
			templateVersion = dbgen.TemplateVersion(t, db, database.TemplateVersion{
				CreatedBy:      user.ID,
				JobID:          templateJob.ID,
				OrganizationID: templateJob.OrganizationID,
			})
			template = dbgen.Template(t, db, database.Template{
				ActiveVersionID:          templateVersion.ID,
				CreatedBy:                user.ID,
				OrganizationID:           templateJob.OrganizationID,
				TimeTilDormant:           int64(timeTilDormant),
				TimeTilDormantAutoDelete: int64(timeTilDormant),
			})
		)

		// Add two dormant workspaces and one active workspace.
		dormantWorkspaces := []database.WorkspaceTable{
			dbgen.Workspace(t, db, database.WorkspaceTable{
				OwnerID:        user.ID,
				TemplateID:     template.ID,
				OrganizationID: templateJob.OrganizationID,
				LastUsedAt:     time.Now().Add(-time.Hour),
			}),
			dbgen.Workspace(t, db, database.WorkspaceTable{
				OwnerID:        user.ID,
				TemplateID:     template.ID,
				OrganizationID: templateJob.OrganizationID,
				LastUsedAt:     time.Now().Add(-time.Hour),
			}),
		}
		dbgen.Workspace(t, db, database.WorkspaceTable{
			OwnerID:        user.ID,
			TemplateID:     template.ID,
			OrganizationID: templateJob.OrganizationID,
			LastUsedAt:     time.Now(),
		})
		for _, ws := range dormantWorkspaces {
			db.UpdateWorkspaceDormantDeletingAt(ctx, database.UpdateWorkspaceDormantDeletingAtParams{