Migrated unit/test/_providers_kms_master_key_provider.py from unittest to pytest

Author: Andres Sanchez

test/unit/test_providers_kms_master_key_provider.py

@@ -11,8 +11,6 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 """Unit test suite from aws_encryption_sdk.key_providers.kms.KMSMasterKeyProvider"""
-import unittest
-
 import botocore.client
 import pytest
 import six
@@ -32,8 +30,9 @@ def test_init_with_regionless_key_ids_and_region_names():
     assert provider.master_key("alias/key_1").config.client.meta.region_name == region_names[0]
 
 
-class TestKMSMasterKeyProvider(unittest.TestCase):
-    def setUp(self):
+class TestKMSMasterKeyProvider(object):
+    @pytest.fixture(autouse=True)
+    def apply_fixtures(self):
         self.mock_botocore_session_patcher = patch("aws_encryption_sdk.key_providers.kms.botocore.session.Session")
         self.mock_botocore_session = self.mock_botocore_session_patcher.start()
         self.mock_boto3_session_patcher = patch("aws_encryption_sdk.key_providers.kms.boto3.session.Session")
@@ -130,10 +129,9 @@ def test_client_no_region_name_with_default(self, mock_add_client):
 
     def test_client_no_region_name_without_default(self):
         test = KMSMasterKeyProvider()
-        with six.assertRaisesRegex(
-            self, UnknownRegionError, "No default region found and no region determinable from key id: *"
-        ):
+        with pytest.raises(UnknownRegionError) as excinfo:
             test._client("")
+        excinfo.match("No default region found and no region determinable from key id: *")
 
     @patch("aws_encryption_sdk.key_providers.kms.KMSMasterKeyProvider._client")
     def test_new_master_key(self, mock_client):