()
| 4238 | } |
| 4239 | |
| 4240 | func (s *MethodTestSuite) TestExtraMethods() { |
| 4241 | s.Run("GetProvisionerDaemons", s.Subtest(func(db database.Store, check *expects) { |
| 4242 | dbtestutil.DisableForeignKeysAndTriggers(s.T(), db) |
| 4243 | d, err := db.UpsertProvisionerDaemon(context.Background(), database.UpsertProvisionerDaemonParams{ |
| 4244 | Provisioners: []database.ProvisionerType{}, |
| 4245 | Tags: database.StringMap(map[string]string{ |
| 4246 | provisionersdk.TagScope: provisionersdk.ScopeOrganization, |
| 4247 | }), |
| 4248 | }) |
| 4249 | s.NoError(err, "insert provisioner daemon") |
| 4250 | check.Args().Asserts(d, policy.ActionRead) |
| 4251 | })) |
| 4252 | s.Run("GetProvisionerDaemonsByOrganization", s.Subtest(func(db database.Store, check *expects) { |
| 4253 | dbtestutil.DisableForeignKeysAndTriggers(s.T(), db) |
| 4254 | org := dbgen.Organization(s.T(), db, database.Organization{}) |
| 4255 | d, err := db.UpsertProvisionerDaemon(context.Background(), database.UpsertProvisionerDaemonParams{ |
| 4256 | OrganizationID: org.ID, |
| 4257 | Provisioners: []database.ProvisionerType{}, |
| 4258 | Tags: database.StringMap(map[string]string{ |
| 4259 | provisionersdk.TagScope: provisionersdk.ScopeOrganization, |
| 4260 | }), |
| 4261 | }) |
| 4262 | s.NoError(err, "insert provisioner daemon") |
| 4263 | ds, err := db.GetProvisionerDaemonsByOrganization(context.Background(), database.GetProvisionerDaemonsByOrganizationParams{OrganizationID: org.ID}) |
| 4264 | s.NoError(err, "get provisioner daemon by org") |
| 4265 | check.Args(database.GetProvisionerDaemonsByOrganizationParams{OrganizationID: org.ID}).Asserts(d, policy.ActionRead).Returns(ds) |
| 4266 | })) |
| 4267 | s.Run("GetProvisionerDaemonsWithStatusByOrganization", s.Subtest(func(db database.Store, check *expects) { |
| 4268 | org := dbgen.Organization(s.T(), db, database.Organization{}) |
| 4269 | d := dbgen.ProvisionerDaemon(s.T(), db, database.ProvisionerDaemon{ |
| 4270 | OrganizationID: org.ID, |
| 4271 | Tags: map[string]string{ |
| 4272 | provisionersdk.TagScope: provisionersdk.ScopeOrganization, |
| 4273 | }, |
| 4274 | }) |
| 4275 | ds, err := db.GetProvisionerDaemonsWithStatusByOrganization(context.Background(), database.GetProvisionerDaemonsWithStatusByOrganizationParams{ |
| 4276 | OrganizationID: org.ID, |
| 4277 | StaleIntervalMS: 24 * time.Hour.Milliseconds(), |
| 4278 | }) |
| 4279 | s.NoError(err, "get provisioner daemon with status by org") |
| 4280 | check.Args(database.GetProvisionerDaemonsWithStatusByOrganizationParams{ |
| 4281 | OrganizationID: org.ID, |
| 4282 | StaleIntervalMS: 24 * time.Hour.Milliseconds(), |
| 4283 | }).Asserts(d, policy.ActionRead).Returns(ds) |
| 4284 | })) |
| 4285 | s.Run("GetEligibleProvisionerDaemonsByProvisionerJobIDs", s.Subtest(func(db database.Store, check *expects) { |
| 4286 | dbtestutil.DisableForeignKeysAndTriggers(s.T(), db) |
| 4287 | org := dbgen.Organization(s.T(), db, database.Organization{}) |
| 4288 | tags := database.StringMap(map[string]string{ |
| 4289 | provisionersdk.TagScope: provisionersdk.ScopeOrganization, |
| 4290 | }) |
| 4291 | j, err := db.InsertProvisionerJob(context.Background(), database.InsertProvisionerJobParams{ |
| 4292 | OrganizationID: org.ID, |
| 4293 | Type: database.ProvisionerJobTypeWorkspaceBuild, |
| 4294 | Tags: tags, |
| 4295 | Provisioner: database.ProvisionerTypeEcho, |
| 4296 | StorageMethod: database.ProvisionerStorageMethodFile, |
| 4297 | Input: json.RawMessage("{}"), |
nothing calls this directly
no test coverage detected