Function Test_ValidateImageCommandJSONPolicyFile

cmd/validate/image_test.go:617–665

(t *testing.T)

}

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

	client := fake.FakeClient{}
	commonMockClient(&client)
	fs := afero.NewMemMapFs()
	ctx := utils.WithFS(context.Background(), fs)
	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"
    config:
      include:
        - '@minimal'
      exclude: []
`
	err := afero.WriteFile(fs, "/policy.json", []byte(testPolicyJSON), 0o644)
	if err != nil {
		panic(err)
	}
	args := append(rootArgs, []string{
		"--image",
		"registry/image:tag",
		"--public-key",
		utils.TestPublicKey,
		"--policy",
		"/policy.json",
	}...)
	cmd.SetArgs(args)

	var out bytes.Buffer
	cmd.SetOut(&out)

	utils.SetTestRekorPublicKey(t)

	err = cmd.Execute()
	assert.NoError(t, err)
}

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