Storage Benchmark

package local

import (
	"context"
	"fmt"
	"testing"

	rbacv1 "k8s.io/api/rbac/v1"
)

const (
	// n is the number of Roles to add per
	n = 1000

	nRules = 100
)

func BenchmarkStorages_AddData(b *testing.B) {
	role := rbacv1.Role{}
	role.Rules = make([]rbacv1.PolicyRule, nRules)
	for r := 0; r < nRules; r++ {
		role.Rules[r] = rbacv1.PolicyRule{
			Verbs:         []string{rbacv1.VerbAll},
			APIGroups:     []string{rbacv1.APIGroupAll},
			Resources:     []string{fmt.Sprintf("foo%d", r)},
			ResourceNames: []string{fmt.Sprintf("name%d", r)},
		}
	}

	b.ResetTimer()

	for i := 0; i < b.N; i += n {
		b.StopTimer()
		d, err := New()
		if err != nil {
			b.Fatal(err)
		}
		ctx := context.Background()
		b.StartTimer()

		for j := 0; j < n; j++ {
			err := d.AddData(ctx, "foo", []string{"rule", fmt.Sprintf("role-%d", j)}, role)
			if err != nil {
				b.Fatal(err)
			}
		}
	}
}