def test_add_to_delayed_write(self): """Test if a cube is in delayed write""" csdb = CacheStateDB(self.config_data) lookup_key = "1&2&3" resolution = 4 time_sample = 5 morton = 66 write_cuboid_key1 = "WRITE-CUBOID&{}&{}&{}&{}&daadsfjk".format( lookup_key, resolution, time_sample, morton) write_cuboid_key2 = "WRITE-CUBOID&{}&{}&{}&{}&fghfghjg".format( lookup_key, resolution, time_sample, morton) keys = csdb.get_all_delayed_write_keys() assert not keys csdb.add_to_delayed_write(write_cuboid_key1, lookup_key, resolution, morton, time_sample, "{dummy resource str}") csdb.add_to_delayed_write(write_cuboid_key2, lookup_key, resolution, morton, time_sample, "{dummy resource str}") keys = csdb.get_all_delayed_write_keys() assert len(keys) == 1 assert keys[0] == "DELAYED-WRITE&{}&{}&{}&{}".format( lookup_key, resolution, time_sample, morton) write_keys = csdb.get_delayed_writes(keys[0]) assert len(write_keys) == 2 assert write_keys[0] == write_cuboid_key1 assert write_keys[1] == write_cuboid_key2
def test_get_all_delayed_write_cuboid_keys(self): """Test getting all delayed write cuboid keys""" csdb = CacheStateDB(self.config_data) lookup_key = "1&2&3" resolution = 4 time_sample = 5 morton = 234 write_cuboid_key1 = "WRITE-CUBOID&{}&{}&{}{}&&daadsfjk".format(lookup_key, resolution, time_sample, morton) write_cuboid_key2 = "WRITE-CUBOID&{}&{}&{}&{}&fghfghjg".format(lookup_key, resolution, time_sample, morton) write_cuboid_key3 = "WRITE-CUBOID&{}&{}&{}&{}&aaauihjg".format(lookup_key, resolution, time_sample, morton) delayed_write_key = "DELAYED-WRITE&{}&{}&{}&{}".format(lookup_key, resolution, time_sample, morton) keys = csdb.get_delayed_writes(delayed_write_key) assert not keys csdb.add_to_delayed_write(write_cuboid_key1, lookup_key, resolution, morton, time_sample, "{dummy resource str}") csdb.add_to_delayed_write(write_cuboid_key2, lookup_key, resolution, morton, time_sample, "{dummy resource str}") csdb.add_to_delayed_write(write_cuboid_key3, lookup_key, resolution, morton, time_sample, "{dummy resource str}") keys = csdb.get_delayed_writes(delayed_write_key) assert len(keys) == 3 assert keys[0] == write_cuboid_key1 assert keys[1] == write_cuboid_key2 assert keys[2] == write_cuboid_key3 keys = csdb.get_delayed_writes(delayed_write_key) assert not keys