fix tests

SasSwart · SasSwart · commit fcd7e2eb9cb0 · 2024-12-09T14:12:14.000Z
diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go
@@ -2121,16 +2121,25 @@ func (s *MethodTestSuite) TestExtraMethods() {
 	}))
 	s.Run("GetProvisionerDaemonsByProvisionerJobs", s.Subtest(func(db database.Store, check *expects) {
 		org := dbgen.Organization(s.T(), db, database.Organization{})
+		tags := database.StringMap(map[string]string{
+			provisionersdk.TagScope: provisionersdk.ScopeOrganization,
+		})
+		j, err := db.InsertProvisionerJob(context.Background(), database.InsertProvisionerJobParams{
+			OrganizationID: org.ID,
+			Type:           database.ProvisionerJobTypeWorkspaceBuild,
+			Tags:           tags,
+			Provisioner:    database.ProvisionerTypeEcho,
+			StorageMethod:  database.ProvisionerStorageMethodFile,
+		})
+		s.NoError(err, "insert provisioner job")
 		d, err := db.UpsertProvisionerDaemon(context.Background(), database.UpsertProvisionerDaemonParams{
 			OrganizationID: org.ID,
-			Tags: database.StringMap(map[string]string{
-				provisionersdk.TagScope: provisionersdk.ScopeOrganization,
-			}),
+			Tags:           tags,
 		})
 		s.NoError(err, "insert provisioner daemon")
-		ds, err := db.GetProvisionerDaemonsByProvisionerJobs(context.Background(), []uuid.UUID{d.ID})
+		ds, err := db.GetProvisionerDaemonsByProvisionerJobs(context.Background(), []uuid.UUID{j.ID})
 		s.NoError(err, "get provisioner daemon by org")
-		check.Args(uuid.UUIDs{org.ID}).Asserts(d, policy.ActionRead).Returns(ds)
+		check.Args(uuid.UUIDs{j.ID}).Asserts(d, policy.ActionRead).Returns(ds)
 	}))
 	s.Run("DeleteOldProvisionerDaemons", s.Subtest(func(db database.Store, check *expects) {
 		_, err := db.UpsertProvisionerDaemon(context.Background(), database.UpsertProvisionerDaemonParams{
