Method SetupTest

runner/pool/pool_test.go:59–128

()

}

func (s *PoolStressTestSuite) SetupTest() {
	dbCfg := garmTesting.GetTestSqliteDBConfig(s.T())
	db, err := database.NewDatabase(context.Background(), dbCfg)
	s.Require().NoError(err)

	s.store = db
	s.adminCtx = garmTesting.ImpersonateAdminContext(context.Background(), db, s.T())

	endpoint := garmTesting.CreateDefaultGithubEndpoint(s.adminCtx, db, s.T())
	creds := garmTesting.CreateTestGithubCredentials(s.adminCtx, "stress-creds", db, s.T(), endpoint)

	repo, err := db.CreateRepository(
		s.adminCtx,
		"test-owner",
		"test-repo",
		creds,
		"test-webhook-secret",
		params.PoolBalancerTypeRoundRobin,
		false,
	)
	s.Require().NoError(err)

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

	s.providerMock = runnerCommonMocks.NewProvider(s.T())
	s.ghcliMock = runnerCommonMocks.NewGithubClient(s.T())

	pool, err := db.CreateEntityPool(s.adminCtx, entity, params.CreatePoolParams{
		ProviderName:   "test-provider",
		MaxRunners:     10,
		MinIdleRunners: 0,
		Image:          "test-image",
		Flavor:         "test-flavor",
		OSType:         "linux",
		OSArch:         "amd64",
		Tags:           []string{"self-hosted", "linux", "x64"},
		Enabled:        true,
	})
	s.Require().NoError(err)
	s.pool = pool

	// Populate the entity cache so ensureMinIdleRunners can find pools.
	cache.SetEntity(entity)
	cache.SetEntityPool(entity.ID, pool)

	s.controllerInfo, err = db.InitController()
	s.Require().NoError(err)

	backoff, err := locking.NewInstanceDeleteBackoff(context.Background())
	s.Require().NoError(err)

	s.mgr = &basePoolManager{
		ctx:              s.adminCtx,
		consumerID:       "test-consumer",
		entity:           entity,
		store:            db,