Function testRoundTripLocalFile

private/buf/buffetch/buffetch_test.go:96–134

(
	t *testing.T,
	filename string,
	expectedData []byte,
	expectedFormat string,
	expectedCompressionType internal.CompressionType,
)

}

func testRoundTripLocalFile(
	t *testing.T,
	filename string,
	expectedData []byte,
	expectedFormat string,
	expectedCompressionType internal.CompressionType,
) {
	logger := slogtestext.NewLogger(t)
	refParser := newRefParser(logger)
	reader := testNewFetchReader(logger)
	writer := testNewFetchWriter(logger)

	ctx := t.Context()
	container := app.NewContainer(nil, nil, nil, nil)

	tmpDir := t.TempDir()
	filePath := filepath.Join(tmpDir, filename)

	parsedRef, err := refParser.getParsedRef(ctx, filePath, allFormats)
	require.NoError(t, err)
	require.Equal(t, expectedFormat, parsedRef.Format())
	fileRef, ok := parsedRef.(internal.FileRef)
	require.True(t, ok)
	require.Equal(t, expectedCompressionType, fileRef.CompressionType())

	writeCloser, err := writer.PutFile(ctx, container, fileRef)
	require.NoError(t, err)
	_, err = writeCloser.Write(expectedData)
	require.NoError(t, err)
	require.NoError(t, writeCloser.Close())

	readCloser, err := reader.GetFile(ctx, container, fileRef)
	require.NoError(t, err)
	actualData, err := io.ReadAll(readCloser)
	require.NoError(t, err)
	require.NoError(t, readCloser.Close())

	require.Equal(t, string(expectedData), string(actualData))
}

func testNewFetchReader(logger *slog.Logger) internal.Reader {
	storageosProvider := storageos.NewProvider(storageos.ProviderWithSymlinks())