2013-02-18 23:25:43 +00:00
|
|
|
package fs
|
|
|
|
|
|
|
|
import (
|
2013-03-12 18:59:27 +00:00
|
|
|
"github.com/dotcloud/docker/fake"
|
2013-02-18 23:25:43 +00:00
|
|
|
"io/ioutil"
|
|
|
|
"os"
|
2013-03-11 12:42:36 +00:00
|
|
|
"testing"
|
2013-02-18 23:25:43 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestLayersInit(t *testing.T) {
|
|
|
|
store := tempStore(t)
|
|
|
|
defer os.RemoveAll(store.Root)
|
|
|
|
// Root should exist
|
|
|
|
if _, err := os.Stat(store.Root); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
// List() should be empty
|
|
|
|
if l := store.List(); len(l) != 0 {
|
|
|
|
t.Fatalf("List() should return %d, not %d", 0, len(l))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestAddLayer(t *testing.T) {
|
|
|
|
store := tempStore(t)
|
|
|
|
defer os.RemoveAll(store.Root)
|
2013-03-11 12:42:36 +00:00
|
|
|
layer, err := store.AddLayer("foo", testArchive(t))
|
2013-02-18 23:25:43 +00:00
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
// Layer path should exist
|
|
|
|
if _, err := os.Stat(layer); err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
// List() should return 1 layer
|
|
|
|
if l := store.List(); len(l) != 1 {
|
|
|
|
t.Fatalf("List() should return %d elements, not %d", 1, len(l))
|
|
|
|
}
|
|
|
|
// Get("foo") should return the correct layer
|
|
|
|
if foo := store.Get("foo"); foo != layer {
|
|
|
|
t.Fatalf("get(\"foo\") should return '%d', not '%d'", layer, foo)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestAddLayerDuplicate(t *testing.T) {
|
|
|
|
store := tempStore(t)
|
|
|
|
defer os.RemoveAll(store.Root)
|
2013-03-11 12:42:36 +00:00
|
|
|
if _, err := store.AddLayer("foobar123", testArchive(t)); err != nil {
|
2013-02-18 23:25:43 +00:00
|
|
|
t.Fatal(err)
|
|
|
|
}
|
2013-03-11 12:42:36 +00:00
|
|
|
if _, err := store.AddLayer("foobar123", testArchive(t)); err == nil {
|
2013-02-18 23:25:43 +00:00
|
|
|
t.Fatalf("Creating duplicate layer should fail")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* HELPER FUNCTIONS
|
|
|
|
*/
|
|
|
|
|
|
|
|
func tempStore(t *testing.T) *LayerStore {
|
|
|
|
tmp, err := ioutil.TempDir("", "docker-fs-layerstore-")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
store, err := NewLayerStore(tmp)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
return store
|
|
|
|
}
|
|
|
|
|
|
|
|
func testArchive(t *testing.T) Archive {
|
|
|
|
archive, err := fake.FakeTar()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
return archive
|
|
|
|
}
|