snapshots: separate implementations into packages

Signed-off-by: Stephen J Day <stephen.day@docker.com>

snapshot/btrfs/btrfs_test.go

@@ -0,0 +1,209 @@
+package btrfs
+
+import (
+	"bytes"
+	"io/ioutil"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/docker/containerd"
+	"github.com/docker/containerd/snapshot/testutil"
+	btrfs "github.com/stevvooe/go-btrfs"
+)
+
+const (
+	mib = 1024 * 1024
+)
+
+func TestBtrfs(t *testing.T) {
+	device := setupBtrfsLoopbackDevice(t)
+	defer removeBtrfsLoopbackDevice(t, device)
+	root, err := ioutil.TempDir(device.mountPoint, "TestBtrfsPrepare-")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(root)
+
+	sm, err := NewBtrfs(device.deviceName, root)
+	if err != nil {
+		t.Fatal(err)
+	}
+	mounts, err := sm.Prepare(filepath.Join(root, "test"), "")
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(mounts)
+
+	for _, mount := range mounts {
+		if mount.Type != "btrfs" {
+			t.Fatalf("wrong mount type: %v != btrfs", mount.Type)
+		}
+
+		// assumes the first, maybe incorrect in the future
+		if !strings.HasPrefix(mount.Options[0], "subvolid=") {
+			t.Fatalf("no subvolid option in %v", mount.Options)
+		}
+	}
+
+	if err := os.MkdirAll(mounts[0].Target, 0755); err != nil {
+		t.Fatal(err)
+	}
+	if err := containerd.MountAll(mounts...); err != nil {
+		t.Fatal(err)
+	}
+	defer testutil.UnmountAll(t, mounts)
+
+	// write in some data
+	if err := ioutil.WriteFile(filepath.Join(mounts[0].Target, "foo"), []byte("content"), 0777); err != nil {
+		t.Fatal(err)
+	}
+
+	// TODO(stevvooe): We don't really make this with the driver, but that
+	// might prove annoying in practice.
+	if err := os.MkdirAll(filepath.Join(root, "snapshots"), 0755); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := sm.Commit(filepath.Join(root, "snapshots/committed"), filepath.Join(root, "test")); err != nil {
+		t.Fatal(err)
+	}
+	defer func() {
+		t.Log("Delete snapshot 1")
+		err := btrfs.SubvolDelete(filepath.Join(root, "snapshots/committed"))
+		if err != nil {
+			t.Error("snapshot delete failed", err)
+		}
+	}()
+
+	mounts, err = sm.Prepare(filepath.Join(root, "test2"), filepath.Join(root, "snapshots/committed"))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if err := os.MkdirAll(filepath.Join(root, "test2"), 0755); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := containerd.MountAll(mounts...); err != nil {
+		t.Fatal(err)
+	}
+	defer testutil.UnmountAll(t, mounts)
+
+	// TODO(stevvooe): Verify contents of "foo"
+	if err := ioutil.WriteFile(filepath.Join(mounts[0].Target, "bar"), []byte("content"), 0777); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := sm.Commit(filepath.Join(root, "snapshots/committed2"), filepath.Join(root, "test2")); err != nil {
+		t.Fatal(err)
+	}
+	defer func() {
+		t.Log("Delete snapshot 2")
+		err := btrfs.SubvolDelete(filepath.Join(root, "snapshots/committed2"))
+		if err != nil {
+			t.Error("snapshot delete failed", err)
+		}
+	}()
+}
+
+type testDevice struct {
+	mountPoint string
+	fileName   string
+	deviceName string
+}
+
+// setupBtrfsLoopbackDevice creates a file, mounts it as a loopback device, and
+// formats it as btrfs.  The device should be cleaned up by calling
+// removeBtrfsLoopbackDevice.
+func setupBtrfsLoopbackDevice(t *testing.T) *testDevice {
+	// create temporary directory for mount point
+	mountPoint, err := ioutil.TempDir("", "containerd-btrfs-test")
+	if err != nil {
+		t.Fatal("Could not create mount point for btrfs test", err)
+	}
+	t.Log("Temporary mount point created", mountPoint)
+
+	// create temporary file for the disk image
+	file, err := ioutil.TempFile("", "containerd-btrfs-test")
+	if err != nil {
+		t.Fatal("Could not create temporary file for btrfs test", err)
+	}
+	t.Log("Temporary file created", file.Name())
+
+	// initialize file with 100 MiB
+	zero := [mib]byte{}
+	for i := 0; i < 100; i++ {
+		_, err = file.Write(zero[:])
+		if err != nil {
+			t.Fatal("Could not write to btrfs file", err)
+		}
+	}
+	file.Close()
+
+	// create device
+	losetup := exec.Command("losetup", "--find", "--show", file.Name())
+	var stdout, stderr bytes.Buffer
+	losetup.Stdout = &stdout
+	losetup.Stderr = &stderr
+	err = losetup.Run()
+	if err != nil {
+		t.Log(stderr.String())
+		t.Fatal("Could not run losetup", err)
+	}
+	deviceName := strings.TrimSpace(stdout.String())
+	t.Log("Created loop device", deviceName)
+
+	// format
+	t.Log("Creating btrfs filesystem")
+	mkfs := exec.Command("mkfs.btrfs", deviceName)
+	err = mkfs.Run()
+	if err != nil {
+		t.Fatal("Could not run mkfs.btrfs", err)
+	}
+
+	// mount
+	t.Logf("Mounting %s at %s", deviceName, mountPoint)
+	mount := exec.Command("mount", deviceName, mountPoint)
+	err = mount.Run()
+	if err != nil {
+		t.Fatal("Could not mount", err)
+	}
+
+	return &testDevice{
+		mountPoint: mountPoint,
+		fileName:   file.Name(),
+		deviceName: deviceName,
+	}
+}
+
+// removeBtrfsLoopbackDevice unmounts the loopback device and deletes the
+// file holding the disk image.
+func removeBtrfsLoopbackDevice(t *testing.T, device *testDevice) {
+	// unmount
+	testutil.Unmount(t, device.mountPoint)
+
+	// detach device
+	t.Log("Removing loop device")
+	losetup := exec.Command("losetup", "--detach", device.deviceName)
+	err := losetup.Run()
+	if err != nil {
+		t.Error("Could not remove loop device", device.deviceName, err)
+	}
+
+	// remove file
+	t.Log("Removing temporary file")
+	err = os.Remove(device.fileName)
+	if err != nil {
+		t.Error(err)
+	}
+
+	// remove mount point
+	t.Log("Removing temporary mount point")
+	err = os.RemoveAll(device.mountPoint)
+	if err != nil {
+		t.Error(err)
+	}
+}