Add rdbms flexible-server tests (Azure#14703)

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/mysql_flexibleservers/operations/_operations.py

@@ -35,7 +35,7 @@ def __init__(self, client, config, serializer, deserializer):
         self._client = client
         self._serialize = serializer
         self._deserialize = deserializer
-        self.api_version = "2020-07-01-privatepreview"
+        self.api_version = "2017-12-01"
 
         self.config = config
 

sdk/rdbms/azure-mgmt-rdbms/azure/mgmt/rdbms/postgresql_flexibleservers/operations/_operations.py

@@ -35,7 +35,7 @@ def __init__(self, client, config, serializer, deserializer):
         self._client = client
         self._serialize = serializer
         self._deserialize = deserializer
-        self.api_version = "2020-02-14-preview"
+        self.api_version = "2017-12-01"
 
         self.config = config
 

sdk/rdbms/azure-mgmt-rdbms/tests/flexible_server_test_common.py

@@ -0,0 +1,88 @@
+def flexible_server_firewall_rule_mgmt_test(instance, resource_group_name, server_name):
+    firewall_rule_name = server_name + "-testfirewallrule"
+    start_ip_address = "0.0.0.0"
+    end_ip_address = "255.255.255.255"
+    new_start_ip_address = "12.12.12.12"
+    new_end_ip_address = "233.233.233.233"
+
+    result = instance.mgmt_client.firewall_rules.create_or_update(resource_group_name, server_name, firewall_rule_name, start_ip_address, end_ip_address)
+    result = result.result()
+    instance.assertEqual(result.start_ip_address, start_ip_address)
+    instance.assertEqual(result.end_ip_address, end_ip_address)
+
+    result = instance.mgmt_client.firewall_rules.create_or_update(resource_group_name, server_name, firewall_rule_name, new_start_ip_address, end_ip_address)
+    result = result.result()
+    instance.assertEqual(result.start_ip_address, new_start_ip_address)
+
+    result = instance.mgmt_client.firewall_rules.create_or_update(resource_group_name, server_name, firewall_rule_name, new_start_ip_address, new_end_ip_address)
+    result = result.result()
+    instance.assertEqual(result.end_ip_address, new_end_ip_address)
+
+    result = instance.mgmt_client.firewall_rules.list_by_server(resource_group_name, server_name)
+    instance.assertIsNotNone(result)
+
+    result = instance.mgmt_client.firewall_rules.get(resource_group_name, server_name, firewall_rule_name)
+
+    result = instance.mgmt_client.firewall_rules.delete(resource_group_name, server_name, firewall_rule_name)
+    result = result.result()
+
+
+def flexible_server_configuration_mgmt_test(instance, resource_group_name, server_name, config_key_name, default_value, user_value):
+
+    result = instance.mgmt_client.configurations.list_by_server(resource_group_name, server_name)
+    instance.assertIsNotNone(result)
+
+    result = instance.mgmt_client.configurations.get(resource_group_name, server_name, config_key_name)
+    instance.assertTrue(result.value, default_value)
+    instance.assertTrue(result.source, "system-default")
+
+    result = instance.mgmt_client.configurations.update(resource_group_name, server_name, config_key_name, value=user_value, source='user-override')
+    result = result.result()
+    instance.assertTrue(result.value, user_value)
+    instance.assertTrue(result.source, "user-override")
+
+
+def flexible_server_database_mgmt_test(instance, resource_group_name, server_name):
+    database_name = "testdatabase"
+    result = instance.mgmt_client.databases.create_or_update(resource_group_name, server_name, database_name, charset='utf8', collation='utf8_general_ci')
+    result = result.result()
+    instance.assertEqual(result.name, database_name)
+    instance.assertEqual(result.charset, 'utf8')
+    instance.assertEqual(result.collation, 'utf8_general_ci')
+
+    result = instance.mgmt_client.databases.get(resource_group_name, server_name, database_name)
+
+    result = instance.mgmt_client.databases.list_by_server(resource_group_name, server_name)
+    instance.assertIsNotNone(result)
+
+    result = instance.mgmt_client.databases.delete(resource_group_name, server_name, database_name)
+    result = result.result()
+
+def flexible_server_serverkey_mgmt_test(instance, resource_group_name, server_name, vault, key_client):
+    parameters = azure.mgmt.rdbms.mysql_flexibleservers.models.Server(
+        location=instance.location,
+        identity = azure.mgmt.rdbms.mysql_flexibleservers.models.Identity(
+        type=azure.mgmt.rdbms.mysql_flexibleservers.models.ResourceIdentityType.system_assigned.value))
+    result = instance.mgmt_client.servers.update(resource_group_name, server_name, parameters)
+    result = result.result()
+    server_identity = result.identity.principal_id
+
+    key = key_client.create_key(key_name, "RSA")
+    kid = key.kid
+    # self.cmd('keyvault set-policy -g {} -n {} --object-id {} --key-permissions wrapKey unwrapKey get list'
+    #              .format(resource_group, vault_name, server_identity))
+
+    key_name = instance._get_server_key_name_from_uri(kid)
+
+    result = instance.mgmt_client.server_keys.create_or_update(resource_group_name, server_name, key_name, uri=kid)
+    result = result.result()
+    print(vars(result))
+
+    result = instance.mgmt_client.server_keys.get(resource_group_name, server_name, key_name)
+    print(vars(result))
+
+    result = instance.mgmt_client.server_keys.list_by_server(resource_group_name, server_name)
+    print(list(result))
+
+    result = instance.mgmt_client.server_keys.delete(resource_group_name, server_name, key_name)
+    result = result.result()