From e56b8a3e1755970e10f5832223901a473f666be1 Mon Sep 17 00:00:00 2001
From: Andrew Starr-Bochicchio <andrewsomething@users.noreply.github.com>
Date: Fri, 26 Jul 2024 17:31:51 -0400
Subject: [PATCH] database: Add Size to DatabaseReplica struct (#706)

---
 databases.go      |  1 +
 databases_test.go | 12 +++++++++---
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/databases.go b/databases.go
index b915391..4e926b7 100644
--- a/databases.go
+++ b/databases.go
@@ -392,6 +392,7 @@ type DatabaseReplica struct {
 	PrivateNetworkUUID string              `json:"private_network_uuid,omitempty"`
 	Tags               []string            `json:"tags,omitempty"`
 	StorageSizeMib     uint64              `json:"storage_size_mib,omitempty"`
+	Size               string              `json:"size"`
 }
 
 // DatabasePool represents a database connection pool
diff --git a/databases_test.go b/databases_test.go
index c3da663..395d58b 100644
--- a/databases_test.go
+++ b/databases_test.go
@@ -1568,6 +1568,7 @@ func TestDatabases_GetReplica(t *testing.T) {
 		PrivateNetworkUUID: "deadbeef-dead-4aa5-beef-deadbeef347d",
 		Tags:               []string{"production", "staging"},
 		StorageSizeMib:     51200,
+		Size:               "db-s-1vcpu-1gb",
 	}
 
 	body := `
@@ -1598,7 +1599,8 @@ func TestDatabases_GetReplica(t *testing.T) {
     },
     "private_network_uuid": "deadbeef-dead-4aa5-beef-deadbeef347d",
 	"tags": ["production", "staging"],
-	"storage_size_mib": 51200
+	"storage_size_mib": 51200,
+	"size": "db-s-1vcpu-1gb"
   }
 }
 `
@@ -1649,6 +1651,7 @@ func TestDatabases_ListReplicas(t *testing.T) {
 			PrivateNetworkUUID: "deadbeef-dead-4aa5-beef-deadbeef347d",
 			Tags:               []string{"production", "staging"},
 			StorageSizeMib:     51200,
+			Size:               "db-s-1vcpu-1gb",
 		},
 	}
 
@@ -1679,7 +1682,8 @@ func TestDatabases_ListReplicas(t *testing.T) {
     },
     "private_network_uuid": "deadbeef-dead-4aa5-beef-deadbeef347d",
 	"tags": ["production", "staging"],
-	"storage_size_mib": 51200
+	"storage_size_mib": 51200,
+	"size": "db-s-1vcpu-1gb"
   }]
 }
 `
@@ -1729,6 +1733,7 @@ func TestDatabases_CreateReplica(t *testing.T) {
 		PrivateNetworkUUID: "deadbeef-dead-4aa5-beef-deadbeef347d",
 		Tags:               []string{"production", "staging"},
 		StorageSizeMib:     51200,
+		Size:               "db-s-2vcpu-4gb",
 	}
 
 	body := `
@@ -1758,7 +1763,8 @@ func TestDatabases_CreateReplica(t *testing.T) {
     },
     "private_network_uuid": "deadbeef-dead-4aa5-beef-deadbeef347d",
 	"tags": ["production", "staging"],
-	"storage_size_mib": 51200
+	"storage_size_mib": 51200,
+	"size": "db-s-2vcpu-4gb"
   }
 }
 `