-
Notifications
You must be signed in to change notification settings - Fork 2
/
snapshots.go
102 lines (85 loc) · 2.83 KB
/
snapshots.go
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
102
package v2
import (
"fmt"
"strings"
"testing"
"github.com/cloud-barista/nhncloud-sdk-go"
"github.com/cloud-barista/nhncloud-sdk-go/acceptance/tools"
"github.com/cloud-barista/nhncloud-sdk-go/openstack/sharedfilesystems/v2/snapshots"
)
// CreateSnapshot will create a snapshot from the share ID with a name. An error will
// be returned if the snapshot could not be created
func CreateSnapshot(t *testing.T, client *gophercloud.ServiceClient, shareID string) (*snapshots.Snapshot, error) {
if testing.Short() {
t.Skip("Skipping test that requres share creation in short mode.")
}
createOpts := snapshots.CreateOpts{
ShareID: shareID,
Name: "My Test Snapshot",
Description: "My Test Description",
}
snapshot, err := snapshots.Create(client, createOpts).Extract()
if err != nil {
t.Logf("Failed to create snapshot")
return nil, err
}
err = waitForSnapshotStatus(t, client, snapshot.ID, "available")
if err != nil {
t.Logf("Failed to get %s snapshot status", snapshot.ID)
return snapshot, err
}
return snapshot, nil
}
// ListSnapshots lists all snapshots that belong to this tenant's project.
// An error will be returned if the snapshots could not be listed..
func ListSnapshots(t *testing.T, client *gophercloud.ServiceClient) ([]snapshots.Snapshot, error) {
r, err := snapshots.ListDetail(client, &snapshots.ListOpts{}).AllPages()
if err != nil {
return nil, err
}
return snapshots.ExtractSnapshots(r)
}
// DeleteSnapshot will delete a snapshot. A fatal error will occur if the snapshot
// failed to be deleted. This works best when used as a deferred function.
func DeleteSnapshot(t *testing.T, client *gophercloud.ServiceClient, snapshot *snapshots.Snapshot) {
err := snapshots.Delete(client, snapshot.ID).ExtractErr()
if err != nil {
t.Errorf("Unable to delete snapshot %s: %v", snapshot.ID, err)
}
err = waitForSnapshotStatus(t, client, snapshot.ID, "deleted")
if err != nil {
t.Errorf("Failed to wait for 'deleted' status for %s snapshot: %v", snapshot.ID, err)
} else {
t.Logf("Deleted snapshot: %s", snapshot.ID)
}
}
func waitForSnapshotStatus(t *testing.T, c *gophercloud.ServiceClient, id, status string) error {
err := tools.WaitFor(func() (bool, error) {
current, err := snapshots.Get(c, id).Extract()
if err != nil {
if _, ok := err.(gophercloud.ErrDefault404); ok {
switch status {
case "deleted":
return true, nil
default:
return false, err
}
}
return false, err
}
if current.Status == status {
return true, nil
}
if strings.Contains(current.Status, "error") {
return true, fmt.Errorf("An error occurred, wrong status: %s", current.Status)
}
return false, nil
})
if err != nil {
mErr := PrintMessages(t, c, id)
if mErr != nil {
return fmt.Errorf("Snapshot status is '%s' and unable to get manila messages: %s", err, mErr)
}
}
return err
}