mirror of
https://github.com/grafana/grafana.git
synced 2024-11-26 19:00:54 -06:00
9d2f5ef62f
* RBAC: Move metadata to own file * RBAC: Rename test files * RBAC: Add wildcard structure and helper function to generate wildcards from prefix * RBAC: Refactor filter to use WildcardsFromPrefix * RBAC: Refactor GetResourceMetadata to use WildcardsFromPrefix
62 lines
2.1 KiB
Go
62 lines
2.1 KiB
Go
package accesscontrol
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func setupTestEnv(b *testing.B, resourceCount, permissionPerResource int) (map[string][]string, map[string]bool) {
|
|
res := map[string][]string{}
|
|
ids := make(map[string]bool, resourceCount)
|
|
|
|
for p := 0; p < permissionPerResource; p++ {
|
|
action := fmt.Sprintf("resources:action%v", p)
|
|
for r := 0; r < resourceCount; r++ {
|
|
scope := fmt.Sprintf("resources:id:%v", r)
|
|
res[action] = append(res[action], scope)
|
|
ids[fmt.Sprintf("%d", r)] = true
|
|
}
|
|
}
|
|
|
|
return res, ids
|
|
}
|
|
|
|
func benchGetMetadata(b *testing.B, resourceCount, permissionPerResource int) {
|
|
permissions, ids := setupTestEnv(b, resourceCount, permissionPerResource)
|
|
b.ResetTimer()
|
|
|
|
var metadata map[string]Metadata
|
|
for n := 0; n < b.N; n++ {
|
|
metadata = GetResourcesMetadata(context.Background(), permissions, "resources:id:", ids)
|
|
assert.Len(b, metadata, resourceCount)
|
|
for _, resourceMetadata := range metadata {
|
|
assert.Len(b, resourceMetadata, permissionPerResource)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Lots of permissions
|
|
func BenchmarkGetResourcesMetadata_10_1000(b *testing.B) { benchGetMetadata(b, 10, 1000) } // ~0.0022s/op
|
|
func BenchmarkGetResourcesMetadata_10_10000(b *testing.B) { benchGetMetadata(b, 10, 10000) } // ~0.019s/op
|
|
func BenchmarkGetResourcesMetadata_10_100000(b *testing.B) { benchGetMetadata(b, 10, 100000) } // ~0.25s/op
|
|
func BenchmarkGetResourcesMetadata_10_1000000(b *testing.B) {
|
|
if testing.Short() {
|
|
b.Skip("Skipping benchmark in short mode")
|
|
}
|
|
benchGetMetadata(b, 10, 1000000)
|
|
} // ~5.8s/op
|
|
|
|
// Lots of resources
|
|
func BenchmarkGetResourcesMetadata_1000_10(b *testing.B) { benchGetMetadata(b, 1000, 10) } // ~0,0023s/op
|
|
func BenchmarkGetResourcesMetadata_10000_10(b *testing.B) { benchGetMetadata(b, 10000, 10) } // ~0.022s/op
|
|
func BenchmarkGetResourcesMetadata_100000_10(b *testing.B) { benchGetMetadata(b, 100000, 10) } // ~0.26s/op
|
|
func BenchmarkGetResourcesMetadata_1000000_10(b *testing.B) {
|
|
if testing.Short() {
|
|
b.Skip("Skipping benchmark in short mode")
|
|
}
|
|
benchGetMetadata(b, 1000000, 10)
|
|
} // ~4.1s/op
|