Function Test_ValidateImageCommandExtraData

cmd/validate/image_test.go:667–768

(t *testing.T)

}

func Test_ValidateImageCommandExtraData(t *testing.T) {
	validateImageCmd := validateImageCmd(happyValidator())
	cmd := setUpCobra(validateImageCmd)

	fs := afero.NewMemMapFs()

	ctx := utils.WithFS(context.Background(), fs)
	client := fake.FakeClient{}
	commonMockClient(&client)

	// Add missing ResolveDigest expectation for VSA processing
	digest, _ := name.NewDigest(testImageDigest)
	client.On("ResolveDigest", mock.Anything).Return(digest.String(), nil)

	ctx = oci.WithClient(ctx, &client)

	mdl := MockDownloader{}
	mdl.On("Download", mock.Anything, "registry/policy:latest", false).Return(&ociMetadata.OCIMetadata{Digest: "sha256:da54bca5477bf4e3449bc37de1822888fa0fbb8d89c640218cb31b987374d357"}, nil)
	mdl.On("Download", mock.Anything, "registry/policy-data:latest", false).Return(&ociMetadata.OCIMetadata{Digest: "sha256:da54bca5477bf4e3449bc37de1822888fa0fbb8d89c640218cb31b987374d357"}, nil)
	ctx = context.WithValue(ctx, source.DownloaderFuncKey, &mdl)

	cmd.SetContext(ctx)

	testPolicyJSON := `sources:
  - policy:
      - "registry/policy:latest"
    data:
      - "registry/policy-data:latest"
    ruleData:
      custom_rule_data:
        prefix_data:
          - registry1
    config:
      include:
        - '@minimal'
      exclude: []
`
	err := afero.WriteFile(fs, "/policy.json", []byte(testPolicyJSON), 0o644)
	if err != nil {
		panic(err)
	}

	testExtraRuleDataYAML := `---
kind: ReleasePlanAdmission
spec:
  application: [some-app]
  data:
    mapping:
      components:
        - name: some-name
          repository: quay.io/some-namespace/msd
`

	err = afero.WriteFile(fs, "/value.yaml", []byte(testExtraRuleDataYAML), 0o644)
	if err != nil {
		panic(err)
	}