def test_creating_pbz2(self):
"""
Test paths of the created instance (PBZ2)
"""
s = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, _test=True)
s.put(1, name="one")
s.dump()
assert os.path.exists(os.path.expanduser(PATH_DUMPS) + DUMP_NAME + PBZ2)
s._cleanup()
def test_initialization_pkl(self):
"""
Test initialization procedure
Expected one backup file after dump procedure
"""
s = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, compressed=False, _test=True)
s.put(1, name="one")
s.dump()
s2 = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, compressed=False, _test=True)
one = len(s2.backup_list)
assert one == 1, f"Got something different: {one}"
s._cleanup()
def test_separation(self):
"""
Test separation procedure
"""
s = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, compressed=False, separations=1, _test=True)
s.put(1, name="one") # 1
s.put(2, name="two") # 2
s.put(3, name="three") # 3
s.put(4, name="four") # 4
s.dump()
s2 = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, compressed=False, _test=True)
length = len(s2.backup_list)
assert length == 4, f"Got something different: {length}"
s._cleanup()
def test_get_item_bpz2(self):
"""
Test get item procedure
"""
s = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, _test=True)
s.put(1, name="one")
s.put(2, name="two")
three = s.get("three")
assert three == False # "Should be False!"
s.put(3, name="three")
s.dump()
# here is new data in storer
three = s.get("three")
assert three == 3, f"got something different: [{three}]" # "Should be 3!"
s._cleanup()
def test_long_loop_loading(self):
"""
Get method in loop:
Checking for not overloading internal data dict
"""
s = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, compressed=False, separations=10000, _test= True)
cycles = 1000
for idx in range(cycles):
value = s.get(name=str(idx))
if not value: s.put(what=idx, name=str(idx))
s.dump()
s2 = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, compressed=False, separations=10000, _test= True)
for idx in range(cycles):
value = s2.get(name=str(idx))
assert value == idx, f"Got something different: {value} != {idx}"
s._cleanup()
def test_loading_separation_pbz2(self):
"""
Test separation procedure
"""
s = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, compressed=True, separations=1, _test=True)
s.put(1, name="one") # _
s.put(2, name="two") # 0
s.put(3, name="three") # 1
s.put(4, name="four") # 2
s.dump() # 3
s2 = Storer(dump_path=PATH_DUMPS, dump_name=DUMP_NAME, verbose=VERBOSE, compressed=True, _test=True)
one = s2.get(name="one")
assert 1 == one, f"Got something different: {one}"
two = s2.get(name="two")
assert 2 == two, f"Got something different: {two}"
three = s2.get(name="three")
assert 3 == three, f"Got something different: {three}"
four = s2.get(name="four")
assert 4 == four, f"Got something different: {four}"
s._cleanup()
