[AppConfig] Refactor tests (Azure#25748)

Author: Yalin Li

sdk/appconfiguration/azure-appconfiguration/tests/async_preparers.py

@@ -0,0 +1,34 @@
+# coding: utf-8
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+from preparers import AppConfigPreparer, trim_kwargs_from_test_function
+
+
+def app_config_decorator_async(func, **kwargs):
+    @AppConfigPreparer()
+    async def wrapper(*args, **kwargs):
+        appconfiguration_connection_string = kwargs.pop("appconfiguration_connection_string")
+        kwargs['appconfiguration_connection_string'] = appconfiguration_connection_string
+        
+        trimmed_kwargs = {k:v for k, v in kwargs.items()}
+        trim_kwargs_from_test_function(func, trimmed_kwargs)
+
+        await func(*args, **trimmed_kwargs)
+
+    return wrapper
+
+def app_config_aad_decorator_async(func, **kwargs):
+    @AppConfigPreparer()
+    async def wrapper(*args, **kwargs):
+        appconfiguration_endpoint_string = kwargs.pop("appconfiguration_endpoint_string")
+        kwargs['appconfiguration_endpoint_string'] = appconfiguration_endpoint_string
+
+        trimmed_kwargs = {k:v for k, v in kwargs.items()}
+        trim_kwargs_from_test_function(func, trimmed_kwargs)
+
+        await func(*args, **trimmed_kwargs)
+
+    return wrapper

sdk/appconfiguration/azure-appconfiguration/tests/async_proxy.py

@@ -0,0 +1,50 @@
+# coding: utf-8
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+from azure.appconfiguration.aio import (
+    AzureAppConfigurationClient,
+)
+from testcase import AppConfigTestCase
+
+
+class AsyncAppConfigTestCase(AppConfigTestCase):
+    def create_aad_client(self, appconfiguration_endpoint_string):
+        cred = self.get_credential(AzureAppConfigurationClient, is_async=True)
+        return AzureAppConfigurationClient(appconfiguration_endpoint_string, cred)
+    
+    def create_client(self, appconfiguration_connection_string):
+        return AzureAppConfigurationClient.from_connection_string(appconfiguration_connection_string)
+
+    async def add_for_test(self, client, config_setting):
+        exist_list = await self.convert_to_list(client.list_configuration_settings(
+            key_filter=config_setting.key, label_filter=config_setting.label
+        ))
+        exist = bool(exist_list)
+        if exist:
+            await client.delete_configuration_setting(key=config_setting.key, label=config_setting.label)
+        await client.add_configuration_setting(config_setting)
+        
+    async def convert_to_list(self, config_settings): # type: (AsyncItemPaged) -> list
+        list = []
+        async for item in config_settings:
+            list.append(item)
+        return list
+    
+    async def set_up(self, appconfiguration_string, is_aad=False):
+        if is_aad:
+            self.client = self.create_aad_client(appconfiguration_string)
+        else:
+            self.client = self.create_client(appconfiguration_string)
+        await self.add_for_test(self.client, self.create_config_setting())
+        await self.add_for_test(self.client, self.create_config_setting_no_label())
+    
+    async def tear_down(self):
+        if self.client is not None:
+            config_settings = self.client.list_configuration_settings()
+            async for config_setting in config_settings:
+                await self.client.delete_configuration_setting(key=config_setting.key, label=config_setting.label)
+        else:
+            raise ValueError("Client is None!")

sdk/appconfiguration/azure-appconfiguration/tests/async_wrapper.py

@@ -0,0 +1,57 @@
+
+# coding: utf-8
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+from devtools_testutils import PowerShellPreparer
+import functools
+import inspect
+
+AppConfigPreparer = functools.partial(
+    PowerShellPreparer,
+    'appconfiguration',
+    appconfiguration_connection_string="Endpoint=https://fake_app_config.azconfig-test.io;Id=0-l4-s0:h5htBaY5Z1LwFz50bIQv;Secret=lamefakesecretlamefakesecretlamefakesecrett=",
+    appconfiguration_endpoint_string="https://fake_app_config.azconfig-test.io",
+)
+
+
+def trim_kwargs_from_test_function(fn, kwargs):
+    # the next function is the actual test function. the kwargs need to be trimmed so
+    # that parameters which are not required will not be passed to it.
+    if not getattr(fn, '__is_preparer', False):
+        try:
+            args, _, kw, _, _, _, _ = inspect.getfullargspec(fn)
+        except AttributeError:
+            args, _, kw, _ = inspect.getargspec(fn) # pylint: disable=deprecated-method
+        if kw is None:
+            args = set(args)
+            for key in [k for k in kwargs if k not in args]:
+                del kwargs[key]
+
+def app_config_decorator(func, **kwargs):
+    @AppConfigPreparer()
+    def wrapper(*args, **kwargs):
+        appconfiguration_connection_string = kwargs.pop("appconfiguration_connection_string")
+        kwargs['appconfiguration_connection_string'] = appconfiguration_connection_string
+
+        trimmed_kwargs = {k:v for k, v in kwargs.items()}
+        trim_kwargs_from_test_function(func, trimmed_kwargs)
+
+        func(*args, **trimmed_kwargs)
+
+    return wrapper
+
+def app_config_aad_decorator(func, **kwargs):
+    @AppConfigPreparer()
+    def wrapper(*args, **kwargs):
+        appconfiguration_endpoint_string = kwargs.pop("appconfiguration_endpoint_string")
+        kwargs['appconfiguration_endpoint_string'] = appconfiguration_endpoint_string
+
+        trimmed_kwargs = {k:v for k, v in kwargs.items()}
+        trim_kwargs_from_test_function(func, trimmed_kwargs)
+
+        func(*args, **trimmed_kwargs)
+
+    return wrapper