def test_key_store(self, tempdir):
store_path = os.path.join(tempdir, "test_store")
store = KeyStore(os.path.join(tempdir, "test_store"))
assert repr(store) == "KeyStore object, file: {}".format(store_path)
key = faker.ed25519_key()
store.add(key)
assert key == store.get_key(key.user_id, key.device_id)
def test_key_store_check_invalid(self, tempdir):
store_path = os.path.join(tempdir, "test_store")
store = KeyStore(os.path.join(tempdir, "test_store"))
key = faker.ed25519_key()
store.add(key)
fake_key = copy.copy(key)
fake_key.key = "FAKE_KEY"
assert fake_key not in store
assert key in store
def test_key_store_add_many(self, tempdir):
store_path = os.path.join(tempdir, "test_store")
store = KeyStore(os.path.join(tempdir, "test_store"))
keys = [
faker.ed25519_key(),
faker.ed25519_key(),
faker.ed25519_key(),
faker.ed25519_key(),
]
store.add_many(keys)
store2 = KeyStore(os.path.join(tempdir, "test_store"))
for key in keys:
assert key in store2
def test_key_store_add_invalid(self, tempdir):
store_path = os.path.join(tempdir, "test_store")
store = KeyStore(os.path.join(tempdir, "test_store"))
key = faker.ed25519_key()
store.add(key)
fake_key = copy.copy(key)
fake_key.key = "FAKE_KEY"
with pytest.raises(OlmTrustError):
store.add(fake_key)