Function TestGetPolicyThroughCache

internal/policy/source/source_test.go:539–600

(t *testing.T)

}

func TestGetPolicyThroughCache(t *testing.T) {
	test := func(t *testing.T, fs afero.Fs, expectedDownloads int) {
		t.Cleanup(func() {
			downloadCache = sync.Map{}
		})

		ctx := utils.WithFS(context.Background(), fs)

		invocations := 0
		data := []byte("hello")
		dl := func(source, dest string) (metadata.Metadata, error) {
			invocations++
			if err := fs.MkdirAll(dest, 0o755); err != nil {
				return nil, err
			}

			return nil, afero.WriteFile(fs, filepath.Join(dest, "data.json"), data, 0o400)
		}

		source := &mockPolicySource{&mock.Mock{}}
		source.On("PolicyUrl").Return("policy-url")
		source.On("Subdir").Return("subdir")

		s1, _, err := getPolicyThroughCache(ctx, source, "/workdir1", dl)
		require.NoError(t, err)

		s2, _, err := getPolicyThroughCache(ctx, source, "/workdir2", dl)
		require.NoError(t, err)

		assert.NotEqual(t, s1, s2)
		assert.Equalf(t, expectedDownloads, invocations, "expected %d invocations, but was %d", expectedDownloads, invocations) // was using cache on second invocation

		dataFile1 := filepath.Join(s1, "data.json")
		data1, err := afero.ReadFile(fs, dataFile1)
		require.NoError(t, err)
		assert.Equal(t, data, data1)

		dataFile2 := filepath.Join(s2, "data.json")
		data2, err := afero.ReadFile(fs, dataFile2)
		require.NoError(t, err)
		assert.Equal(t, data, data2)

		if fs, ok := fs.(afero.Symlinker); ok {
			info, ok, err := fs.LstatIfPossible(s2)
			require.True(t, ok)
			require.NoError(t, err)
			assert.True(t, info.Mode()&os.ModeSymlink == os.ModeSymlink)
		}
	}

	t.Run("symlinkable", func(t *testing.T) {
		temp := t.TempDir()
		// need to use the OsFs as it implements Symlinker
		fs := afero.NewBasePathFs(afero.NewOsFs(), temp)

		test(t, fs, 1)
	})