Add template unit test

Emyrk · Emyrk · commit 837f66a23715 · 2023-02-02T17:02:53.000-06:00
diff --git a/coderd/authzquery/authz_test.go b/coderd/authzquery/authz_test.go
@@ -71,3 +71,10 @@ func TestAuthzQueryRecursive(t *testing.T) {
 		reflect.ValueOf(q).Method(i).Call(ins)
 	}
 }
+
+func must[T any](value T, err error) T {
+	if err != nil {
+		panic(err)
+	}
+	return value
+}
diff --git a/coderd/authzquery/job_test.go b/coderd/authzquery/job_test.go
@@ -0,0 +1,94 @@
+package authzquery_test
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/google/uuid"
+
+	"github.com/coder/coder/coderd/database"
+	"github.com/coder/coder/coderd/database/dbgen"
+	"github.com/coder/coder/coderd/rbac"
+)
+
+func (suite *MethodTestSuite) TestProvsionerJob() {
+	suite.Run("Build/GetProvisionerJobByID", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			w := dbgen.Workspace(t, db, database.Workspace{})
+			j := dbgen.ProvisionerJob(t, db, database.ProvisionerJob{
+				Type: database.ProvisionerJobTypeWorkspaceBuild,
+			})
+			_ = dbgen.WorkspaceBuild(t, db, database.WorkspaceBuild{JobID: j.ID, WorkspaceID: w.ID})
+			return methodCase(inputs(j.ID), asserts(w, rbac.ActionRead))
+		})
+	})
+	suite.Run("TemplateVersion/GetProvisionerJobByID", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			j := dbgen.ProvisionerJob(t, db, database.ProvisionerJob{
+				Type: database.ProvisionerJobTypeTemplateVersionImport,
+			})
+			tpl := dbgen.Template(t, db, database.Template{})
+			v := dbgen.TemplateVersion(t, db, database.TemplateVersion{
+				TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true},
+				JobID:      j.ID,
+			})
+			return methodCase(inputs(j.ID), asserts(v.RBACObject(tpl), rbac.ActionRead))
+		})
+	})
+	suite.Run("TemplateVersionDryRun/GetProvisionerJobByID", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			tpl := dbgen.Template(t, db, database.Template{})
+			v := dbgen.TemplateVersion(t, db, database.TemplateVersion{
+				TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true},
+			})
+			j := dbgen.ProvisionerJob(t, db, database.ProvisionerJob{
+				Type: database.ProvisionerJobTypeTemplateVersionDryRun,
+				Input: must(json.Marshal(struct {
+					TemplateVersionID uuid.UUID `json:"template_version_id"`
+				}{TemplateVersionID: v.ID})),
+			})
+			return methodCase(inputs(j.ID), asserts(v.RBACObject(tpl), rbac.ActionRead))
+		})
+	})
+	suite.Run("Build/UpdateProvisionerJobWithCancelByID", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			tpl := dbgen.Template(t, db, database.Template{AllowUserCancelWorkspaceJobs: true})
+			w := dbgen.Workspace(t, db, database.Workspace{TemplateID: tpl.ID})
+			j := dbgen.ProvisionerJob(t, db, database.ProvisionerJob{
+				Type: database.ProvisionerJobTypeWorkspaceBuild,
+			})
+			_ = dbgen.WorkspaceBuild(t, db, database.WorkspaceBuild{JobID: j.ID, WorkspaceID: w.ID})
+			return methodCase(inputs(database.UpdateProvisionerJobWithCancelByIDParams{ID: j.ID}), asserts(w, rbac.ActionUpdate))
+		})
+	})
+	suite.Run("TemplateVersion/UpdateProvisionerJobWithCancelByID", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			j := dbgen.ProvisionerJob(t, db, database.ProvisionerJob{
+				Type: database.ProvisionerJobTypeTemplateVersionImport,
+			})
+			tpl := dbgen.Template(t, db, database.Template{})
+			v := dbgen.TemplateVersion(t, db, database.TemplateVersion{
+				TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true},
+				JobID:      j.ID,
+			})
+			return methodCase(inputs(database.UpdateProvisionerJobWithCancelByIDParams{ID: j.ID}),
+				asserts(v.RBACObject(tpl), []rbac.Action{rbac.ActionRead, rbac.ActionUpdate}))
+		})
+	})
+	suite.Run("TemplateVersionDryRun/UpdateProvisionerJobWithCancelByID", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			tpl := dbgen.Template(t, db, database.Template{})
+			v := dbgen.TemplateVersion(t, db, database.TemplateVersion{
+				TemplateID: uuid.NullUUID{UUID: tpl.ID, Valid: true},
+			})
+			j := dbgen.ProvisionerJob(t, db, database.ProvisionerJob{
+				Type: database.ProvisionerJobTypeTemplateVersionDryRun,
+				Input: must(json.Marshal(struct {
+					TemplateVersionID uuid.UUID `json:"template_version_id"`
+				}{TemplateVersionID: v.ID})),
+			})
+			return methodCase(inputs(database.UpdateProvisionerJobWithCancelByIDParams{ID: j.ID}),
+				asserts(v.RBACObject(tpl), []rbac.Action{rbac.ActionRead, rbac.ActionUpdate}))
+		})
+	})
+}

Original file line number	Diff line number	Diff line change
`@@ -71,3 +71,10 @@ func TestAuthzQueryRecursive(t *testing.T) {`
`71`	`71`	`reflect.ValueOf(q).Method(i).Call(ins)`
`72`	`72`	`}`
`73`	`73`	`}`
	`74`	`+`
	`75`	`+func must[T any](value T, err error) T {`
	`76`	`+ if err != nil {`
	`77`	`+ panic(err)`
	`78`	`+ }`
	`79`	`+ return value`
	`80`	`+}`