From 714fd9efff4e0e35d1a2ae069b683ba86926bbdc Mon Sep 17 00:00:00 2001
From: Yuriy Taraday <yorik.sar@gmail.com>
Date: Sun, 6 May 2018 00:57:24 +0400
Subject: [PATCH] Add test for volume upload and download

---
 libvirt_integration_test.go | 53 +++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/libvirt_integration_test.go b/libvirt_integration_test.go
index 719108b..9e8f27b 100644
--- a/libvirt_integration_test.go
+++ b/libvirt_integration_test.go
@@ -17,6 +17,7 @@
 package libvirt
 
 import (
+	"bytes"
 	"encoding/xml"
 	"net"
 	"sync"
@@ -252,6 +253,58 @@ func TestXMLIntegration(t *testing.T) {
 	}
 }
 
+func TestVolumeUploadDownloadIntegration(t *testing.T) {
+	testdata := []byte("Hello, world!")
+	l := New(testConn(t))
+
+	if err := l.Connect(); err != nil {
+		t.Error(err)
+	}
+	defer l.Disconnect()
+
+	pool, err := l.StoragePool("test")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	var volObj struct {
+		XMLName  xml.Name `xml:"volume"`
+		Name     string   `xml:"name"`
+		Capacity struct {
+			Value uint64 `xml:",chardata"`
+		} `xml:"capacity"`
+		Target struct {
+			Format struct {
+				Type string `xml:"type,attr"`
+			} `xml:"format"`
+		} `xml:"target"`
+	}
+	volObj.Name = "testvol"
+	volObj.Capacity.Value = uint64(len(testdata))
+	volObj.Target.Format.Type = "raw"
+	xmlVol, err := xml.Marshal(volObj)
+	if err != nil {
+		t.Fatal(err)
+	}
+	vol, err := l.StorageVolCreateXML(pool, string(xmlVol), 0)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer l.StorageVolDelete(vol, 0)
+	err = l.StorageVolUpload(vol, bytes.NewBuffer(testdata), 0, 0, 0)
+	if err != nil {
+		t.Fatal(err)
+	}
+	var buf bytes.Buffer
+	err = l.StorageVolDownload(vol, &buf, 0, 0, 0)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if bytes.Compare(testdata, buf.Bytes()) != 0 {
+		t.Fatal("download not what we uploaded")
+	}
+}
+
 // verify we're able to concurrently communicate with libvirtd.
 // see: https://github.com/digitalocean/go-libvirt/pull/52
 func Test_concurrentWrite(t *testing.T) {