def test_encode_decode(self):
n_keys = 8
n_passwds = 8
LEN = 12
for keys in range(n_keys):
length = int(random.random() * LEN)
key = os.urandom(LEN - length).encode('hex')
padded_key = transform_passwd(key)
for passwds in range(n_passwds):
passwd = os.urandom(length).encode('hex')
original = make_gsm_dict(passwd)
# encode
encrypted = encryptData(padded_key, pickle.dumps(original))
# decode
pickledobj = decryptData(padded_key, encrypted)
decrypted = pickle.load(StringIO(pickledobj))
self.assertEqual(decrypted, original)
def update(self, uuid, conn_id, secrets, update=True):
path = os.path.join(self.secrets_path, uuid)
with open(path, "w") as f:
data = encryptData(self.key, pickle.dumps(secrets))
f.write(data)