Function TestConcurrentVerifyMaterial

pkg/policies/concurrency_test.go:83–131

TestConcurrentVerifyMaterial runs VerifyMaterial from multiple goroutines.

(t *testing.T)

// TestConcurrentVerifyMaterial runs VerifyMaterial from multiple goroutines.
func TestConcurrentVerifyMaterial(t *testing.T) {
	logger := zerolog.Nop()

	schema := &v12.CraftingSchema{
		Policies: &v12.Policies{
			Materials: []*v12.PolicyAttachment{
				{
					Policy: &v12.PolicyAttachment_Ref{Ref: "file://testdata/sbom_syft.yaml"},
				},
			},
		},
	}

	material := &v1.Attestation_Material{
		Id:           "sbom",
		MaterialType: v12.CraftingSchema_Material_SBOM_SPDX_JSON,
		M: &v1.Attestation_Material_Artifact_{
			Artifact: &v1.Attestation_Material_Artifact{},
		},
	}

	pv := NewPolicyVerifier(schema.GetPolicies(), nil, &logger)

	const goroutines = 10
	var wg sync.WaitGroup
	errs := make([]error, goroutines)
	results := make([][]*v1.PolicyEvaluation, goroutines)

	for i := range goroutines {
		wg.Add(1)
		go func() {
			defer wg.Done()
			res, err := pv.VerifyMaterial(context.Background(), material, "testdata/sbom-spdx.json")
			errs[i] = err
			results[i] = res
		}()
	}

	wg.Wait()

	for i := range goroutines {
		assert.NoError(t, errs[i], "goroutine %d failed", i)
	}

	for i := 1; i < goroutines; i++ {
		assert.Equal(t, len(results[0]), len(results[i]),
			"goroutine %d returned different number of evaluations", i)
	}
}

// TestWithMaxConcurrency verifies the WithMaxConcurrency option is applied.
func TestWithMaxConcurrency(t *testing.T) {