1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
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
|
