Migrate "test/unit/test_providers_kms_master_key.py" from unittest to pytest (#121)

* Migrating unit/test_deserialize.py from unittest from pytest

* Migrate unit/test_provides_kms_master_key.py from unittest to pytest

* Removed unit test not corresponding to this branch

* Added yield and tearDown to apply_fixtures function

* Lintering issues, fixed with tox -re autoformat

Author: ansanper

test/unit/test_providers_kms_master_key.py

@@ -11,11 +11,8 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 """Unit test suite for aws_encryption_sdk.key_providers.kms.KMSMasterKey"""
-import unittest
-
 import botocore.client
 import pytest
-import six
 from botocore.exceptions import ClientError
 from mock import MagicMock, patch, sentinel
 
@@ -30,8 +27,9 @@
 pytestmark = [pytest.mark.unit, pytest.mark.local]
 
 
-class TestKMSMasterKey(unittest.TestCase):
-    def setUp(self):
+class TestKMSMasterKey(object):
+    @pytest.fixture(autouse=True)
+    def apply_fixture(self):
         self.mock_client = MagicMock()
         self.mock_client.__class__ = botocore.client.BaseClient
         self.mock_client.generate_data_key.return_value = {
@@ -59,13 +57,13 @@ def setUp(self):
             key_id=VALUES["arn"], client=self.mock_client, grant_tokens=self.mock_grant_tokens
         )
         self.mock_kms_mkc_3 = KMSMasterKeyConfig(key_id="ex_key_info", client=self.mock_client)
+        yield
+        # Run tearDown
+        self.mock_data_key_len_check_patcher.stop()
 
     def test_parent(self):
         assert issubclass(KMSMasterKey, MasterKey)
 
-    def tearDown(self):
-        self.mock_data_key_len_check_patcher.stop()
-
     def test_config_bare(self):
         test = KMSMasterKeyConfig(key_id=VALUES["arn"], client=self.mock_client)
         assert test.client is self.mock_client
@@ -111,14 +109,16 @@ def test_generate_data_key_with_grant_tokens(self):
     def test_generate_data_key_unsuccessful_clienterror(self):
         self.mock_client.generate_data_key.side_effect = ClientError({"Error": {}}, "This is an error!")
         test = KMSMasterKey(config=self.mock_kms_mkc_3)
-        with six.assertRaisesRegex(self, GenerateKeyError, "Master Key .* unable to generate data key"):
+        with pytest.raises(GenerateKeyError) as excinfo:
             test._generate_data_key(self.mock_algorithm)
+        excinfo.match("Master Key .* unable to generate data key")
 
     def test_generate_data_key_unsuccessful_keyerror(self):
         self.mock_client.generate_data_key.side_effect = KeyError
         test = KMSMasterKey(config=self.mock_kms_mkc_3)
-        with six.assertRaisesRegex(self, GenerateKeyError, "Master Key .* unable to generate data key"):
+        with pytest.raises(GenerateKeyError) as excinfo:
             test._generate_data_key(self.mock_algorithm)
+        excinfo.match("Master Key .* unable to generate data key")
 
     def test_encrypt_data_key(self):
         test = KMSMasterKey(config=self.mock_kms_mkc_3)
@@ -146,14 +146,16 @@ def test_encrypt_data_key_with_grant_tokens(self):
     def test_encrypt_data_key_unsuccessful_clienterror(self):
         self.mock_client.encrypt.side_effect = ClientError({"Error": {}}, "This is an error!")
         test = KMSMasterKey(config=self.mock_kms_mkc_3)
-        with six.assertRaisesRegex(self, EncryptKeyError, "Master Key .* unable to encrypt data key"):
+        with pytest.raises(EncryptKeyError) as excinfo:
             test._encrypt_data_key(self.mock_data_key, self.mock_algorithm)
+        excinfo.match("Master Key .* unable to encrypt data key")
 
     def test_encrypt_data_key_unsuccessful_keyerror(self):
         self.mock_client.encrypt.side_effect = KeyError
         test = KMSMasterKey(config=self.mock_kms_mkc_3)
-        with six.assertRaisesRegex(self, EncryptKeyError, "Master Key .* unable to encrypt data key"):
+        with pytest.raises(EncryptKeyError) as excinfo:
             test._encrypt_data_key(self.mock_data_key, self.mock_algorithm)
+        excinfo.match("Master Key .* unable to encrypt data key")
 
     def test_decrypt_data_key(self):
         test = KMSMasterKey(config=self.mock_kms_mkc_1)
@@ -186,11 +188,13 @@ def test_decrypt_data_key_with_grant_tokens(self):
     def test_decrypt_data_key_unsuccessful_clienterror(self):
         self.mock_client.decrypt.side_effect = ClientError({"Error": {}}, "This is an error!")
         test = KMSMasterKey(config=self.mock_kms_mkc_3)
-        with six.assertRaisesRegex(self, DecryptKeyError, "Master Key .* unable to decrypt data key"):
+        with pytest.raises(DecryptKeyError) as excinfo:
             test._decrypt_data_key(encrypted_data_key=self.mock_encrypted_data_key, algorithm=sentinel.algorithm)
+        excinfo.match("Master Key .* unable to decrypt data key")
 
     def test_decrypt_data_key_unsuccessful_keyerror(self):
         self.mock_client.decrypt.side_effect = KeyError
         test = KMSMasterKey(config=self.mock_kms_mkc_3)
-        with six.assertRaisesRegex(self, DecryptKeyError, "Master Key .* unable to decrypt data key"):
+        with pytest.raises(DecryptKeyError) as excinfo:
             test._decrypt_data_key(encrypted_data_key=self.mock_encrypted_data_key, algorithm=sentinel.algorithm)
+        excinfo.match("Master Key .* unable to decrypt data key")