IntenseWebs/grafana
3
0
Fork 0
mirror of https://github.com/grafana/grafana.git synced 2025-02-25 18:55:37 -06:00
Code Issues Packages Projects Releases Wiki Activity

Chore: Replace mock guardian with the actual one in folder API benchmarks (#76231)

Browse Source 
Benchmarks: Replace mock guardian with the actual one
This commit is contained in:
Sofia Papagiannaki 2023-10-10 13:31:55 +03:00 committed by GitHub
parent 90631360eb
commit 99e4894636
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
pkg/api/folder_bench_test.go
View File

@ -182,13 +182,6 @@ func setupDB(b testing.TB) benchScenario {
userSvc, err := userimpl.ProvideService(db, orgService, cfg, teamSvc, cache, &quotatest.FakeQuotaService{}, bundleregistry.ProvideService()) userSvc, err := userimpl.ProvideService(db, orgService, cfg, teamSvc, cache, &quotatest.FakeQuotaService{}, bundleregistry.ProvideService())
require.NoError(b, err) require.NoError(b, err)
origNewGuardian := guardian.New
guardian.MockDashboardGuardian(&guardian.FakeDashboardGuardian{CanSaveValue: true, CanViewValue: true})
b.Cleanup(func() {
guardian.New = origNewGuardian
})
var orgID int64 = 1 var orgID int64 = 1
userIDs := make([]int64, 0, TEAM_MEMBER_NUM) userIDs := make([]int64, 0, TEAM_MEMBER_NUM)
@ -293,6 +286,8 @@ func setupDB(b testing.TB) benchScenario {
Created: now, Created: now,
}, },
) )
signedInUser.Permissions[orgID][dashboards.ActionFoldersRead] = append(signedInUser.Permissions[orgID][dashboards.ActionFoldersRead], dashboards.ScopeFoldersProvider.GetResourceScopeUID(f0.UID))
signedInUser.Permissions[orgID][dashboards.ActionDashboardsRead] = append(signedInUser.Permissions[orgID][dashboards.ActionDashboardsRead], dashboards.ScopeFoldersProvider.GetResourceScopeUID(f0.UID))
for j := 0; j < LEVEL0_DASHBOARD_NUM; j++ { for j := 0; j < LEVEL0_DASHBOARD_NUM; j++ {
str := fmt.Sprintf("dashboard_%d_%d", i, j) str := fmt.Sprintf("dashboard_%d_%d", i, j)
@ -440,16 +435,21 @@ func setupServer(b testing.TB, sc benchScenario, features *featuremgmt.FeatureMa
starSvc := startest.NewStarServiceFake() starSvc := startest.NewStarServiceFake()
starSvc.ExpectedUserStars = &star.GetUserStarsResult{UserStars: make(map[int64]bool)} starSvc.ExpectedUserStars = &star.GetUserStarsResult{UserStars: make(map[int64]bool)}
hs := &HTTPServer{ hs := &HTTPServer{
CacheService: localcache.New(5*time.Minute, 10*time.Minute), CacheService: localcache.New(5*time.Minute, 10*time.Minute),
Cfg: sc.cfg, Cfg: sc.cfg,
SQLStore: sc.db, SQLStore: sc.db,
Features: features, Features: features,
QuotaService: quotaSrv, QuotaService: quotaSrv,
SearchService: search.ProvideService(sc.cfg, sc.db, starSvc, dashboardSvc), SearchService: search.ProvideService(sc.cfg, sc.db, starSvc, dashboardSvc),
folderService: folderServiceWithFlagOn, folderService: folderServiceWithFlagOn,
DashboardService: dashboardSvc,
} }
hs.AccessControl = acimpl.ProvideAccessControl(hs.Cfg)
guardian.InitAccessControlGuardian(hs.Cfg, hs.AccessControl, hs.DashboardService)
m.Get("/api/folders", hs.GetFolders) m.Get("/api/folders", hs.GetFolders)
m.Get("/api/search", hs.Search) m.Get("/api/search", hs.Search)