gogreen/droplets_test.go

837 lines
20 KiB
Go

package godo
import (
"encoding/json"
"fmt"
"net/http"
"reflect"
"testing"
)
func TestDroplets_ListDroplets(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, `{
"droplets": [
{
"id": 1
},
{
"id": 2
}
],
"meta": {
"total": 2
}
}`)
})
droplets, resp, err := client.Droplets.List(ctx, nil)
if err != nil {
t.Errorf("Droplets.List returned error: %v", err)
}
expectedDroplets := []Droplet{{ID: 1}, {ID: 2}}
if !reflect.DeepEqual(droplets, expectedDroplets) {
t.Errorf("Droplets.List\nDroplets: got=%#v\nwant=%#v", droplets, expectedDroplets)
}
expectedMeta := &Meta{Total: 2}
if !reflect.DeepEqual(resp.Meta, expectedMeta) {
t.Errorf("Droplets.List\nMeta: got=%#v\nwant=%#v", resp.Meta, expectedMeta)
}
}
func TestDroplets_ListDropletsByTag(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
if r.URL.Query().Get("tag_name") != "testing-1" {
t.Errorf("Droplets.ListByTag did not request with a tag parameter")
}
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, `{
"droplets": [
{
"id": 1
},
{
"id": 2
}
],
"meta": {
"total": 2
}
}`)
})
droplets, resp, err := client.Droplets.ListByTag(ctx, "testing-1", nil)
if err != nil {
t.Errorf("Droplets.ListByTag returned error: %v", err)
}
expectedDroplets := []Droplet{{ID: 1}, {ID: 2}}
if !reflect.DeepEqual(droplets, expectedDroplets) {
t.Errorf("Droplets.ListByTag returned droplets %+v, expected %+v", droplets, expectedDroplets)
}
expectedMeta := &Meta{Total: 2}
if !reflect.DeepEqual(resp.Meta, expectedMeta) {
t.Errorf("Droplets.ListByTag returned meta %+v, expected %+v", resp.Meta, expectedMeta)
}
}
func TestDroplets_ListDropletsByName(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
name := "testing"
if r.URL.Query().Get("name") != name {
t.Errorf("Droplets.ListByName request did not contain the 'name=%s' query parameter", name)
}
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, `{
"droplets": [
{
"id": 1,
"name": "testing"
},
{
"id": 2,
"name": "testing"
}
],
"meta": {
"total": 2
}
}`)
})
droplets, _, err := client.Droplets.ListByName(ctx, "testing", nil)
if err != nil {
t.Errorf("Droplets.ListByTag returned error: %v", err)
}
expected := []Droplet{{ID: 1, Name: "testing"}, {ID: 2, Name: "testing"}}
if !reflect.DeepEqual(droplets, expected) {
t.Errorf("Droplets.ListByTag returned droplets %+v, expected %+v", droplets, expected)
}
}
func TestDroplets_ListDropletsMultiplePages(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodGet)
dr := dropletsRoot{
Droplets: []Droplet{
{ID: 1},
{ID: 2},
},
Links: nil,
}
b, err := json.Marshal(dr)
if err != nil {
t.Fatal(err)
}
fmt.Fprint(w, string(b))
})
_, _, err := client.Droplets.List(ctx, nil)
if err != nil {
t.Fatal(err)
}
}
func TestDroplets_RetrievePageByNumber(t *testing.T) {
setup()
defer teardown()
jBlob := `
{
"droplets": [{"id":1},{"id":2}],
"links": []
}`
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, jBlob)
})
opt := &ListOptions{Page: 2}
_, _, err := client.Droplets.List(ctx, opt)
if err != nil {
t.Fatal(err)
}
}
func TestDroplets_GetDroplet(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets/12345", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, `{"droplet":{"id":12345}}`)
})
droplets, _, err := client.Droplets.Get(ctx, 12345)
if err != nil {
t.Errorf("Droplet.Get returned error: %v", err)
}
expected := &Droplet{ID: 12345}
if !reflect.DeepEqual(droplets, expected) {
t.Errorf("Droplets.Get\n got=%#v\nwant=%#v", droplets, expected)
}
}
func TestDroplets_Create(t *testing.T) {
setup()
defer teardown()
createRequest := &DropletCreateRequest{
Name: "name",
Region: "region",
Size: "size",
Image: DropletCreateImage{
ID: 1,
},
Volumes: []DropletCreateVolume{
{ID: "hello-im-another-volume"},
{Name: "should be ignored due to Name", ID: "aaa-111-bbb-222-ccc"},
},
Tags: []string{"one", "two"},
VPCUUID: "880b7f98-f062-404d-b33c-458d545696f6",
}
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
expected := map[string]interface{}{
"name": "name",
"region": "region",
"size": "size",
"image": float64(1),
"ssh_keys": nil,
"backups": false,
"ipv6": false,
"private_networking": false,
"monitoring": false,
"volumes": []interface{}{
map[string]interface{}{"id": "hello-im-another-volume"},
map[string]interface{}{"id": "aaa-111-bbb-222-ccc"},
},
"tags": []interface{}{"one", "two"},
"vpc_uuid": "880b7f98-f062-404d-b33c-458d545696f6",
}
jsonBlob := `
{
"droplet": {
"id": 1,
"vpc_uuid": "880b7f98-f062-404d-b33c-458d545696f6"
},
"links": [
{
"id": 1,
"href": "http://example.com",
"rel": "create"
}
]
}
`
var v map[string]interface{}
err := json.NewDecoder(r.Body).Decode(&v)
if err != nil {
t.Fatalf("decode json: %v", err)
}
if !reflect.DeepEqual(v, expected) {
t.Errorf("Request body\n got=%#v\nwant=%#v", v, expected)
}
fmt.Fprintf(w, jsonBlob)
})
droplet, resp, err := client.Droplets.Create(ctx, createRequest)
if err != nil {
t.Errorf("Droplets.Create returned error: %v", err)
}
if id := droplet.ID; id != 1 {
t.Errorf("expected id '%d', received '%d'", 1, id)
}
vpcid := "880b7f98-f062-404d-b33c-458d545696f6"
if id := droplet.VPCUUID; id != vpcid {
t.Errorf("expected VPC uuid '%s', received '%s'", vpcid, id)
}
if a := resp.Links[0]; a.ID != 1 {
t.Errorf("expected action id '%d', received '%d'", 1, a.ID)
}
}
func TestDroplets_CreateWithoutDropletAgent(t *testing.T) {
setup()
defer teardown()
boolVal := false
createRequest := &DropletCreateRequest{
Name: "name",
Region: "region",
Size: "size",
Image: DropletCreateImage{
ID: 1,
},
Volumes: []DropletCreateVolume{
{ID: "hello-im-another-volume"},
{Name: "should be ignored due to Name", ID: "aaa-111-bbb-222-ccc"},
},
Tags: []string{"one", "two"},
VPCUUID: "880b7f98-f062-404d-b33c-458d545696f6",
WithDropletAgent: &boolVal,
}
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
expected := map[string]interface{}{
"name": "name",
"region": "region",
"size": "size",
"image": float64(1),
"ssh_keys": nil,
"backups": false,
"ipv6": false,
"private_networking": false,
"monitoring": false,
"volumes": []interface{}{
map[string]interface{}{"id": "hello-im-another-volume"},
map[string]interface{}{"id": "aaa-111-bbb-222-ccc"},
},
"tags": []interface{}{"one", "two"},
"vpc_uuid": "880b7f98-f062-404d-b33c-458d545696f6",
"with_droplet_agent": false,
}
jsonBlob := `
{
"droplet": {
"id": 1,
"vpc_uuid": "880b7f98-f062-404d-b33c-458d545696f6"
},
"links": [
{
"id": 1,
"href": "http://example.com",
"rel": "create"
}
]
}
`
var v map[string]interface{}
err := json.NewDecoder(r.Body).Decode(&v)
if err != nil {
t.Fatalf("decode json: %v", err)
}
if !reflect.DeepEqual(v, expected) {
t.Errorf("Request body\n got=%#v\nwant=%#v", v, expected)
}
fmt.Fprintf(w, jsonBlob)
})
droplet, resp, err := client.Droplets.Create(ctx, createRequest)
if err != nil {
t.Errorf("Droplets.Create returned error: %v", err)
}
if id := droplet.ID; id != 1 {
t.Errorf("expected id '%d', received '%d'", 1, id)
}
vpcid := "880b7f98-f062-404d-b33c-458d545696f6"
if id := droplet.VPCUUID; id != vpcid {
t.Errorf("expected VPC uuid '%s', received '%s'", vpcid, id)
}
if a := resp.Links[0]; a.ID != 1 {
t.Errorf("expected action id '%d', received '%d'", 1, a.ID)
}
}
func TestDroplets_WithDropletAgentJsonMarshal(t *testing.T) {
boolF := false
boolT := true
tests := []struct {
in *DropletCreateRequest
want string
}{
{
in: &DropletCreateRequest{Name: "foo", WithDropletAgent: &boolF},
want: `{"name":"foo","region":"","size":"","image":0,"ssh_keys":null,"backups":false,"ipv6":false,"private_networking":false,"monitoring":false,"tags":null,"with_droplet_agent":false}`,
},
{
in: &DropletCreateRequest{Name: "foo", WithDropletAgent: &boolT},
want: `{"name":"foo","region":"","size":"","image":0,"ssh_keys":null,"backups":false,"ipv6":false,"private_networking":false,"monitoring":false,"tags":null,"with_droplet_agent":true}`,
},
{
in: &DropletCreateRequest{Name: "foo"},
want: `{"name":"foo","region":"","size":"","image":0,"ssh_keys":null,"backups":false,"ipv6":false,"private_networking":false,"monitoring":false,"tags":null}`,
},
}
for _, tt := range tests {
got, err := json.Marshal(tt.in)
if err != nil {
t.Fatalf("error: %v", err)
}
if !reflect.DeepEqual(tt.want, string(got)) {
t.Errorf("expected: %v, got: %v", tt.want, string(got))
}
}
}
func TestDroplets_CreateWithDisabledPublicNetworking(t *testing.T) {
setup()
defer teardown()
createRequest := &DropletCreateRequest{
Name: "name",
Region: "region",
Size: "size",
Image: DropletCreateImage{
ID: 1,
},
Volumes: []DropletCreateVolume{
{ID: "hello-im-another-volume"},
{Name: "should be ignored due to Name", ID: "aaa-111-bbb-222-ccc"},
},
Tags: []string{"one", "two"},
VPCUUID: "880b7f98-f062-404d-b33c-458d545696f6",
}
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
expected := map[string]interface{}{
"name": "name",
"region": "region",
"size": "size",
"image": float64(1),
"ssh_keys": nil,
"backups": false,
"ipv6": false,
"private_networking": false,
"monitoring": false,
"volumes": []interface{}{
map[string]interface{}{"id": "hello-im-another-volume"},
map[string]interface{}{"id": "aaa-111-bbb-222-ccc"},
},
"tags": []interface{}{"one", "two"},
"vpc_uuid": "880b7f98-f062-404d-b33c-458d545696f6",
}
jsonBlob := `
{
"droplet": {
"id": 1,
"vpc_uuid": "880b7f98-f062-404d-b33c-458d545696f6"
},
"links": [
{
"id": 1,
"href": "http://example.com",
"rel": "create"
}
]
}
`
var v map[string]interface{}
err := json.NewDecoder(r.Body).Decode(&v)
if err != nil {
t.Fatalf("decode json: %v", err)
}
if !reflect.DeepEqual(v, expected) {
t.Errorf("Request body\n got=%#v\nwant=%#v", v, expected)
}
fmt.Fprintf(w, jsonBlob)
})
droplet, _, err := client.Droplets.Create(ctx, createRequest)
if err != nil {
t.Errorf("Droplets.Create returned error: %v", err)
}
if id := droplet.ID; id != 1 {
t.Errorf("expected id '%d', received '%d'", 1, id)
}
}
func TestDroplets_CreateMultiple(t *testing.T) {
setup()
defer teardown()
createRequest := &DropletMultiCreateRequest{
Names: []string{"name1", "name2"},
Region: "region",
Size: "size",
Image: DropletCreateImage{
ID: 1,
},
Tags: []string{"one", "two"},
VPCUUID: "880b7f98-f062-404d-b33c-458d545696f6",
}
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
expected := map[string]interface{}{
"names": []interface{}{"name1", "name2"},
"region": "region",
"size": "size",
"image": float64(1),
"ssh_keys": nil,
"backups": false,
"ipv6": false,
"private_networking": false,
"monitoring": false,
"tags": []interface{}{"one", "two"},
"vpc_uuid": "880b7f98-f062-404d-b33c-458d545696f6",
}
jsonBlob := `
{
"droplets": [
{
"id": 1,
"vpc_uuid": "880b7f98-f062-404d-b33c-458d545696f6"
},
{
"id": 2,
"vpc_uuid": "880b7f98-f062-404d-b33c-458d545696f6"
}
],
"links": [
{
"id": 1,
"href": "http://example.com",
"rel": "multiple_create"
}
]
}
`
var v map[string]interface{}
err := json.NewDecoder(r.Body).Decode(&v)
if err != nil {
t.Fatalf("decode json: %v", err)
}
if !reflect.DeepEqual(v, expected) {
t.Errorf("Request body = %#v, expected %#v", v, expected)
}
fmt.Fprintf(w, jsonBlob)
})
droplets, resp, err := client.Droplets.CreateMultiple(ctx, createRequest)
if err != nil {
t.Errorf("Droplets.CreateMultiple returned error: %v", err)
}
if id := droplets[0].ID; id != 1 {
t.Errorf("expected id '%d', received '%d'", 1, id)
}
if id := droplets[1].ID; id != 2 {
t.Errorf("expected id '%d', received '%d'", 2, id)
}
vpcid := "880b7f98-f062-404d-b33c-458d545696f6"
if id := droplets[0].VPCUUID; id != vpcid {
t.Errorf("expected VPC uuid '%s', received '%s'", vpcid, id)
}
if id := droplets[1].VPCUUID; id != vpcid {
t.Errorf("expected VPC uuid '%s', received '%s'", vpcid, id)
}
if a := resp.Links[0]; a.ID != 1 {
t.Errorf("expected action id '%d', received '%d'", 1, a.ID)
}
}
func TestDroplets_Destroy(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets/12345", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodDelete)
})
_, err := client.Droplets.Delete(ctx, 12345)
if err != nil {
t.Errorf("Droplet.Delete returned error: %v", err)
}
}
func TestDroplets_DestroyByTag(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets", func(w http.ResponseWriter, r *http.Request) {
if r.URL.Query().Get("tag_name") != "testing-1" {
t.Errorf("Droplets.DeleteByTag did not request with a tag parameter")
}
testMethod(t, r, http.MethodDelete)
})
_, err := client.Droplets.DeleteByTag(ctx, "testing-1")
if err != nil {
t.Errorf("Droplet.Delete returned error: %v", err)
}
}
func TestDroplets_Kernels(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets/12345/kernels", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, `{
"kernels": [
{
"id": 1
},
{
"id": 2
}
],
"meta": {
"total": 2
}
}`)
})
opt := &ListOptions{Page: 2}
kernels, resp, err := client.Droplets.Kernels(ctx, 12345, opt)
if err != nil {
t.Errorf("Droplets.Kernels returned error: %v", err)
}
expectedKernels := []Kernel{{ID: 1}, {ID: 2}}
if !reflect.DeepEqual(kernels, expectedKernels) {
t.Errorf("Droplets.Kernels\nKernels got=%#v\nwant=%#v", kernels, expectedKernels)
}
expectedMeta := &Meta{Total: 2}
if !reflect.DeepEqual(resp.Meta, expectedMeta) {
t.Errorf("Droplets.Kernels\nMeta: got=%#v\nwant=%#v", resp.Meta, expectedMeta)
}
}
func TestDroplets_Snapshots(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets/12345/snapshots", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, `{
"snapshots": [
{
"id": "1"
},
{
"id": "2"
}
],
"meta": {
"total": 2
}
}`)
})
opt := &ListOptions{Page: 2}
snapshots, resp, err := client.Droplets.Snapshots(ctx, 12345, opt)
if err != nil {
t.Errorf("Droplets.Snapshots returned error: %v", err)
}
expectedSnapshots := []Image{{ID: "1"}, {ID: "2"}}
if !reflect.DeepEqual(snapshots, expectedSnapshots) {
t.Errorf("Droplets.Snapshots\nSnapshots got=%#v\nwant=%#v", snapshots, expectedSnapshots)
}
expectedMeta := &Meta{Total: 2}
if !reflect.DeepEqual(resp.Meta, expectedMeta) {
t.Errorf("Droplets.Snapshots\nMeta: got=%#v\nwant=%#v", resp.Meta, expectedMeta)
}
}
func TestDroplets_Backups(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets/12345/backups", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, `{
"backups": [
{
"id": "1"
},
{
"id": "2"
}
],
"meta": {
"total": 2
}
}`)
})
opt := &ListOptions{Page: 2}
backups, resp, err := client.Droplets.Backups(ctx, 12345, opt)
if err != nil {
t.Errorf("Droplets.Backups returned error: %v", err)
}
expectedBackups := []Image{{ID: "1"}, {ID: "2"}}
if !reflect.DeepEqual(backups, expectedBackups) {
t.Errorf("Droplets.Backups\nBackups got=%#v\nwant=%#v", backups, expectedBackups)
}
expectedMeta := &Meta{Total: 2}
if !reflect.DeepEqual(resp.Meta, expectedMeta) {
t.Errorf("Droplets.Backups\nMeta: got=%#v\nwant=%#v", resp.Meta, expectedMeta)
}
}
func TestDroplets_Actions(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets/12345/actions", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, `{
"actions": [
{
"id": 1
},
{
"id": 2
}
],
"meta": {
"total": 2
}
}`)
})
opt := &ListOptions{Page: 2}
actions, resp, err := client.Droplets.Actions(ctx, 12345, opt)
if err != nil {
t.Errorf("Droplets.Actions returned error: %v", err)
}
expectedActions := []Action{{ID: 1}, {ID: 2}}
if !reflect.DeepEqual(actions, expectedActions) {
t.Errorf("Droplets.Actions\nActions got=%#v\nwant=%#v", actions, expectedActions)
}
expectedMeta := &Meta{Total: 2}
if !reflect.DeepEqual(resp.Meta, expectedMeta) {
t.Errorf("Droplets.Actions\nMeta: got=%#v\nwant=%#v", resp.Meta, expectedMeta)
}
}
func TestDroplets_Neighbors(t *testing.T) {
setup()
defer teardown()
mux.HandleFunc("/v2/droplets/12345/neighbors", func(w http.ResponseWriter, r *http.Request) {
testMethod(t, r, http.MethodGet)
fmt.Fprint(w, `{"droplets": [{"id":1},{"id":2}]}`)
})
neighbors, _, err := client.Droplets.Neighbors(ctx, 12345)
if err != nil {
t.Errorf("Droplets.Neighbors returned error: %v", err)
}
expected := []Droplet{{ID: 1}, {ID: 2}}
if !reflect.DeepEqual(neighbors, expected) {
t.Errorf("Droplets.Neighbors\n got=%#v\nwant=%#v", neighbors, expected)
}
}
func TestNetworkV4_String(t *testing.T) {
network := &NetworkV4{
IPAddress: "192.168.1.2",
Netmask: "255.255.255.0",
Gateway: "192.168.1.1",
}
stringified := network.String()
expected := `godo.NetworkV4{IPAddress:"192.168.1.2", Netmask:"255.255.255.0", Gateway:"192.168.1.1", Type:""}`
if expected != stringified {
t.Errorf("NetworkV4.String\n got=%#v\nwant=%#v", stringified, expected)
}
}
func TestNetworkV6_String(t *testing.T) {
network := &NetworkV6{
IPAddress: "2604:A880:0800:0010:0000:0000:02DD:4001",
Netmask: 64,
Gateway: "2604:A880:0800:0010:0000:0000:0000:0001",
}
stringified := network.String()
expected := `godo.NetworkV6{IPAddress:"2604:A880:0800:0010:0000:0000:02DD:4001", Netmask:64, Gateway:"2604:A880:0800:0010:0000:0000:0000:0001", Type:""}`
if expected != stringified {
t.Errorf("NetworkV6.String\n got=%#v\nwant=%#v", stringified, expected)
}
}
func TestDroplets_IPMethods(t *testing.T) {
var d Droplet
ipv6 := "1000:1000:1000:1000:0000:0000:004D:B001"
d.Networks = &Networks{
V4: []NetworkV4{
{IPAddress: "192.168.0.1", Type: "public"},
{IPAddress: "10.0.0.1", Type: "private"},
},
V6: []NetworkV6{
{IPAddress: ipv6, Type: "public"},
},
}
ip, err := d.PublicIPv4()
if err != nil {
t.Errorf("unknown error")
}
if got, expected := ip, "192.168.0.1"; got != expected {
t.Errorf("Droplet.PublicIPv4 returned %s; expected %s", got, expected)
}
ip, err = d.PrivateIPv4()
if err != nil {
t.Errorf("unknown error")
}
if got, expected := ip, "10.0.0.1"; got != expected {
t.Errorf("Droplet.PrivateIPv4 returned %s; expected %s", got, expected)
}
ip, err = d.PublicIPv6()
if err != nil {
t.Errorf("unknown error")
}
if got, expected := ip, ipv6; got != expected {
t.Errorf("Droplet.PublicIPv6 returned %s; expected %s", got, expected)
}
}