def test_list_function(self):
args={"ddb_table": "test-creds"}
keypot_instance=Keypot(args)
result_list=keypot_instance.list_from_ddb()
self.assertIsInstance(result_list, list)
#based on previous tests, should have at least two values
self.assertTrue(len(result_list)>=2)
def test_decrypt_file(self):
args={"ddb_table":test_ddb_table,"parameter_key":test_ddb_name_file, "kms_key": test_kms_key}
keypot_instance=Keypot(args)
retreived_value=keypot_instance.do_decrypt()
try:
with open(test_private_key_name_output, "wb") as f:
f.write(retreived_value.encode("UTF-8"))
finally:
if f:
f.close()
#compare both files, make sure they came out exactly the same
self.assertTrue(filecmp.cmp(test_private_key_name,test_private_key_name_output))
def remove_all_ddb_entries(self):
#Cleans up any potential failed runs before exiting
args={"ddb_table":test_ddb_table}
setup_keypot=Keypot(args)
setup_keypot.parameter_key=test_ddb_name_text_var
print('Removing:' + str(test_ddb_name_text_var) + ' from ' + test_ddb_table )
setup_keypot.do_delete()
setup_keypot.parameter_key=test_ddb_name_file
print('Removing:' + str(test_ddb_name_file) + ' from ' + test_ddb_table )
setup_keypot.do_delete()
def test_encrypt_file_no_override_empty(self):
args={"ddb_table":test_ddb_table,"parameter_key":test_ddb_name_file,"parameter_file":test_private_key_name, "kms_key": test_kms_key}
keypot_instance=Keypot(args)
self.assertIsInstance(keypot_instance.do_encrypt(),str)
def test_encrypt_value_with_overwrite(self):
args={"ddb_table":test_ddb_table,"parameter_key":test_ddb_name_text_var,"parameter_value":test_ddb_name_text_val2, "overwrite": "TRUE", "kms_key": test_kms_key}
keypot_instance=Keypot(args)
self.assertIsInstance(keypot_instance.do_encrypt(),str)
def test_encrypt_value_override_false_override_failure(self):
args={"ddb_table":test_ddb_table,"parameter_key":test_ddb_name_text_var,"parameter_value":test_ddb_name_text_val1, "overwrite": "FALSE", "kms_key": test_kms_key}
keypot_instance=Keypot(args)
with self.assertRaises(KeypotError) as ke:
keypot_instance.do_encrypt()
def test_encrypt_value_no_override_empty(self):
args={"ddb_table":test_ddb_table,"parameter_key":test_ddb_name_text_var,"parameter_value":test_ddb_name_text_val1,"kms_key": test_kms_key}
keypot_instance=Keypot(args)
self.assertIsInstance(keypot_instance.do_encrypt(),str)
def test_delete_file_key(self):
args={"ddb_table":test_ddb_table,"parameter_key":test_ddb_name_file}
keypot_instance=Keypot(args)
self.assertIsInstance(keypot_instance.do_delete(),str)
def test_read_text_value2(self):
args={"ddb_table":test_ddb_table,"parameter_key":test_ddb_name_text_var, "kms_key": test_kms_key}
keypot_instance=Keypot(args)
retreived_value=keypot_instance.do_decrypt()
self.assertEqual(retreived_value,test_ddb_name_text_val2)
