Merge pull request #266 from planetscale/add-with-postgresql-list-option

Add WithPostgreSQL() ListOption for cluster size queries

Author: mscoutermarsh

planetscale/client.go

@@ -116,6 +116,14 @@ func WithRates() ListOption {
 	}
 }
 
+// WithPostgreSQL returns a ListOption that sets the "postgresql" URL parameter.
+func WithPostgreSQL() ListOption {
+	return func(opt *ListOptions) error {
+		opt.URLValues.Set("postgresql", "true")
+		return nil
+	}
+}
+
 // WithRegion returns a ListOption sets the "region" URL parameter.
 func WithRegion(region string) ListOption {
 	return func(opt *ListOptions) error {

planetscale/organizations_test.go

@@ -242,3 +242,59 @@ func TestOrganizations_ListClusterSKUsWithRates(t *testing.T) {
 
 	c.Assert(orgs, qt.DeepEquals, want)
 }
+
+func TestOrganizations_ListClusterSKUsWithPostgreSQL(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.URL.String(), qt.Equals, "/v1/organizations/my-cool-org/cluster-size-skus?postgresql=true")
+		out := `[
+		{
+			"name": "PS_10_AWS_ARM",
+			"type": "ClusterSizeSku",
+			"display_name": "PS-10-AWS-ARM",
+			"cpu": "1/8",
+			"provider_instance_type": null,
+			"storage": null,
+			"ram": 1,
+			"sort_order": 1,
+			"enabled": true,
+			"provider": "aws",
+			"rate": 34,
+			"replica_rate": null,
+			"default_vtgate": null,
+			"default_vtgate_rate": null
+		}
+	]`
+
+		_, err := w.Write([]byte(out))
+		c.Assert(err, qt.IsNil)
+	}))
+
+	client, err := NewClient(WithBaseURL(ts.URL))
+	c.Assert(err, qt.IsNil)
+
+	ctx := context.Background()
+
+	orgs, err := client.Organizations.ListClusterSKUs(ctx, &ListOrganizationClusterSKUsRequest{
+		Organization: "my-cool-org",
+	}, WithPostgreSQL())
+
+	c.Assert(err, qt.IsNil)
+	want := []*ClusterSKU{
+		{
+			Name:        "PS_10_AWS_ARM",
+			DisplayName: "PS-10-AWS-ARM",
+			CPU:         "1/8",
+			Memory:      1,
+			Enabled:     true,
+			Provider:    Pointer("aws"),
+			Rate:        Pointer[int64](34),
+			SortOrder:   1,
+		},
+	}
+
+	c.Assert(orgs, qt.DeepEquals, want)
+}