42 lines
882 B
Go
42 lines
882 B
Go
|
package filecache
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"io/ioutil"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"gitlab.com/slon/shad-go/distbuild/pkg/build"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestFileCache(t *testing.T) {
|
||
|
|
tmpDir, err := ioutil.TempDir("", "filecache")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
cache, err := New(tmpDir)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
_, abort, err := cache.Write(build.ID{01})
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NoError(t, abort())
|
||
|
|
|
||
|
|
_, _, err = cache.Get(build.ID{01})
|
||
|
|
require.Truef(t, errors.Is(err, ErrNotFound), "real error: %v", err)
|
||
|
|
|
||
|
|
f, _, err := cache.Write(build.ID{02})
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
_, err = f.Write([]byte("foo bar"))
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NoError(t, f.Close())
|
||
|
|
|
||
|
|
path, unlock, err := cache.Get(build.ID{02})
|
||
|
|
require.NoError(t, err)
|
||
|
|
defer unlock()
|
||
|
|
|
||
|
|
content, err := ioutil.ReadFile(path)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, []byte("foo bar"), content)
|
||
|
|
}
|