fix codebuild tests

Author: RitvikKapila

.gitignore

@@ -32,10 +32,7 @@ __pycache__
 # PyTest
 .pytest_cache
 # Ignore key materials generated by examples or tests
-user_public_key_file_name.pem
-user_private_key_file_name.pem
-my-encrypted-data.ct
-my-decrypted-data.dat
+test_keyrings/
 
 # PyCharm
 .idea/

examples/test/keyrings/test_i_file_streaming_example.py

@@ -1,6 +1,8 @@
 # Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 """Test suite for the file streaming keyring example."""
+import os
+
 import pytest
 
 from ...src.keyrings.file_streaming_example import encrypt_and_decrypt_with_keyring
@@ -10,9 +12,44 @@
 
 def test_encrypt_and_decrypt_with_keyring():
     """Test function for encrypt and decrypt for file streaming example using Raw AES keyring."""
-    plaintext_filename = "test_keyrings/my-secret-data.dat"
-    ciphertext_filename = 'test_keyrings/my-encrypted-data.ct'
-    new_plaintext_filename = 'test_keyrings/my-decrypted-data.dat'
+    test_keyrings_directory = 'test_keyrings'
+    if not os.path.exists(test_keyrings_directory):
+        os.makedirs(test_keyrings_directory)
+
+    plaintext_filename = test_keyrings_directory + '/my-secret-data.dat'
+
+    plaintext_data = '''Lorem ipsum dolor sit amet, consectetur adipiscing elit.
+Praesent non feugiat leo. Aenean iaculis tellus ut velit consectetur,
+quis convallis orci eleifend. Sed eu dictum sapien. Nulla facilisi. Suspendisse potenti.
+Proin vehicula vehicula maximus. Donec varius et elit vel rutrum. Nulla lacinia neque turpis
+quis consequat orci pharetra et. Etiam consequat ullamcorper mauris. Vivamus molestie mollis
+mauris a gravida. Curabitur sed bibendum nisl. Cras varius tortor non erat sodales, quis congu
+tellus laoreet. Etiam fermentum purus eu diam sagittis, vitae commodo est vehicula.
+Nulla feugiat viverra orci vel interdum. Quisque pulvinar elit eget nulla facilisis varius.
+Mauris at suscipit sem. Aliquam in purus ut velit fringilla volutpat id non mi.
+Curabitur quis nunc eleifend, ornare lectus non, fringilla quam. Nam maximus volutpat placerat.
+Nulla ullamcorper lorem velit, nec sagittis ex tristique posuere. Aliquam fringilla magna commod
+libero faucibus tempor. Vestibulum non ligula tincidunt, finibus sapien in, sollicitudin
+ex. Pellentesque congue laoreet mi in condimentum. Cras convallis nisi ac nunc tincidunt
+venenatis. Suspendisse urna elit, cursus eu lacus a, aliquet porttitor mi.
+Nulla vel congue nibh, sed condimentum dui. Ut ante ligula, blandit eu finibus nec,
+scelerisque quis eros. Maecenas gravida odio eget nibh dictum, dictum varius lacus interdum.
+Integer quis nulla vulputate, rhoncus diam vitae, mollis mauris. Sed ut porttitor dolor.
+Fusce ut justo a ex bibendum imperdiet nec sit amet magna. Sed ullamcorper luctus augue,
+tempor viverra elit interdum sed. Cras sit amet arcu eu turpis molestie sollicitudin.
+Curabitur fermentum varius nibh, ut aliquet nisi. Aliquam id tempus tellus.
+Nulla porttitor nulla at nibh interdum, quis sollicitudin erat egestas.
+Ut blandit mauris quis efficitur efficitur. Morbi neque sapien, posuere ut aliquam eget,
+aliquam at velit. Morbi sit amet rhoncus felis, et hendrerit sem. Nulla porta dictum ligula
+eget iaculis. Cras lacinia ligula quis risus ultrices, sed consectetur metus imperdiet.
+Nullam id enim vestibulum nibh ultricies auctor. Morbi neque lacus, faucibus vitae commodo quis,
+malesuada sed velit.'''
+
+    with open(plaintext_filename, "w", encoding="utf-8") as f:
+        f.write(plaintext_data)
+
+    ciphertext_filename = test_keyrings_directory + '/my-encrypted-data.ct'
+    new_plaintext_filename = test_keyrings_directory + '/my-decrypted-data.dat'
     encrypt_and_decrypt_with_keyring(plaintext_filename,
                                      ciphertext_filename,
                                      new_plaintext_filename)

examples/test/keyrings/test_i_raw_rsa_keyring_example.py

@@ -33,6 +33,8 @@ def test_encrypt_and_decrypt_with_keyring_with_user_defined_keys():
     user_private_key = user_private_key.decode('utf-8')
 
     test_keyrings_directory = 'test_keyrings'
+    if not os.path.exists(test_keyrings_directory):
+        os.makedirs(test_keyrings_directory)
 
     # Define the file names for the keys
     user_public_key_file_name = test_keyrings_directory + '/user_public_key_file_name.pem'