add benchmark for writeFiles

Signed-off-by: Matthias Bertschy <[email protected]>
kubescape · Jan 18, 2024 · 97f7eb8 · 97f7eb8
1 parent 69d97dd
commit 97f7eb8
Showing 1 changed file with 15 additions and 32 deletions.
diff --git a/pkg/registry/file/storage_test.go b/pkg/registry/file/storage_test.go
@@ -546,40 +546,23 @@ func TestStorageImpl_Versioner(t *testing.T) {
 	}
 }
 
-func TestStorageImpl_writeFiles(t *testing.T) {
-	type args struct {
-		key     string
-		obj     runtime.Object
-		metaOut runtime.Object
-	}
-	tests := []struct {
-		name string
-		args args
-	}{
-		{
-			name: "test",
-			args: args{
-				key: "/spdx.softwarecomposition.kubescape.io/sbomspdxv2p3s/kubescape/toto",
-				obj: &v1beta1.SBOMSPDXv2p3{
-					ObjectMeta: v1.ObjectMeta{
-						Name: "toto",
-					},
-					Spec: v1beta1.SBOMSPDXv2p3Spec{
-						Metadata: v1beta1.SPDXMeta{
-							Tool: v1beta1.ToolMeta{Name: "titi"},
-						},
-					},
-				},
-				metaOut: &v1beta1.SBOMSPDXv2p3{},
+func BenchmarkWriteFiles(b *testing.B) {
+	s := NewStorageImpl(afero.NewMemMapFs(), DefaultStorageRoot).(*StorageImpl)
+	ctx := context.TODO()
+	key := "/spdx.softwarecomposition.kubescape.io/sbomspdxv2p3s/kubescape/toto"
+	obj := &v1beta1.SBOMSPDXv2p3{
+		ObjectMeta: v1.ObjectMeta{
+			Name: "toto",
+		},
+		Spec: v1beta1.SBOMSPDXv2p3Spec{
+			Metadata: v1beta1.SPDXMeta{
+				Tool: v1beta1.ToolMeta{Name: "titi"},
 			},
 		},
 	}
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			s := NewStorageImpl(afero.NewMemMapFs(), DefaultStorageRoot).(*StorageImpl)
-			err := s.writeFiles(context.TODO(), tt.args.key, tt.args.obj, tt.args.metaOut)
-			assert.NoError(t, err)
-			assert.Equal(t, tt.args.metaOut.(*v1beta1.SBOMSPDXv2p3).Spec, v1beta1.SBOMSPDXv2p3Spec{})
-		})
+	metaOut := &v1beta1.SBOMSPDXv2p3{}
+	for i := 0; i < b.N; i++ {
+		_ = s.writeFiles(ctx, key, obj, metaOut)
 	}
+	b.ReportAllocs()
 }