1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
package testing
import (
"fmt"
"io"
"io/ioutil"
"testing"
"github.com/gophercloud/gophercloud/openstack/imageservice/v2/imagedata"
th "github.com/gophercloud/gophercloud/testhelper"
fakeclient "github.com/gophercloud/gophercloud/testhelper/client"
)
func TestUpload(t *testing.T) {
th.SetupHTTP()
defer th.TeardownHTTP()
HandlePutImageDataSuccessfully(t)
err := imagedata.Upload(
fakeclient.ServiceClient(),
"da3b75d9-3f4a-40e7-8a2c-bfab23927dea",
readSeekerOfBytes([]byte{5, 3, 7, 24})).ExtractErr()
th.AssertNoErr(t, err)
}
func TestStage(t *testing.T) {
th.SetupHTTP()
defer th.TeardownHTTP()
HandleStageImageDataSuccessfully(t)
err := imagedata.Stage(
fakeclient.ServiceClient(),
"da3b75d9-3f4a-40e7-8a2c-bfab23927dea",
readSeekerOfBytes([]byte{5, 3, 7, 24})).ExtractErr()
th.AssertNoErr(t, err)
}
func readSeekerOfBytes(bs []byte) io.ReadSeeker {
return &RS{bs: bs}
}
// implements io.ReadSeeker
type RS struct {
bs []byte
offset int
}
func (rs *RS) Read(p []byte) (int, error) {
leftToRead := len(rs.bs) - rs.offset
if 0 < leftToRead {
bytesToWrite := min(leftToRead, len(p))
for i := 0; i < bytesToWrite; i++ {
p[i] = rs.bs[rs.offset]
rs.offset++
}
return bytesToWrite, nil
}
return 0, io.EOF
}
func min(a int, b int) int {
if a < b {
return a
}
return b
}
func (rs *RS) Seek(offset int64, whence int) (int64, error) {
var offsetInt = int(offset)
if whence == 0 {
rs.offset = offsetInt
} else if whence == 1 {
rs.offset = rs.offset + offsetInt
} else if whence == 2 {
rs.offset = len(rs.bs) - offsetInt
} else {
return 0, fmt.Errorf("For parameter `whence`, expected value in {0,1,2} but got: %#v", whence)
}
return int64(rs.offset), nil
}
func TestDownload(t *testing.T) {
th.SetupHTTP()
defer th.TeardownHTTP()
HandleGetImageDataSuccessfully(t)
rdr, err := imagedata.Download(fakeclient.ServiceClient(), "da3b75d9-3f4a-40e7-8a2c-bfab23927dea").Extract()
th.AssertNoErr(t, err)
bs, err := ioutil.ReadAll(rdr)
th.AssertNoErr(t, err)
th.AssertByteArrayEquals(t, []byte{34, 87, 0, 23, 23, 23, 56, 255, 254, 0}, bs)
}
|
