connect/disconnect region endpoints

Signed-off-by: Harsh Thakur <[email protected]>

fake_client.go

@@ -126,6 +126,8 @@ type Clienter interface {
 	// Regions
 	ListRegions() ([]Region, error)
 	CreateRegion(r *CreateRegionRequest) (*Region, error)
+	ConnectRegion(r *ConnectRegionRequest) error
+	DisconnectRegion(r *DisconnectRegionRequest) error
 
 	// Snapshots
 	// CreateSnapshot(name string, r *SnapshotConfig) (*Snapshot, error)
@@ -1017,6 +1019,16 @@ func (c *FakeClient) CreateRegion(r *CreateRegionRequest) (*Region, error) {
 	return &region, nil
 }
 
+// ConnectRegion implemented in a fake way for automated tests
+func (c *FakeClient) ConnectRegion(r *ConnectRegionRequest) error {
+	return nil
+}
+
+// DisconnectRegion implemented in a fake way for automated tests
+func (c *FakeClient) DisconnectRegion(r *DisconnectRegionRequest) error {
+	return nil
+}
+
 // CreateSnapshot implemented in a fake way for automated tests
 // func (c *FakeClient) CreateSnapshot(name string, r *SnapshotConfig) (*Snapshot, error) {
 // 	snapshot := Snapshot{

region.go

@@ -46,6 +46,16 @@ type CreateRegionRequest struct {
 	Features                 map[string]bool `json:"features" `
 }
 
+// DisconnectRegionRequest is the request to disconnect a region
+type DisconnectRegionRequest struct {
+	Code string `json:"code"`
+}
+
+// ConnectRegionRequest is the request to connect a region
+type ConnectRegionRequest struct {
+	Code string `json:"code"`
+}
+
 // ListRegions returns all load balancers owned by the calling API account
 func (c *Client) ListRegions() ([]Region, error) {
 	resp, err := c.SendGetRequest("/v2/regions")
@@ -129,3 +139,21 @@ func (c *Client) CreateRegion(r *CreateRegionRequest) (*Region, error) {
 
 	return &region, nil
 }
+
+// ConnectRegion connects a region to CivoAPI
+func (c *Client) ConnectRegion(r *ConnectRegionRequest) error {
+	_, err := c.SendPostRequest("/v2/regions/connect", r)
+	if err != nil {
+		return decodeError(err)
+	}
+	return nil
+}
+
+// DisconnectRegion disconnects a region to CivoAPI
+func (c *Client) DisconnectRegion(r *DisconnectRegionRequest) error {
+	_, err := c.SendPostRequest("/v2/regions/disconnect", r)
+	if err != nil {
+		return decodeError(err)
+	}
+	return nil
+}

region_test.go

@@ -97,3 +97,35 @@ func TestCreateRegion(t *testing.T) {
 	}
 
 }
+
+func TestConnectRegion(t *testing.T) {
+	client, server, _ := NewClientForTesting(map[string]string{
+		"/v2/regions/connect": `{code: "TEST1"}`,
+	})
+	defer server.Close()
+
+	connectRegionRequest := &ConnectRegionRequest{
+		Code: "TEST1",
+	}
+
+	err := client.ConnectRegion(connectRegionRequest)
+	if err != nil {
+		t.Errorf("Request returned an error: %s", err)
+	}
+}
+
+func TestDisconnectRegion(t *testing.T) {
+	client, server, _ := NewClientForTesting(map[string]string{
+		"/v2/regions/disconnect": `{code: "TEST1"}`,
+	})
+	defer server.Close()
+
+	disconnectRegionRequest := &DisconnectRegionRequest{
+		Code: "TEST1",
+	}
+
+	err := client.DisconnectRegion(disconnectRegionRequest)
+	if err != nil {
+		t.Errorf("Request returned an error: %s", err)
+	}
+}