Added support for DBaaS suspend/resume (#680)

* Added support for DBaaS suspend/resume

* Added unit tests

* Add comments

Author: ezilber-akamai

mysql.go

@@ -232,3 +232,16 @@ func (c *Client) PatchMySQLDatabase(ctx context.Context, databaseID int) error {
 	e := formatAPIPath("databases/mysql/instances/%d/patch", databaseID)
 	return doPOSTRequestNoRequestResponseBody(ctx, c, e)
 }
+
+// SuspendMySQLDatabase suspends a MySQL Managed Database, releasing idle resources and keeping only necessary data.
+// All service data is lost if there are no backups available.
+func (c *Client) SuspendMySQLDatabase(ctx context.Context, databaseID int) error {
+	e := formatAPIPath("databases/mysql/instances/%d/suspend", databaseID)
+	return doPOSTRequestNoRequestResponseBody(ctx, c, e)
+}
+
+// ResumeMySQLDatabase resumes a suspended MySQL Managed Database
+func (c *Client) ResumeMySQLDatabase(ctx context.Context, databaseID int) error {
+	e := formatAPIPath("databases/mysql/instances/%d/resume", databaseID)
+	return doPOSTRequestNoRequestResponseBody(ctx, c, e)
+}

postgres.go

@@ -252,3 +252,16 @@ func (c *Client) CreatePostgresDatabaseBackup(ctx context.Context, databaseID in
 	e := formatAPIPath("databases/postgresql/instances/%d/backups", databaseID)
 	return doPOSTRequestNoResponseBody(ctx, c, e, opts)
 }
+
+// SuspendPostgresDatabase suspends a PostgreSQL Managed Database, releasing idle resources and keeping only necessary data.
+// All service data is lost if there are no backups available.
+func (c *Client) SuspendPostgresDatabase(ctx context.Context, databaseID int) error {
+	e := formatAPIPath("databases/postgresql/instances/%d/suspend", databaseID)
+	return doPOSTRequestNoRequestResponseBody(ctx, c, e)
+}
+
+// ResumePostgresDatabase resumes a suspended PostgreSQL Managed Database
+func (c *Client) ResumePostgresDatabase(ctx context.Context, databaseID int) error {
+	e := formatAPIPath("databases/postgresql/instances/%d/resume", databaseID)
+	return doPOSTRequestNoRequestResponseBody(ctx, c, e)
+}

test/integration/fixtures/TestDatabase_MySQL_Suite.yaml

@@ -126,7 +126,29 @@ func TestDatabase_MySQL_Suite(t *testing.T) {
 	if err := client.WaitForDatabaseStatus(
 		context.Background(), database.ID, linodego.DatabaseEngineTypeMySQL,
 		linodego.DatabaseStatusActive, 2400); err != nil {
-		t.Fatalf("failed to wait for database updating: %s", err)
+		t.Fatalf("failed to wait for database active: %s", err)
+	}
+
+	if err := client.SuspendMySQLDatabase(context.Background(), database.ID); err != nil {
+		t.Fatalf("failed to suspend database: %s", err)
+	}
+
+	// Wait for the DB to enter suspended status
+	if err := client.WaitForDatabaseStatus(
+		context.Background(), database.ID, linodego.DatabaseEngineTypeMySQL,
+		linodego.DatabaseStatusSuspended, 240); err != nil {
+		t.Fatalf("failed to wait for database suspended: %s", err)
+	}
+
+	if err := client.ResumeMySQLDatabase(context.Background(), database.ID); err != nil {
+		t.Fatalf("failed to resume database: %s", err)
+	}
+
+	// Wait for the DB to re-enter active status
+	if err := client.WaitForDatabaseStatus(
+		context.Background(), database.ID, linodego.DatabaseEngineTypeMySQL,
+		linodego.DatabaseStatusActive, 2400); err != nil {
+		t.Fatalf("failed to wait for database active: %s", err)
 	}
 }
 

test/integration/fixtures/TestDatabase_Postgres_Suite.yaml

@@ -132,6 +132,28 @@ func TestDatabase_Postgres_Suite(t *testing.T) {
 		linodego.DatabaseStatusActive, 2400); err != nil {
 		t.Fatalf("failed to wait for database updating: %s", err)
 	}
+
+	if err := client.SuspendPostgresDatabase(context.Background(), database.ID); err != nil {
+		t.Fatalf("failed to suspend database: %s", err)
+	}
+
+	// Wait for the DB to enter suspended status
+	if err := client.WaitForDatabaseStatus(
+		context.Background(), database.ID, linodego.DatabaseEngineTypePostgres,
+		linodego.DatabaseStatusSuspended, 2400); err != nil {
+		t.Fatalf("failed to wait for database suspended: %s", err)
+	}
+
+	if err := client.ResumePostgresDatabase(context.Background(), database.ID); err != nil {
+		t.Fatalf("failed to resume database: %s", err)
+	}
+
+	// Wait for the DB to re-enter active status
+	if err := client.WaitForDatabaseStatus(
+		context.Background(), database.ID, linodego.DatabaseEngineTypePostgres,
+		linodego.DatabaseStatusActive, 2400); err != nil {
+		t.Fatalf("failed to wait for database active: %s", err)
+	}
 }
 
 type postgresDatabaseModifier func(options *linodego.PostgresCreateOptions)

test/integration/mysql_test.go

@@ -192,3 +192,23 @@ func TestDatabaseMySQL_Patch(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestDatabaseMySQL_Suspend(t *testing.T) {
+	client := createMockClient(t)
+
+	httpmock.RegisterRegexpResponder("POST", mockRequestURL(t, "databases/mysql/instances/123/suspend"), httpmock.NewStringResponder(200, "{}"))
+
+	if err := client.SuspendMySQLDatabase(context.Background(), 123); err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestDatabaseMySQL_Resume(t *testing.T) {
+	client := createMockClient(t)
+
+	httpmock.RegisterRegexpResponder("POST", mockRequestURL(t, "databases/mysql/instances/123/resume"), httpmock.NewStringResponder(200, "{}"))
+
+	if err := client.ResumeMySQLDatabase(context.Background(), 123); err != nil {
+		t.Fatal(err)
+	}
+}

test/integration/postgres_test.go

@@ -191,3 +191,23 @@ func TestDatabasePostgreSQL_Patch(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestDatabasePostgreSQL_Suspend(t *testing.T) {
+	client := createMockClient(t)
+
+	httpmock.RegisterRegexpResponder("POST", mockRequestURL(t, "databases/postgresql/instances/123/suspend"), httpmock.NewStringResponder(200, "{}"))
+
+	if err := client.SuspendPostgresDatabase(context.Background(), 123); err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestDatabasePostgreSQL_Resume(t *testing.T) {
+	client := createMockClient(t)
+
+	httpmock.RegisterRegexpResponder("POST", mockRequestURL(t, "databases/postgresql/instances/123/resume"), httpmock.NewStringResponder(200, "{}"))
+
+	if err := client.ResumePostgresDatabase(context.Background(), 123); err != nil {
+		t.Fatal(err)
+	}
+}