Merge pull request #16 from imperva/matt/az-flexible

Support AZURE POSTGRESQL FLEXIBLE server type
imperva · Sep 25, 2024 · a02b887 · a02b887
2 parents 94d0aaf + 47c3c6d
commit a02b887
Show file tree

Hide file tree

Showing 2 changed files with 88 additions and 0 deletions.
diff --git a/dsfhub/client_data_source_test_data.go b/dsfhub/client_data_source_test_data.go
@@ -2277,6 +2277,76 @@ var dataSourceTestDataJson = `{
                 "gateway_id": "string"
             }
         },
+        "AZURE POSTGRESQL FLEXIBLE": {
+            "data": {
+                "assetData": {
+                    "admin_email": "string",
+                    "asset_display_name": "string",
+                    "asset_id": "string",
+                    "asset_source": "string",
+                    "audit_info": {
+                        "policy_template_name": "<template name>"
+                    },
+                    "audit_pull_enabled": true,
+                    "connections": [
+                        {
+                            "connectionData": {
+                                "amazon_secret": {
+                                    "field_mapping": {
+                                        "<local_field1>": "<remote_field1>",
+                                        "<local_field2>": "<remote_field2>"
+                                    },
+                                    "secret_asset_id": "<AWS_asset_id>",
+                                    "secret_name": "<secret_name>"
+                                },
+                                "autocommit": true,
+                                "ca_file": "string",
+                                "cert_file": "string",
+                                "driver": "string",
+                                "hashicorp_secret": {
+                                    "field_mapping": {
+                                        "<local_field1>": "<remote_field1>",
+                                        "<local_field2>": "<remote_field2>"
+                                    },
+                                    "path": "secret/",
+                                    "secret_asset_id": "<hashicorp_asset_id>",
+                                    "secret_name": "<secret_name>"
+                                },
+                                "key_file": "string",
+                                "odbc_connection_string": "string",
+                                "passphrase": "string",
+                                "password": "string",
+                                "ssl": true,
+                                "ssl_server_cert": "string",
+                                "username": "string"
+                            },
+                            "reason": "string"
+                        }
+                    ],
+                    "criticality": 1,
+                    "database_name": "string",
+                    "db_engine": "string",
+                    "enable_audit_management": true,
+                    "enable_audit_monitoring": true,
+                    "entitlement_enabled": true,
+                    "gateway_service": "string",
+                    "jsonar_uid": "string",
+                    "location": "string",
+                    "logs_destination_asset_id": "string",
+                    "managed_by": "string",
+                    "max_concurrent_conn": "string",
+                    "owned_by": "string",
+                    "parent_asset_id": "string",
+                    "sdm_enabled": true,
+                    "Server Host Name": "string",
+                    "Server Port": "string",
+                    "used_for": "Production",
+                    "version": 1.0
+                },
+                "asset_id": "string",
+                "gateway_id": "string"
+            }
+        },
         "AZURE SQL MANAGED INSTANCE": {
             "data": {
                 "assetData": {

diff --git a/dsfhub/resource_data_source_schema.go b/dsfhub/resource_data_source_schema.go
@@ -27,6 +27,7 @@ var ignoreDataSourceParamsByServerType = map[string]map[string]bool{
 	"AZURE MS SQL SERVER":               {"arn": true, "asset_display_name": true},
 	"AZURE MYSQL":                       {"arn": true, "asset_display_name": true},
 	"AZURE POSTGRESQL":                  {"arn": true, "asset_display_name": true},
+	"AZURE POSTGRESQL FLEXIBLE":         {"arn": true, "asset_display_name": true},
 	"AZURE SQL MANAGED INSTANCE":        {"arn": true, "asset_display_name": true},
 	"AZURE STORAGE ACCOUNT":             {"arn": true, "asset_display_name": true},
 }
@@ -619,6 +620,23 @@ var requiredDataSourceFieldsJson = `{
                 "location"
             ]
         },
+        "AZURE POSTGRESQL FLEXIBLE": {
+            "auth_mechanisms": {
+                "password": [
+                "reason",
+                "password",
+                "username"
+                ]
+            },
+            "required": [
+                "server_host_name",
+                "admin_email",
+                "asset_display_name",
+                "asset_id",
+                "location",
+                "gateway_id"
+            ]
+        },
         "AZURE SQL MANAGED INSTANCE": {
             "auth_mechanisms": {
                 "password": [