Method TestScaleSetOperations

database/sql/scalesets_test.go:141–364

()

}

func (s *ScaleSetsTestSuite) TestScaleSetOperations() {
	// create a scale set for the organization
	createScaleSetPrams := params.CreateScaleSetParams{
		Name:              "test-scaleset",
		ProviderName:      "test-provider",
		MaxRunners:        10,
		MinIdleRunners:    5,
		Image:             "test-image",
		Flavor:            "test-flavor",
		OSType:            commonParams.Linux,
		OSArch:            commonParams.Amd64,
		ExtraSpecs:        json.RawMessage(`{"test": 1}`),
		GitHubRunnerGroup: "test-group",
	}

	var orgScaleSet params.ScaleSet
	var repoScaleSet params.ScaleSet
	var enterpriseScaleSet params.ScaleSet
	var err error

	s.T().Run("create org scaleset", func(_ *testing.T) {
		orgScaleSet, err = s.Store.CreateEntityScaleSet(s.adminCtx, s.orgEntity, createScaleSetPrams)
		s.Require().NoError(err)
		s.Require().NotNil(orgScaleSet)
		s.Require().Equal(orgScaleSet.Name, createScaleSetPrams.Name)
		s.T().Cleanup(func() {
			err := s.Store.DeleteScaleSetByID(s.adminCtx, orgScaleSet.ID)
			if err != nil {
				s.FailNow(fmt.Sprintf("failed to delete scaleset: %s", err))
			}
		})
	})

	s.T().Run("create repo scaleset", func(_ *testing.T) {
		repoScaleSet, err = s.Store.CreateEntityScaleSet(s.adminCtx, s.repoEntity, createScaleSetPrams)
		s.Require().NoError(err)
		s.Require().NotNil(repoScaleSet)
		s.Require().Equal(repoScaleSet.Name, createScaleSetPrams.Name)
		s.T().Cleanup(func() {
			err := s.Store.DeleteScaleSetByID(s.adminCtx, repoScaleSet.ID)
			if err != nil {
				s.FailNow(fmt.Sprintf("failed to delete scaleset: %s", err))
			}
		})
	})

	s.T().Run("create enterprise scaleset", func(_ *testing.T) {
		enterpriseScaleSet, err = s.Store.CreateEntityScaleSet(s.adminCtx, s.enterpriseEntity, createScaleSetPrams)
		s.Require().NoError(err)
		s.Require().NotNil(enterpriseScaleSet)
		s.Require().Equal(enterpriseScaleSet.Name, createScaleSetPrams.Name)

		s.T().Cleanup(func() {
			err := s.Store.DeleteScaleSetByID(s.adminCtx, enterpriseScaleSet.ID)
			if err != nil {
				s.FailNow(fmt.Sprintf("failed to delete scaleset: %s", err))
			}
		})