def test_wait_for_page_in_timeout(self):
"""Test to make sure page in timeout works properly"""
start_time = datetime.now()
with self.assertRaises(SpdbError):
csdb = CacheStateDB(self.config_data)
ch = csdb.create_page_in_channel()
csdb.wait_for_page_in(["MY_TEST_KEY1", "MY_TEST_KEY2"], ch, 1)
assert (datetime.now() - start_time).seconds < 3
def test_wait_for_page_in_timeout(self):
"""Test to make sure page in timeout works properly"""
start_time = datetime.now()
with self.assertRaises(SpdbError):
csdb = CacheStateDB(self.config_data)
ch = csdb.create_page_in_channel()
csdb.wait_for_page_in(["MY_TEST_KEY1", "MY_TEST_KEY2"], ch, 1)
assert (datetime.now() - start_time).seconds < 3
def test_wait_for_page_in(self):
"""Test to make sure waiting for all the keys to be paged in works properly"""
# Create test instance
csdb1 = CacheStateDB(self.config_data)
csdb2 = CacheStateDB(self.config_data)
# Create page in channel in the first instance
ch = csdb1.create_page_in_channel()
# Publish a message
csdb2.notify_page_in_complete(ch, "MY_TEST_KEY1")
csdb2.notify_page_in_complete(ch, "MY_TEST_KEY2")
# Wait for page in
csdb1.wait_for_page_in(["MY_TEST_KEY1", "MY_TEST_KEY2"], ch, 5)
def test_wait_for_page_in(self):
"""Test to make sure waiting for all the keys to be paged in works properly"""
# Create test instance
csdb1 = CacheStateDB(self.config_data)
csdb2 = CacheStateDB(self.config_data)
# Create page in channel in the first instance
ch = csdb1.create_page_in_channel()
# Publish a message
csdb2.notify_page_in_complete(ch, "MY_TEST_KEY1")
csdb2.notify_page_in_complete(ch, "MY_TEST_KEY2")
# Wait for page in
csdb1.wait_for_page_in(["MY_TEST_KEY1", "MY_TEST_KEY2"], ch, 5)