pkg/server/test_env.go - main - githubmirror/grafana

pkg/server/test_env.go
85 строк3 KB
Gonzalo Trigueros Manzanas
Provisioning: wire quota getter so it can be dynamically update during integration tests (#118691)
04 мар 2026, 11:12
04 мар 2026, 11:12fca78e0
package server

import (
	"github.com/stretchr/testify/mock"

	githubconnection "github.com/grafana/grafana/apps/provisioning/pkg/connection/github"
	"github.com/grafana/grafana/apps/provisioning/pkg/quotas"
	"github.com/grafana/grafana/apps/provisioning/pkg/repository/github"
	"github.com/grafana/grafana/apps/secret/pkg/decrypt"
	"github.com/grafana/grafana/pkg/infra/db"
	"github.com/grafana/grafana/pkg/infra/httpclient"
	"github.com/grafana/grafana/pkg/plugins/manager/registry"
	"github.com/grafana/grafana/pkg/services/auth"
	"github.com/grafana/grafana/pkg/services/featuremgmt"
	"github.com/grafana/grafana/pkg/services/grpcserver"
	"github.com/grafana/grafana/pkg/services/notifications"
	"github.com/grafana/grafana/pkg/services/oauthtoken/oauthtokentest"
	"github.com/grafana/grafana/pkg/setting"
	"github.com/grafana/grafana/pkg/storage/unified/resource"
	"github.com/grafana/grafana/pkg/web"
)

func ProvideTestEnv(
	testingT interface {
		mock.TestingT
		Cleanup(func())
	},
	server *Server,
	db db.DB,
	cfg *setting.Cfg,
	ns *notifications.NotificationServiceMock,
	grpcServer grpcserver.Provider,
	pluginRegistry registry.Service,
	httpClientProvider httpclient.Provider,
	oAuthTokenService *oauthtokentest.Service,
	featureMgmt featuremgmt.FeatureToggles,
	resourceClient resource.ResourceClient,
	idService auth.IDService,
	githubRepoFactory *github.Factory,
	githubConnectionFactory githubconnection.GithubFactory,
	decryptService decrypt.DecryptService,
	quotaGetter quotas.QuotaGetter,
) (*TestEnv, error) {
	return &TestEnv{
		TestingT:                testingT,
		Server:                  server,
		SQLStore:                db,
		Cfg:                     cfg,
		NotificationService:     ns,
		GRPCServer:              grpcServer,
		PluginRegistry:          pluginRegistry,
		HTTPClientProvider:      httpClientProvider,
		OAuthTokenService:       oAuthTokenService,
		FeatureToggles:          featureMgmt,
		ResourceClient:          resourceClient,
		IDService:               idService,
		GithubRepoFactory:       githubRepoFactory,
		GithubConnectionFactory: githubConnectionFactory,
		DecryptService:          decryptService,
		QuotaGetter:             quotaGetter,
	}, nil
}

type TestEnv struct {
	TestingT interface {
		mock.TestingT
		Cleanup(func())
	}
	Server                  *Server
	SQLStore                db.DB
	Cfg                     *setting.Cfg
	NotificationService     *notifications.NotificationServiceMock
	GRPCServer              grpcserver.Provider
	PluginRegistry          registry.Service
	HTTPClientProvider      httpclient.Provider
	OAuthTokenService       *oauthtokentest.Service
	RequestMiddleware       web.Middleware
	FeatureToggles          featuremgmt.FeatureToggles
	ResourceClient          resource.ResourceClient
	IDService               auth.IDService
	GithubRepoFactory       *github.Factory
	GithubConnectionFactory githubconnection.GithubFactory
	DecryptService          decrypt.DecryptService
	QuotaGetter             quotas.QuotaGetter
}