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 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)