Method TestEntityPoolOperations

database/sql/pools_test.go:210–289

()

}

func (s *PoolsTestSuite) TestEntityPoolOperations() {
	ep := garmTesting.CreateDefaultGithubEndpoint(s.ctx, s.Store, s.T())
	creds := garmTesting.CreateTestGithubCredentials(s.ctx, "test-creds", s.Store, s.T(), ep)
	s.T().Cleanup(func() { s.Store.DeleteGithubCredentials(s.ctx, creds.ID) })
	repo, err := s.Store.CreateRepository(s.ctx, "test-owner", "test-repo", creds, "test-secret", params.PoolBalancerTypeRoundRobin, false)
	s.Require().NoError(err)
	s.Require().NotEmpty(repo.ID)
	s.T().Cleanup(func() { s.Store.DeleteRepository(s.ctx, repo.ID) })

	entity, err := repo.GetEntity()
	s.Require().NoError(err)

	createPoolParams := params.CreatePoolParams{
		ProviderName: "test-provider",
		MaxRunners:   5,
		Image:        "test-image",
		Flavor:       "test-flavor",
		OSType:       commonParams.Linux,
		OSArch:       commonParams.Amd64,
		Tags:         []string{"test-tag"},
	}

	pool, err := s.Store.CreateEntityPool(s.ctx, entity, createPoolParams)
	s.Require().NoError(err)
	s.Require().NotEmpty(pool.ID)
	s.T().Cleanup(func() { s.Store.DeleteEntityPool(s.ctx, entity, pool.ID) })

	entityPool, err := s.Store.GetEntityPool(s.ctx, entity, pool.ID)
	s.Require().NoError(err)
	s.Require().Equal(pool.ID, entityPool.ID)
	s.Require().Equal(pool.ProviderName, entityPool.ProviderName)

	updatePoolParams := params.UpdatePoolParams{
		Enabled: garmTesting.Ptr(true),
		Flavor:  "new-flavor",
		Image:   "new-image",
		RunnerPrefix: params.RunnerPrefix{
			Prefix: "new-prefix",
		},
		MaxRunners:             garmTesting.Ptr(uint(100)),
		MinIdleRunners:         garmTesting.Ptr(uint(50)),
		OSType:                 commonParams.Windows,
		OSArch:                 commonParams.Amd64,
		Tags:                   []string{"new-tag"},
		RunnerBootstrapTimeout: garmTesting.Ptr(uint(10)),
		ExtraSpecs:             json.RawMessage(`{"extra": "specs"}`),
		GitHubRunnerGroup:      garmTesting.Ptr("new-group"),
		Priority:               garmTesting.Ptr(uint(1)),
	}
	pool, err = s.Store.UpdateEntityPool(s.ctx, entity, pool.ID, updatePoolParams)
	s.Require().NoError(err)
	s.Require().Equal(*updatePoolParams.Enabled, pool.Enabled)
	s.Require().Equal(updatePoolParams.Flavor, pool.Flavor)
	s.Require().Equal(updatePoolParams.Image, pool.Image)
	s.Require().Equal(updatePoolParams.Prefix, pool.Prefix)
	s.Require().Equal(*updatePoolParams.MaxRunners, pool.MaxRunners)
	s.Require().Equal(*updatePoolParams.MinIdleRunners, pool.MinIdleRunners)
	s.Require().Equal(updatePoolParams.OSType, pool.OSType)