Skip to content

Commit

Permalink
Migrate servermanager to testProxy (Do not merge) (Azure#25481)
Browse files Browse the repository at this point in the history
* migrate tests

* Update dev_requirements.txt

* Update conftest.py

* update
  • Loading branch information
BigCat20196 authored Aug 8, 2022
1 parent 59c53cd commit 122f89a
Show file tree
Hide file tree
Showing 5 changed files with 63 additions and 222 deletions.
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
-e ../../../tools/azure-sdk-tools
-e ../../../tools/azure-devtools
azure-core
../../identity/azure-identity
54 changes: 54 additions & 0 deletions sdk/servermanager/azure-mgmt-servermanager/tests/conftest.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,54 @@
# --------------------------------------------------------------------------
#
# Copyright (c) Microsoft Corporation. All rights reserved.
#
# The MIT License (MIT)
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the ""Software""), to
# deal in the Software without restriction, including without limitation the
# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
# sell copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
# IN THE SOFTWARE.
#
# --------------------------------------------------------------------------
import os
import platform
import pytest
import sys

from dotenv import load_dotenv

from devtools_testutils import test_proxy, add_general_regex_sanitizer, add_body_key_sanitizer, add_header_regex_sanitizer

# Ignore async tests for Python < 3.5
collect_ignore_glob = []
if sys.version_info < (3, 5) or platform.python_implementation() == "PyPy":
collect_ignore_glob.append("*_async.py")

load_dotenv()

@pytest.fixture(scope="session", autouse=True)
def add_sanitizers(test_proxy):
subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000")
tenant_id = os.environ.get("AZURE_TENANT_ID", "00000000-0000-0000-0000-000000000000")
client_id = os.environ.get("AZURE_CLIENT_ID", "00000000-0000-0000-0000-000000000000")
client_secret = os.environ.get("AZURE_CLIENT_SECRET", "00000000-0000-0000-0000-000000000000")
add_general_regex_sanitizer(regex=subscription_id, value="00000000-0000-0000-0000-000000000000")
add_general_regex_sanitizer(regex=tenant_id, value="00000000-0000-0000-0000-000000000000")
add_general_regex_sanitizer(regex=client_id, value="00000000-0000-0000-0000-000000000000")
add_general_regex_sanitizer(regex=client_secret, value="00000000-0000-0000-0000-000000000000")
add_header_regex_sanitizer(key="Set-Cookie", value="[set-cookie;]")
add_header_regex_sanitizer(key="Cookie", value="cookie;")
add_body_key_sanitizer(json_path="$..access_token", value="access_token")
Original file line number Diff line number Diff line change
Expand Up @@ -8,17 +8,17 @@
import unittest

import azure.mgmt.servermanager
from devtools_testutils import AzureMgmtTestCase, ResourceGroupPreparer
from devtools_testutils import AzureMgmtRecordedTestCase, ResourceGroupPreparer, recorded_by_proxy

class MgmtServerManagerTest(AzureMgmtTestCase):
class TestMgmtServerManager(AzureMgmtRecordedTestCase):

def setUp(self):
super(MgmtServerManagerTest, self).setUp()
def setup_method(self, method):
self.client = self.create_mgmt_client(
azure.mgmt.servermanager.ServerManagement
)

@ResourceGroupPreparer()
@recorded_by_proxy
def test_servermanager_gateways(self, resource_group, location):
gateway_name = self.get_resource_name('pygateway')
region = 'centralus'
Expand All @@ -29,16 +29,16 @@ def test_servermanager_gateways(self, resource_group, location):
region
)
gateway = gateway_async.result()
self.assertEqual(gateway.name, gateway_name)
assert gateway.name == gateway_name

gateway = self.client.gateway.get(
resource_group.name,
gateway_name
)
self.assertEqual(gateway.name, gateway_name)
assert gateway.name == gateway_name

gateways = list(self.client.gateway.list())
self.assertEqual(len(gateways), 1)
assert len(gateways) == 1

#------------------------------------------------------------------------------
if __name__ == '__main__':
Expand Down

This file was deleted.

Loading

0 comments on commit 122f89a

Please sign in to comment.