From 4f7309da6ec73edc3c2c0f8544e9530871b767da Mon Sep 17 00:00:00 2001
From: Rahul Bhardwaj <rahulbhardwaj@digitalocean.com>
Date: Wed, 3 Apr 2024 21:47:06 +0530
Subject: [PATCH] Add Opensearch to Godo

---
 databases.go      |  1 +
 databases_test.go | 55 ++++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 55 insertions(+), 1 deletion(-)

diff --git a/databases.go b/databases.go
index 2deb17b7..ea027980 100644
--- a/databases.go
+++ b/databases.go
@@ -698,6 +698,7 @@ type DatabaseOptions struct {
 	PostgresSQLOptions DatabaseEngineOptions `json:"pg"`
 	RedisOptions       DatabaseEngineOptions `json:"redis"`
 	KafkaOptions       DatabaseEngineOptions `json:"kafka"`
+	OpensearchOptions  DatabaseEngineOptions `json:"opensearch"`
 }
 
 // DatabaseEngineOptions represents the configuration options that are available for a given database engine
diff --git a/databases_test.go b/databases_test.go
index 2035ccf2..aba28fd2 100644
--- a/databases_test.go
+++ b/databases_test.go
@@ -2078,7 +2078,56 @@ func TestDatabases_GetDatabaseOptions(t *testing.T) {
 						"num_nodes": 3,
 						"sizes": [
 							"gd-2vcpu-8gb",
-	            "gd-4vcpu-16gb"
+	            			"gd-4vcpu-16gb"
+						]
+					}
+				]
+			},
+			"opensearch": {
+				"regions": [
+					"ams3",
+					"tor1"
+				],
+				"versions": [
+					"1",
+					"2"
+				],
+				"layouts": [
+					{
+						"num_nodes": 1,
+						"sizes": [
+							"db-s-2vcpu-4gb",
+							"db-s-4vcpu-8gb"
+						]
+					},
+					{
+						"num_nodes": 3,
+						"sizes": [
+							"db-s-2vcpu-4gb",
+							"m3-2vcpu-16gb",
+							"db-s-4vcpu-8gb",
+							"m3-4vcpu-32gb"
+						]
+					},
+					{
+						"num_nodes": 6,
+						"sizes": [
+							"m3-2vcpu-16gb",
+							"m3-4vcpu-32gb"
+						]
+					},
+					{
+						"num_nodes": 9,
+						"sizes": [
+							"m3-2vcpu-16gb",
+							"m3-4vcpu-32gb"
+						]
+					},
+					{
+						"num_nodes": 15,
+						"sizes": [
+							"m3-2vcpu-16gb",
+							"m3-4vcpu-32gb"
 						]
 					}
 				]
@@ -2124,21 +2173,25 @@ func TestDatabases_GetDatabaseOptions(t *testing.T) {
 	require.NotNil(t, options.RedisOptions)
 	require.NotNil(t, options.MySQLOptions)
 	require.NotNil(t, options.KafkaOptions)
+	require.NotNil(t, options.OpensearchOptions)
 	require.Greater(t, len(options.MongoDBOptions.Regions), 0)
 	require.Greater(t, len(options.PostgresSQLOptions.Regions), 0)
 	require.Greater(t, len(options.RedisOptions.Regions), 0)
 	require.Greater(t, len(options.MySQLOptions.Regions), 0)
 	require.Greater(t, len(options.KafkaOptions.Regions), 0)
+	require.Greater(t, len(options.OpensearchOptions.Regions), 0)
 	require.Greater(t, len(options.MongoDBOptions.Versions), 0)
 	require.Greater(t, len(options.PostgresSQLOptions.Versions), 0)
 	require.Greater(t, len(options.RedisOptions.Versions), 0)
 	require.Greater(t, len(options.MySQLOptions.Versions), 0)
 	require.Greater(t, len(options.KafkaOptions.Versions), 0)
+	require.Greater(t, len(options.OpensearchOptions.Versions), 0)
 	require.Greater(t, len(options.MongoDBOptions.Layouts), 0)
 	require.Greater(t, len(options.PostgresSQLOptions.Layouts), 0)
 	require.Greater(t, len(options.RedisOptions.Layouts), 0)
 	require.Greater(t, len(options.MySQLOptions.Layouts), 0)
 	require.Greater(t, len(options.KafkaOptions.Layouts), 0)
+	require.Greater(t, len(options.OpensearchOptions.Layouts), 0)
 }
 
 func TestDatabases_CreateDatabaseUserWithMySQLSettings(t *testing.T) {