clipper/backend/filestore/fs_test.go

166 lines
3.7 KiB
Go
Raw Normal View History

package filestore_test
import (
"context"
"io/ioutil"
"os"
"path"
"strings"
"testing"
"git.netflux.io/rob/clipper/filestore"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestFileStoreGetObject(t *testing.T) {
store, err := filestore.NewFileSystemStore("testdata/", "/")
require.NoError(t, err)
reader, err := store.GetObject(context.Background(), "file.txt")
require.NoError(t, err)
defer reader.Close()
content, err := ioutil.ReadAll(reader)
require.NoError(t, err)
assert.Equal(t, "hello world", string(content))
}
func TestFileStoreGetObjectWithRange(t *testing.T) {
testCases := []struct {
name string
start, end int64
wantErr string
wantContent string
}{
{
name: "happy path, complete object",
start: 0,
end: 12,
wantContent: "hello world",
},
{
name: "happy path, partial object",
start: 6,
end: 10,
wantContent: "worl",
},
{
name: "empty range",
start: 1,
end: 1,
wantContent: "",
},
{
name: "bad range",
start: -10,
end: 0,
wantErr: "error seeking in file: seek testdata/file.txt: invalid argument",
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
store, err := filestore.NewFileSystemStore("testdata/", "/")
require.NoError(t, err)
reader, err := store.GetObjectWithRange(context.Background(), "file.txt", tc.start, tc.end)
if tc.wantErr == "" {
defer reader.Close()
require.NoError(t, err)
content, readErr := ioutil.ReadAll(reader)
require.NoError(t, readErr)
assert.Equal(t, tc.wantContent, string(content))
} else {
assert.EqualError(t, err, tc.wantErr)
}
})
}
}
func TestFileStoreGetURL(t *testing.T) {
testCases := []struct {
name string
baseURL string
key string
wantURL string
}{
{
name: "URL with host and no path, no trailing slash",
baseURL: "https://foo.example.com",
key: "bar",
wantURL: "https://foo.example.com/bar",
},
{
name: "URL with host and no path, trailing slash",
baseURL: "https://foo.example.com/",
key: "bar",
wantURL: "https://foo.example.com/bar",
},
{
name: "URL with host and path, no trailing slash",
baseURL: "https://foo.example.com/bar",
key: "baz",
wantURL: "https://foo.example.com/bar/baz",
},
{
name: "URL with host and path, trailing slash",
baseURL: "https://foo.example.com/bar/",
key: "baz",
wantURL: "https://foo.example.com/bar/baz",
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
store, err := filestore.NewFileSystemStore("testdata/", tc.baseURL)
require.NoError(t, err)
url, err := store.GetURL(context.Background(), tc.key)
require.NoError(t, err)
assert.Equal(t, tc.wantURL, url)
})
}
}
func TestFileStorePutObject(t *testing.T) {
rootPath := t.TempDir()
testCases := []struct {
name string
key string
content string
wantCount int64
wantErr string
}{
{
name: "happy path, no created directory",
key: "foo.txt",
content: "hello world",
wantCount: 11,
},
{
name: "happy path, with sub-directories",
key: "foo/bar/baz.txt",
content: "hello world",
wantCount: 11,
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
store, err := filestore.NewFileSystemStore(rootPath, "/")
require.NoError(t, err)
n, err := store.PutObject(context.Background(), tc.key, strings.NewReader(tc.content), "text/plain")
require.NoError(t, err)
content, err := os.ReadFile(path.Join(rootPath, tc.key))
require.NoError(t, err)
assert.Equal(t, tc.wantCount, n)
assert.Equal(t, tc.content, string(content))
})
}
}