100 lines
2 KiB
Go
100 lines
2 KiB
Go
package artifacttest
|
|
|
|
import (
|
|
"errors"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"gitlab.com/slon/shad-go/distbuild/pkg/artifact"
|
|
"gitlab.com/slon/shad-go/distbuild/pkg/build"
|
|
)
|
|
|
|
type testCache struct {
|
|
*artifact.Cache
|
|
tmpDir string
|
|
}
|
|
|
|
func (c *testCache) cleanup() {
|
|
_ = os.RemoveAll(c.tmpDir)
|
|
}
|
|
|
|
func newTestCache(t *testing.T) *testCache {
|
|
tmpDir, err := ioutil.TempDir("", "")
|
|
require.NoError(t, err)
|
|
|
|
cache, err := artifact.NewCache(tmpDir)
|
|
if err != nil {
|
|
_ = os.RemoveAll(tmpDir)
|
|
}
|
|
require.NoError(t, err)
|
|
|
|
return &testCache{Cache: cache, tmpDir: tmpDir}
|
|
}
|
|
|
|
func TestCache(t *testing.T) {
|
|
c := newTestCache(t)
|
|
defer c.cleanup()
|
|
|
|
idA := build.ID{'a'}
|
|
|
|
path, commit, _, err := c.Create(idA)
|
|
require.NoError(t, err)
|
|
|
|
_, _, _, err = c.Create(idA)
|
|
require.Truef(t, errors.Is(err, artifact.ErrWriteLocked), "%v", err)
|
|
|
|
_, err = os.Create(filepath.Join(path, "a.txt"))
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, commit())
|
|
|
|
path, unlock, err := c.Get(idA)
|
|
require.NoError(t, err)
|
|
defer unlock()
|
|
|
|
_, err = os.Stat(filepath.Join(path, "a.txt"))
|
|
require.NoError(t, err)
|
|
|
|
require.Truef(t, errors.Is(c.Remove(idA), artifact.ErrReadLocked), "%v", err)
|
|
|
|
idB := build.ID{'b'}
|
|
_, _, err = c.Get(idB)
|
|
require.Truef(t, errors.Is(err, artifact.ErrNotFound), "%v", err)
|
|
|
|
require.NoError(t, c.Range(func(artifact build.ID) error {
|
|
require.Equal(t, idA, artifact)
|
|
return nil
|
|
}))
|
|
}
|
|
|
|
func TestAbortWrite(t *testing.T) {
|
|
c := newTestCache(t)
|
|
defer c.cleanup()
|
|
|
|
idA := build.ID{'a'}
|
|
|
|
_, _, abort, err := c.Create(idA)
|
|
require.NoError(t, err)
|
|
require.NoError(t, abort())
|
|
|
|
_, _, err = c.Get(idA)
|
|
require.Truef(t, errors.Is(err, artifact.ErrNotFound), "%v", err)
|
|
}
|
|
|
|
func TestArtifactExists(t *testing.T) {
|
|
c := newTestCache(t)
|
|
defer c.cleanup()
|
|
|
|
idA := build.ID{'a'}
|
|
|
|
_, commit, _, err := c.Create(idA)
|
|
require.NoError(t, err)
|
|
require.NoError(t, commit())
|
|
|
|
_, _, _, err = c.Create(idA)
|
|
require.Truef(t, errors.Is(err, artifact.ErrExists), "%v", err)
|
|
}
|