Merge pull request #81 from planetscale/iheanyi/add-organization-regions

Add ListRegions to OrganizationsService.

Author: iheanyi

Diff for: planetscale/organizations.go

@@ -22,6 +22,13 @@ type GetOrganizationRequest struct {
 type OrganizationsService interface {
 	Get(context.Context, *GetOrganizationRequest) (*Organization, error)
 	List(context.Context) ([]*Organization, error)
+	ListRegions(context.Context, *ListOrganizationRegionsRequest) ([]*Region, error)
+}
+
+// ListRegionsRequest encapsulates the request for getting a list of regions for
+// an organization.
+type ListOrganizationRegionsRequest struct {
+	Organization string
 }
 
 // Organization represents a PlanetScale organization.
@@ -76,3 +83,21 @@ func (o *organizationsService) List(ctx context.Context) ([]*Organization, error
 
 	return orgResponse.Organizations, nil
 }
+
+type listRegionsResponse struct {
+	Regions []*Region `json:"data"`
+}
+
+func (o *organizationsService) ListRegions(ctx context.Context, listReq *ListOrganizationRegionsRequest) ([]*Region, error) {
+	req, err := o.client.newRequest(http.MethodGet, fmt.Sprintf("%s/%s/regions", organizationsAPIPath, listReq.Organization), nil)
+	if err != nil {
+		return nil, err
+	}
+
+	listResponse := &listRegionsResponse{}
+	if err := o.client.do(ctx, req, &listResponse); err != nil {
+		return nil, err
+	}
+
+	return listResponse.Regions, nil
+}

Diff for: planetscale/organizations_test.go

@@ -85,3 +85,45 @@ func TestOrganizations_Get(t *testing.T) {
 
 	c.Assert(org, qt.DeepEquals, want)
 }
+
+func TestOrganizations_ListRegions(t *testing.T) {
+	c := qt.New(t)
+
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(200)
+		out := `{
+"data": [
+		{
+			"id": "my-cool-org",
+			"type": "Region",
+			"slug": "us-east",
+			"display_name": "US East",
+			"enabled": true
+		}
+	]
+}`
+
+		_, 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.ListRegions(ctx, &ListOrganizationRegionsRequest{
+		Organization: "my-cool-org",
+	})
+
+	c.Assert(err, qt.IsNil)
+	want := []*Region{
+		{
+			Name:    "US East",
+			Slug:    "us-east",
+			Enabled: true,
+		},
+	}
+
+	c.Assert(orgs, qt.DeepEquals, want)
+}