Add FloatingIPLimit to Account type.

digitalocean · Nov 24, 2015 · 09c1ef1 · 09c1ef1
1 parent 4ac7bea
commit 09c1ef1
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 14 deletions.
diff --git a/account.go b/account.go
@@ -17,12 +17,13 @@ var _ AccountService = &AccountServiceOp{}
 
 // Account represents a DigitalOcean Account
 type Account struct {
-	DropletLimit  int    `json:"droplet_limit,omitempty"`
-	Email         string `json:"email,omitempty"`
-	UUID          string `json:"uuid,omitempty"`
-	EmailVerified bool   `json:"email_verified,omitempty"`
-	Status        string `json:"status,omitempty"`
-	StatusMessage string `json:"status_message,omitempty"`
+	DropletLimit    int    `json:"droplet_limit,omitempty"`
+	FloatingIPLimit int    `json:"floating_ip_limit,omitempty"`
+	Email           string `json:"email,omitempty"`
+	UUID            string `json:"uuid,omitempty"`
+	EmailVerified   bool   `json:"email_verified,omitempty"`
+	Status          string `json:"status,omitempty"`
+	StatusMessage   string `json:"status_message,omitempty"`
 }
 
 type accountRoot struct {

diff --git a/account_test.go b/account_test.go
@@ -17,6 +17,7 @@ func TestAccountGet(t *testing.T) {
 		response := `
 		{ "account": {
 			"droplet_limit": 25,
+			"floating_ip_limit": 25,
 			"email": "sammy@digitalocean.com",
 			"uuid": "b6fr89dbf6d9156cace5f3c78dc9851d957381ef",
 			"email_verified": true
@@ -31,7 +32,7 @@ func TestAccountGet(t *testing.T) {
 		t.Errorf("Account.Get returned error: %v", err)
 	}
 
-	expected := &Account{DropletLimit: 25, Email: "sammy@digitalocean.com",
+	expected := &Account{DropletLimit: 25, FloatingIPLimit: 25, Email: "sammy@digitalocean.com",
 		UUID: "b6fr89dbf6d9156cace5f3c78dc9851d957381ef", EmailVerified: true}
 	if !reflect.DeepEqual(acct, expected) {
 		t.Errorf("Account.Get returned %+v, expected %+v", acct, expected)
@@ -40,16 +41,17 @@ func TestAccountGet(t *testing.T) {
 
 func TestAccountString(t *testing.T) {
 	acct := &Account{
-		DropletLimit:  25,
-		Email:         "sammy@digitalocean.com",
-		UUID:          "b6fr89dbf6d9156cace5f3c78dc9851d957381ef",
-		EmailVerified: true,
-		Status:        "active",
-		StatusMessage: "message",
+		DropletLimit:    25,
+		FloatingIPLimit: 25,
+		Email:           "sammy@digitalocean.com",
+		UUID:            "b6fr89dbf6d9156cace5f3c78dc9851d957381ef",
+		EmailVerified:   true,
+		Status:          "active",
+		StatusMessage:   "message",
 	}
 
 	stringified := acct.String()
-	expected := `godo.Account{DropletLimit:25, Email:"sammy@digitalocean.com", UUID:"b6fr89dbf6d9156cace5f3c78dc9851d957381ef", EmailVerified:true, Status:"active", StatusMessage:"message"}`
+	expected := `godo.Account{DropletLimit:25, FloatingIPLimit:25, Email:"sammy@digitalocean.com", UUID:"b6fr89dbf6d9156cace5f3c78dc9851d957381ef", EmailVerified:true, Status:"active", StatusMessage:"message"}`
 	if expected != stringified {
 		t.Errorf("Account.String returned %+v, expected %+v", stringified, expected)
 	}