4 from pyoram.util.virtual_heap import \
6 from pyoram.encrypted_storage.encrypted_heap_storage import \
11 # get a unique filename in the current directory
13 fid, tmpname = tempfile.mkstemp(dir=os.getcwd())
16 print("Storage Name: %s" % (tmpname))
19 header_data = b'a message'
24 initialize = lambda i: \
25 bytes(bytearray([i] * block_size * blocks_per_bucket))
26 vheap = SizedVirtualHeap(
29 blocks_per_bucket=blocks_per_bucket)
31 with EncryptedHeapStorage.setup(
36 header_data=header_data,
38 blocks_per_bucket=blocks_per_bucket,
39 initialize=initialize) as f:
40 assert tmpname == f.storage_name
41 assert f.header_data == header_data
42 print(f.read_path(vheap.random_bucket()))
44 assert os.path.exists(tmpname)
46 with EncryptedHeapStorage(tmpname, key=key) as f:
47 assert tmpname == f.storage_name
48 assert f.header_data == header_data
49 print(f.read_path(vheap.random_bucket()))
56 if __name__ == "__main__":
57 main() # pragma: no cover