adding pytest tests

Author: RitvikKapila

performance_tests/test/keyrings/test_aws_kms_keyring.py

@@ -5,6 +5,8 @@
 import time
 
 import click
+import click.testing
+import pytest
 from tqdm import tqdm
 
 from aws_encryption_sdk_performance_tests.keyrings.aws_kms_keyring import (
@@ -164,5 +166,35 @@ def decrypt(
                                                     decrypt_kms_keyring])
 
 
+@pytest.fixture
+def runner():
+    """Click runner"""
+    return click.testing.CliRunner()
+
+
+def test_create(runner):
+    """Test the create_keyring function"""
+    result = runner.invoke(create_kms_keyring.commands['create'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_create_given_kms_client(runner):
+    """Test the create_keyring_given_kms_client function"""
+    result = runner.invoke(create_kms_keyring_given_kms_client.commands['create-given-kms-client'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_encrypt(runner):
+    """Test the encrypt_using_keyring function"""
+    result = runner.invoke(encrypt_kms_keyring.commands['encrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_decrypt(runner):
+    """Test the decrypt_using_keyring function"""
+    result = runner.invoke(decrypt_kms_keyring.commands['decrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
 if __name__ == "__main__":
     kms_keyring_test()

performance_tests/test/keyrings/test_raw_aes_keyring.py

@@ -5,6 +5,8 @@
 import time
 
 import click
+import click.testing
+import pytest
 from tqdm import tqdm
 
 from aws_encryption_sdk_performance_tests.keyrings.raw_aes_keyring import (
@@ -120,5 +122,29 @@ def decrypt(
                                                         decrypt_raw_aes_keyring])
 
 
+@pytest.fixture
+def runner():
+    """Click runner"""
+    return click.testing.CliRunner()
+
+
+def test_create(runner):
+    """Test the create_keyring function"""
+    result = runner.invoke(create_raw_aes_keyring.commands['create'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_encrypt(runner):
+    """Test the encrypt_using_keyring function"""
+    result = runner.invoke(encrypt_raw_aes_keyring.commands['encrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_decrypt(runner):
+    """Test the decrypt_using_keyring function"""
+    result = runner.invoke(decrypt_raw_aes_keyring.commands['decrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
 if __name__ == "__main__":
     raw_aes_keyring_test()

performance_tests/test/keyrings/test_raw_rsa_keyring.py

@@ -5,6 +5,8 @@
 import time
 
 import click
+import click.testing
+import pytest
 from tqdm import tqdm
 
 from aws_encryption_sdk_performance_tests.keyrings.raw_rsa_keyring import (
@@ -127,5 +129,29 @@ def decrypt(
                                                         decrypt_raw_rsa_keyring])
 
 
+@pytest.fixture
+def runner():
+    """Click runner"""
+    return click.testing.CliRunner()
+
+
+def test_create(runner):
+    """Test the create_keyring function"""
+    result = runner.invoke(create_raw_rsa_keyring.commands['create'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_encrypt(runner):
+    """Test the encrypt_using_keyring function"""
+    result = runner.invoke(encrypt_raw_rsa_keyring.commands['encrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_decrypt(runner):
+    """Test the decrypt_using_keyring function"""
+    result = runner.invoke(decrypt_raw_rsa_keyring.commands['decrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
 if __name__ == "__main__":
     raw_rsa_keyring_test()

performance_tests/test/master_key_providers/test_aws_kms_master_key_provider.py

@@ -5,6 +5,8 @@
 import time
 
 import click
+import click.testing
+import pytest
 from tqdm import tqdm
 
 from aws_encryption_sdk_performance_tests.master_key_providers.aws_kms_master_key_provider import (
@@ -129,5 +131,29 @@ def decrypt(
                                                          decrypt_kms_key_provider])
 
 
+@pytest.fixture
+def runner():
+    """Click runner"""
+    return click.testing.CliRunner()
+
+
+def test_create(runner):
+    """Test the create_key_provider function"""
+    result = runner.invoke(create_kms_key_provider.commands['create'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_encrypt(runner):
+    """Test the encrypt_using_key_provider function"""
+    result = runner.invoke(encrypt_kms_key_provider.commands['encrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_decrypt(runner):
+    """Test the decrypt_using_key_provider function"""
+    result = runner.invoke(decrypt_kms_key_provider.commands['decrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
 if __name__ == "__main__":
     kms_key_provider_test()

performance_tests/test/master_key_providers/test_raw_aes_master_key_provider.py

@@ -5,6 +5,8 @@
 import time
 
 import click
+import click.testing
+import pytest
 from tqdm import tqdm
 
 from aws_encryption_sdk_performance_tests.master_key_providers.raw_aes_master_key_provider import (
@@ -120,5 +122,29 @@ def decrypt(
                                                              decrypt_raw_aes_key_provider])
 
 
+@pytest.fixture
+def runner():
+    """Click runner"""
+    return click.testing.CliRunner()
+
+
+def test_create(runner):
+    """Test the create_key_provider function"""
+    result = runner.invoke(create_raw_aes_key_provider.commands['create'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_encrypt(runner):
+    """Test the encrypt_using_key_provider function"""
+    result = runner.invoke(encrypt_raw_aes_key_provider.commands['encrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_decrypt(runner):
+    """Test the decrypt_using_key_provider function"""
+    result = runner.invoke(decrypt_raw_aes_key_provider.commands['decrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
 if __name__ == "__main__":
     raw_aes_key_provider_test()

performance_tests/test/master_key_providers/test_raw_rsa_master_key_provider.py

@@ -5,6 +5,8 @@
 import time
 
 import click
+import click.testing
+import pytest
 from tqdm import tqdm
 
 from aws_encryption_sdk_performance_tests.master_key_providers.raw_rsa_master_key_provider import (
@@ -120,5 +122,29 @@ def decrypt(
                                                              decrypt_raw_rsa_key_provider])
 
 
+@pytest.fixture
+def runner():
+    """Click runner"""
+    return click.testing.CliRunner()
+
+
+def test_create(runner):
+    """Test the create_key_provider function"""
+    result = runner.invoke(create_raw_rsa_key_provider.commands['create'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_encrypt(runner):
+    """Test the encrypt_using_key_provider function"""
+    result = runner.invoke(encrypt_raw_rsa_key_provider.commands['encrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
+def test_decrypt(runner):
+    """Test the decrypt_using_key_provider function"""
+    result = runner.invoke(decrypt_raw_rsa_key_provider.commands['decrypt'], ['--n_iters', 1])
+    assert result.exit_code == 0
+
+
 if __name__ == "__main__":
     raw_rsa_key_provider_test()