Add storage size parameters to create database and branch requests

planetscale/databases.go

@@ -15,6 +15,12 @@ const (
 	DatabaseEnginePostgres DatabaseEngine = "postgresql"
 )
 
+// StorageConfig represents storage size configuration for a database or branch.
+type StorageConfig struct {
+	MinimumStorageBytes *int64 `json:"minimum_storage_bytes,omitempty"`
+	MaximumStorageBytes *int64 `json:"maximum_storage_bytes,omitempty"`
+}
+
 // CreateDatabaseRequest encapsulates the request for creating a new database.
 type CreateDatabaseRequest struct {
 	Organization string
@@ -25,6 +31,7 @@ type CreateDatabaseRequest struct {
 	Kind         DatabaseEngine `json:"kind,omitempty"`
 	Replicas     *int           `json:"replicas,omitempty"`
 	MajorVersion string         `json:"major_version,omitempty"`
+	Storage      *StorageConfig `json:"storage,omitempty"`
 }
 
 // DatabaseRequest encapsulates the request for getting a single database.

planetscale/databases_test.go

@@ -205,6 +205,61 @@ func TestDatabases_CreateWithReplicasZero(t *testing.T) {
 	c.Assert(db, qt.DeepEquals, want)
 }
 
+func TestDatabases_CreateWithStorage(t *testing.T) {
+	c := qt.New(t)
+
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(200)
+		c.Assert(r.Method, qt.Equals, http.MethodPost)
+
+		var body map[string]any
+		err := json.NewDecoder(r.Body).Decode(&body)
+		c.Assert(err, qt.IsNil)
+
+		storage, ok := body["storage"].(map[string]any)
+		c.Assert(ok, qt.IsTrue, qt.Commentf("storage field should be a nested object"))
+		c.Assert(storage["minimum_storage_bytes"], qt.Equals, float64(10737418240))
+		c.Assert(storage["maximum_storage_bytes"], qt.Equals, float64(107374182400))
+
+		out := `{"id":"planetscale-go-test-db","type":"database","name":"planetscale-go-test-db","notes":"","created_at":"2021-01-14T10:19:23.000Z","updated_at":"2021-01-14T10:19:23.000Z", "region": { "slug": "us-west", "display_name": "US West" },"state":"ready","kind":"postgresql"}`
+		_, err = w.Write([]byte(out))
+		c.Assert(err, qt.IsNil)
+	}))
+
+	client, err := NewClient(WithBaseURL(ts.URL))
+	c.Assert(err, qt.IsNil)
+
+	ctx := context.Background()
+	minStorage := int64(10737418240)
+	maxStorage := int64(107374182400)
+
+	db, err := client.Databases.Create(ctx, &CreateDatabaseRequest{
+		Organization: testOrg,
+		Region:       "us-west",
+		Name:         testDatabase,
+		Kind:         DatabaseEnginePostgres,
+		Storage: &StorageConfig{
+			MinimumStorageBytes: &minStorage,
+			MaximumStorageBytes: &maxStorage,
+		},
+	})
+
+	want := &Database{
+		Name:  testDatabase,
+		State: DatabaseReady,
+		Kind:  DatabaseEnginePostgres,
+		Region: Region{
+			Slug: "us-west",
+			Name: "US West",
+		},
+		CreatedAt: time.Date(2021, time.January, 14, 10, 19, 23, 0, time.UTC),
+		UpdatedAt: time.Date(2021, time.January, 14, 10, 19, 23, 0, time.UTC),
+	}
+
+	c.Assert(err, qt.IsNil)
+	c.Assert(db, qt.DeepEquals, want)
+}
+
 func TestDatabases_Get(t *testing.T) {
 	c := qt.New(t)
 

planetscale/postgres_branches.go

@@ -42,7 +42,8 @@ type CreatePostgresBranchRequest struct {
 	ParentBranch string `json:"parent_branch"`
 	BackupID     string `json:"backup_id,omitempty"`
 	ClusterName  string `json:"cluster_name,omitempty"`
-	MajorVersion string `json:"major_version,omitempty"`
+	MajorVersion string         `json:"major_version,omitempty"`
+	Storage      *StorageConfig `json:"storage,omitempty"`
 }
 
 // ListPostgresBranchesRequest encapsulates the request to list Postgres branches for a database.

planetscale/postgres_branches_test.go

@@ -2,6 +2,7 @@ package planetscale
 
 import (
 	"context"
+	"encoding/json"
 	"net/http"
 	"net/http/httptest"
 	"testing"
@@ -324,3 +325,58 @@ func TestPostgresBranches_ListClusterSKUs(t *testing.T) {
 	c.Assert(err, qt.IsNil)
 	c.Assert(skus, qt.DeepEquals, want)
 }
+
+func TestPostgresBranches_CreateWithStorage(t *testing.T) {
+	c := qt.New(t)
+
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(200)
+		c.Assert(r.Method, qt.Equals, http.MethodPost)
+
+		var body map[string]any
+		err := json.NewDecoder(r.Body).Decode(&body)
+		c.Assert(err, qt.IsNil)
+
+		storage, ok := body["storage"].(map[string]any)
+		c.Assert(ok, qt.IsTrue, qt.Commentf("storage field should be a nested object"))
+		c.Assert(storage["minimum_storage_bytes"], qt.Equals, float64(10737418240))
+		c.Assert(storage["maximum_storage_bytes"], qt.Equals, float64(107374182400))
+
+		out := `{"id":"postgres-test-branch","name":"postgres-test-branch","created_at":"2021-01-14T10:19:23.000Z","updated_at":"2021-01-14T10:19:23.000Z", "region": {"slug": "us-west", "display_name": "US West"}}`
+		_, err = w.Write([]byte(out))
+		c.Assert(err, qt.IsNil)
+	}))
+
+	client, err := NewClient(WithBaseURL(ts.URL))
+	c.Assert(err, qt.IsNil)
+
+	ctx := context.Background()
+	minStorage := int64(10737418240)
+	maxStorage := int64(107374182400)
+
+	branch, err := client.PostgresBranches.Create(ctx, &CreatePostgresBranchRequest{
+		Organization: "my-org",
+		Database:     "postgres-test-db",
+		Region:       "us-west",
+		Name:         testPostgresBranch,
+		ParentBranch: "main",
+		Storage: &StorageConfig{
+			MinimumStorageBytes: &minStorage,
+			MaximumStorageBytes: &maxStorage,
+		},
+	})
+
+	want := &PostgresBranch{
+		ID:   "postgres-test-branch",
+		Name: testPostgresBranch,
+		Region: Region{
+			Slug: "us-west",
+			Name: "US West",
+		},
+		CreatedAt: time.Date(2021, time.January, 14, 10, 19, 23, 0, time.UTC),
+		UpdatedAt: time.Date(2021, time.January, 14, 10, 19, 23, 0, time.UTC),
+	}
+
+	c.Assert(err, qt.IsNil)
+	c.Assert(branch, qt.DeepEquals, want)
+}