Function TestFetchSources

cmd/inspect/inspect_policy_test.go:91–135

(t *testing.T)

}

func TestFetchSources(t *testing.T) {
	fs := afero.NewMemMapFs()
	ctx := utils.WithFS(context.Background(), fs)

	downloader := mockDownloader{}
	ctx = context.WithValue(ctx, source.DownloaderFuncKey, &downloader)

	createDir := func(args mock.Arguments) {
		dir := args.String(0)

		if err := fs.MkdirAll(dir, 0o755); err != nil {
			panic(err)
		}
		if err := afero.WriteFile(fs, fmt.Sprintf("%s/foo.rego", args.String(0)), []byte("package foo\n\nbar = 1"), 0o644); err != nil {
			panic(err)
		}
	}

	downloader.On("Download", mock.Anything, "one", false).Return(&fileMetadata.FSMetadata{}, nil).Run(createDir)
	downloader.On("Download", mock.Anything, "two", false).Return(&fileMetadata.FSMetadata{}, nil).Run(createDir)
	downloader.On("Download", mock.Anything, "three", false).Return(&fileMetadata.FSMetadata{}, nil).Run(createDir)

	inspectPolicyCmd := inspectPolicyCmd()
	cmd := setUpCobra(inspectPolicyCmd)
	cmd.SetContext(ctx)
	buffy := bytes.Buffer{}
	cmd.SetOut(&buffy)

	cmd.SetArgs([]string{
		"inspect",
		"policy",
		"--source",
		"one",
		"--source",
		"two",
		"--source",
		"three",
	})

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

	assert.Equal(t, "[one,two,three]", inspectPolicyCmd.Flag("source").Value.String())
	assert.Equal(t, "# Source: file::one\n\n# Source: file::three\n\n# Source: file::two\n\n", buffy.String())
}

func TestSourcesAndPolicyCantBeBothProvided(t *testing.T) {
	fs := afero.NewMemMapFs()