From edd909006b8d655eb19f34626bd1921951646045 Mon Sep 17 00:00:00 2001
From: Sylvain Baubeau <sbaubeau@redhat.com>
Date: Thu, 15 Jan 2015 15:19:20 +0100
Subject: [PATCH] rbd: add tests for read/write and snapshots

---
 rbd/rbd_test.go | 103 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 103 insertions(+)

diff --git a/rbd/rbd_test.go b/rbd/rbd_test.go
index 2e85a94..d7756f0 100644
--- a/rbd/rbd_test.go
+++ b/rbd/rbd_test.go
@@ -7,6 +7,7 @@ import (
     "github.com/stretchr/testify/assert"
     "os/exec"
     "sort"
+    "encoding/json"
 )
 
 func GetUUID() string {
@@ -58,3 +59,105 @@ func TestGetImageNames(t *testing.T) {
     conn.DeletePool(poolname)
     conn.Shutdown()
 }
+
+func TestIOReaderWriter(t *testing.T) {
+    conn, _ := rados.NewConn()
+    conn.ReadDefaultConfigFile()
+    conn.Connect()
+
+    poolname := GetUUID()
+    err := conn.MakePool(poolname)
+    assert.NoError(t, err)
+
+    ioctx, err := conn.OpenIOContext(poolname)
+    assert.NoError(t, err)
+
+    name := GetUUID()
+    img, err := rbd.Create(ioctx, name, 1<<22)
+    assert.NoError(t, err)
+
+    err = img.Open()
+    assert.NoError(t, err)
+
+    stats, err := img.Stat()
+    assert.NoError(t, err)
+
+    encoder := json.NewEncoder(img)
+    encoder.Encode(stats)
+
+    err = img.Flush()
+    assert.NoError(t, err)
+
+    _, err = img.Seek(0, 0)
+    assert.NoError(t, err)
+
+    var stats2 *rbd.ImageInfo
+    decoder := json.NewDecoder(img)
+    decoder.Decode(&stats2)
+
+    assert.Equal(t, &stats, &stats2)
+
+    _, err = img.Seek(0, 0)
+    bytes_in := []byte("input data")
+    _, err = img.Write(bytes_in)
+    assert.NoError(t, err)
+
+    _, err = img.Seek(0, 0)
+    assert.NoError(t, err)
+
+    bytes_out := make([]byte, len(bytes_in))
+    n_out, err := img.Read(bytes_out)
+
+    assert.Equal(t, n_out, len(bytes_in))
+    assert.Equal(t, bytes_in, bytes_out)
+
+    err = img.Close()
+    assert.NoError(t, err)
+
+    img.Remove()
+    assert.NoError(t, err)
+
+    ioctx.Destroy()
+    conn.DeletePool(poolname)
+    conn.Shutdown()
+}
+
+func TestCreateSnapshot(t *testing.T) {
+    conn, _ := rados.NewConn()
+    conn.ReadDefaultConfigFile()
+    conn.Connect()
+
+    poolname := GetUUID()
+    err := conn.MakePool(poolname)
+    assert.NoError(t, err)
+
+    ioctx, err := conn.OpenIOContext(poolname)
+    assert.NoError(t, err)
+
+    name := GetUUID()
+    img, err := rbd.Create(ioctx, name, 1<<22)
+    assert.NoError(t, err)
+
+    err = img.Open()
+    assert.NoError(t, err)
+
+    snapshot, err := img.CreateSnapshot("mysnap")
+    assert.NoError(t, err)
+
+    err = img.Close()
+    err = img.Open("mysnap")
+    assert.NoError(t, err)
+
+    snapshot.Remove()
+    assert.NoError(t, err)
+
+    err = img.Close()
+    assert.NoError(t, err)
+
+    img.Remove()
+    assert.NoError(t, err)
+
+    ioctx.Destroy()
+    conn.DeletePool(poolname)
+    conn.Shutdown()
+}