Skip to content

Commit ee1cb69

Browse files
ansanpermattsb42-aws
authored andcommitted
Migrate "test/unit/test_serialize.py" from unittest to pytest (#125)
* Migrating unit/test_deserialize.py from unittest from pytest * Migrate unit/test_provides_kms_master_key.py from unittest to pytest * Migrated unit/test/_providers_kms_master_key_provider.py from unittest to pytest * unit/test_providers_raw_master_key.py * Migrate unit/test_providers_raw_master_key_provider.py from unittest from pytest * Migrate unit/test_serialize.py from unittest to pytest * Removed unit tests not corresponding to this branch * Added yield and tearDown to apply_fixtures function
1 parent d6d1f87 commit ee1cb69

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

test/unit/test_serialize.py

+5-6
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,6 @@
1111
# ANY KIND, either express or implied. See the License for the specific
1212
# language governing permissions and limitations under the License.
1313
"""Unit test suite for aws_encryption_sdk.internal.formatting.serialize"""
14-
import unittest
15-
1614
import pytest
1715
from mock import MagicMock, patch, sentinel
1816

@@ -51,8 +49,9 @@ def test_serialize_frame_invalid_sequence_number(sequence_number, error_message)
5149
excinfo.match(error_message)
5250

5351

54-
class TestSerialize(unittest.TestCase):
55-
def setUp(self):
52+
class TestSerialize(object):
53+
@pytest.fixture(autouse=True)
54+
def apply_fixtures(self):
5655
self.mock_algorithm = MagicMock()
5756
self.mock_algorithm.encryption_algorithm.block_size = VALUES["block_size"]
5857
self.mock_algorithm.algorithm_id = VALUES["algorithm_id"]
@@ -84,8 +83,8 @@ def setUp(self):
8483
self.mock_signer = MagicMock()
8584
self.mock_signer.update.return_value = None
8685
self.mock_signer.finalize.return_value = VALUES["signature"]
87-
88-
def tearDown(self):
86+
yield
87+
# Run tearDown
8988
self.mock_serialize_acc_patcher.stop()
9089
self.mock_encrypt_patcher.stop()
9190
self.mock_valid_frame_length_patcher.stop()

0 commit comments

Comments
 (0)