add tests for volume snapshot endpoints

Author: TheRealSibasishBehera

volume_snapshot_test.go

@@ -0,0 +1,98 @@
+package civogo
+
+import (
+	"reflect"
+	"testing"
+)
+
+func TestListVolumeSnapshots(t *testing.T) {
+	client, server, _ := NewClientForTesting(map[string]string{
+		"/v2/snapshots?region=TEST&resource_type=volume": `[{
+			"name": "test-snapshot",
+			"snapshot_id": "12345",
+			"snapshot_description": "snapshot for test",
+			"volume_id": "12345",
+			"restore_size": 20,
+			"state": "available",
+			"creation_time": "2020-01-01T00:00:00Z"
+		}]`,
+	})
+	defer server.Close()
+
+	got, err := client.ListVolumeSnapshots()
+
+	if err != nil {
+		t.Errorf("Request returned an error: %s", err)
+		return
+	}
+
+	expected := []VolumeSnapshot{
+		{
+			Name:                "test-snapshot",
+			SnapshotID:          "12345",
+			SnapshotDescription: "snapshot for test",
+			VolumeID:            "12345",
+			RestoreSize:         20,
+			State:               "available",
+			CreationTime:        "2020-01-01T00:00:00Z",
+		},
+	}
+
+	if !reflect.DeepEqual(got, expected) {
+		t.Errorf("Expected %+v, got %+v", expected, got)
+	}
+}
+
+func TestGetVolumeSnapshot(t *testing.T) {
+	client, server, _ := NewClientForTesting(map[string]string{
+		"/v2/snapshots/snapshot-uuid?region=TEST&resource_type=volume": `{
+			"name": "test-snapshot",
+			"snapshot_id": "snapshot-uuid",
+			"snapshot_description": "snapshot for testing",
+			"volume_id": "12345",
+			"restore_size": 20,
+			"state": "available",
+			"creation_time": "2020-01-01T00:00:00Z"
+		}`,
+	})
+	defer server.Close()
+	got, err := client.GetVolumeSnapshot("snapshot-uuid")
+
+	if err != nil {
+		t.Errorf("Request returned an error: %s", err)
+		return
+	}
+
+	expected := VolumeSnapshot{
+		Name:                "test-snapshot",
+		SnapshotID:          "snapshot-uuid",
+		SnapshotDescription: "snapshot for testing",
+		VolumeID:            "12345",
+		RestoreSize:         20,
+		State:               "available",
+		CreationTime:        "2020-01-01T00:00:00Z",
+	}
+
+	if !reflect.DeepEqual(got, expected) {
+		t.Errorf("Expected %+v, got %+v", expected, got)
+	}
+}
+
+func TestDeleteVolumeSnapshot(t *testing.T) {
+	client, server, _ := NewClientForTesting(map[string]string{
+		"/v2/snapshots/12346": `{"result": "success"}`,
+	})
+	defer server.Close()
+	got, err := client.DeleteVolumeSnapshot("12346")
+
+	if err != nil {
+		t.Errorf("Request returned an error: %s", err)
+		return
+	}
+
+	expected := &SimpleResponse{Result: "success"}
+
+	if !reflect.DeepEqual(got, expected) {
+		t.Errorf("Expected %+v, got %+v", expected, got)
+	}
+}

volume_test.go

@@ -260,3 +260,124 @@ func TestResizeVolume(t *testing.T) {
 		t.Errorf("Expected %+v, got %+v", expected, got)
 	}
 }
+
+func TestCreateVolumeSnapshot(t *testing.T) {
+	client, server, _ := NewClientForTesting(map[string]string{
+		"/v2/volumes/12346/snapshot": `{
+			"snapshot_id": "12345",
+			"name": "test-snapshot",
+			"snapshot_description": "snapshot for testing",
+			"volume_id": "12346",
+			"state": "available",
+			"creation_time": "2020-01-01T00:00:00Z"
+		}`,
+	})
+	defer server.Close()
+	cfg := &VolumeSnapshotConfig{Name: "my-snapshot"}
+	got, err := client.CreateVolumeSnapshot("12346", cfg)
+	if err != nil {
+		t.Errorf("Request returned an error: %s", err)
+		return
+	}
+
+	expected := &VolumeSnapshot{
+		SnapshotID:          "12345",
+		Name:                "test-snapshot",
+		SnapshotDescription: "snapshot for testing",
+		VolumeID:            "12346",
+		State:               "available",
+		CreationTime:        "2020-01-01T00:00:00Z",
+	}
+	if !reflect.DeepEqual(got, expected) {
+		t.Errorf("Expected %+v, got %+v", expected, got)
+	}
+}
+
+func TestGetVolumeSnapshotByVolumeID(t *testing.T) {
+	client, server, _ := NewClientForTesting(map[string]string{
+		"/v2/volumes/12346/snapshot/12345": `{
+			"snapshot_id": "12345",
+			"name": "test-snapshot",
+			"snapshot_description": "snapshot for testing",
+			"volume_id": "12346",
+			"state": "available",
+			"creation_time": "2020-01-01T00:00:00Z"
+		}`,
+	})
+	defer server.Close()
+
+	got, err := client.GetVolumeSnapshotByVolumeID("12346", "12345")
+	if err != nil {
+		t.Errorf("Request returned an error: %s", err)
+		return
+	}
+
+	expected := VolumeSnapshot{
+		SnapshotID:          "12345",
+		Name:                "test-snapshot",
+		SnapshotDescription: "snapshot for testing",
+		VolumeID:            "12346",
+		State:               "available",
+		CreationTime:        "2020-01-01T00:00:00Z",
+	}
+
+	if !reflect.DeepEqual(got, expected) {
+		t.Errorf("Expected %+v, got %+v", expected, got)
+	}
+}
+
+func TestListVolumeSnapshotsByVolumeID(t *testing.T) {
+	client, server, _ := NewClientForTesting(map[string]string{
+		"/v2/volumes/12346/snapshots": `[{
+			"snapshot_id": "12345",
+			"name": "test-snapshot",
+			"snapshot_description": "snapshot for testing",
+			"volume_id": "12346",
+			"state": "available",
+			"creation_time": "2020-01-01T00:00:00Z"
+		}]`,
+	})
+	defer server.Close()
+
+	got, err := client.ListVolumeSnapshotsByVolumeID("12346")
+	if err != nil {
+		t.Errorf("Request returned an error: %s", err)
+		return
+	}
+
+	expected := []VolumeSnapshot{
+		{
+			SnapshotID:          "12345",
+			Name:                "test-snapshot",
+			SnapshotDescription: "snapshot for testing",
+			VolumeID:            "12346",
+			State:               "available",
+			CreationTime:        "2020-01-01T00:00:00Z",
+		},
+	}
+
+	if !reflect.DeepEqual(got, expected) {
+		t.Errorf("Expected %+v, got %+v", expected, got)
+	}
+}
+
+func TestDeleteVolumeAndAllSnapshot(t *testing.T) {
+	client, server, _ := NewClientForTesting(map[string]string{
+		"/v2/volumes/12346?delete_snapshot=true": `{"result": "success"}`,
+	})
+	defer server.Close()
+
+	got, err := client.DeleteVolumeAndAllSnapshot("12346")
+	if err != nil {
+		t.Errorf("Request returned an error: %s", err)
+		return
+	}
+
+	expected := &SimpleResponse{
+		Result: "success",
+	}
+
+	if !reflect.DeepEqual(got, expected) {
+		t.Errorf("Expected %+v, got %+v", expected, got)
+	}
+}