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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import inspect
import os.path
import zlib
from .preparers import AbstractPreparer
class HttpStatusCode(object):
OK = 200
Created = 201
Accepted = 202
NoContent = 204
NotFound = 404
def get_resource_name(name_prefix, identifier):
# Append a suffix to the name, based on the fully qualified test name
# We use a checksum of the test name so that each test gets different
# resource names, but each test will get the same name on repeat runs,
# which is needed for playback.
# Most resource names have a length limit, so we use a crc32
checksum = zlib.adler32(identifier) & 0xFFFFFFFF
name = "{}{}".format(name_prefix, hex(checksum)[2:]).rstrip("L")
if name.endswith("L"):
name = name[:-1]
return name
def get_qualified_method_name(obj, method_name):
# example of qualified test name:
# test_mgmt_network.test_public_ip_addresses
_, filename = os.path.split(inspect.getsourcefile(type(obj)))
module_name, _ = os.path.splitext(filename)
return "{0}.{1}".format(module_name, method_name)
class AzureMgmtPreparer(AbstractPreparer):
def __init__(
self,
name_prefix,
random_name_length,
disable_recording=True,
playback_fake_resource=None,
client_kwargs=None,
random_name_enabled=False,
):
super(AzureMgmtPreparer, self).__init__(name_prefix, random_name_length, disable_recording=disable_recording)
self.client = None
self.resource = playback_fake_resource
self.client_kwargs = client_kwargs or {}
self.random_name_enabled = random_name_enabled
@property
def is_live(self):
return self.test_class_instance.is_live
def create_random_name(self):
if self.random_name_enabled:
return super(AzureMgmtPreparer, self).create_random_name()
return self.test_class_instance.get_preparer_resource_name(self.name_prefix)
@property
def moniker(self):
"""Override moniker generation for backwards compatibility.
AbstractPreparer preparers, by default, generate "monikers" which replace
resource names in request URIs by tacking on a resource count to
name_prefix. By contrast, SDK tests used the fully qualified (module + method)
test name and the hashing process in get_resource_name.
This property override implements the SDK test name generation so that
the URIs don't change and tests don't need to be re-recorded.
"""
if not self.resource_moniker:
self.resource_moniker = self.random_name
return self.resource_moniker
def create_mgmt_client(self, client_class, **kwargs):
return self.test_class_instance.create_mgmt_client(client_class, **kwargs)
|
