From c44178f2f789e22b43a35c4091105999a079c571 Mon Sep 17 00:00:00 2001 From: mattsb42-aws Date: Fri, 13 Sep 2019 13:19:11 -0700 Subject: [PATCH] add tests for invalid values passed to generate_random --- tests/test_kms/test_kms.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/test_kms/test_kms.py b/tests/test_kms/test_kms.py index 31ba717e3..d9389094b 100644 --- a/tests/test_kms/test_kms.py +++ b/tests/test_kms/test_kms.py @@ -971,6 +971,21 @@ def test_generate_random(number_of_bytes): len(response["Plaintext"]).should.equal(number_of_bytes) +@parameterized(( + (2048, botocore.exceptions.ClientError), + (1025, botocore.exceptions.ClientError), + (0, botocore.exceptions.ParamValidationError), + (-1, botocore.exceptions.ParamValidationError), + (-1024, botocore.exceptions.ParamValidationError) +)) +@mock_kms +def test_generate_random_invalid_number_of_bytes(number_of_bytes, error_type): + client = boto3.client("kms", region_name="us-west-2") + + with assert_raises(error_type): + client.generate_random(NumberOfBytes=number_of_bytes) + + @mock_kms def test_enable_key_rotation_key_not_found(): client = boto3.client("kms", region_name="us-east-1")