def test_remove_task_from_cache_value_not_in_cache(self):
"""Test that a task is removed."""
task_list = [1, 2, 3, 4]
_cache = WorkerCache()
_cache.set_host_specific_task_list(task_list)
self.assertEqual(_cache.host_specific_worker_cache, task_list)
_cache.remove_task_from_cache(5)
self.assertEqual(_cache.host_specific_worker_cache, task_list)
def test_remove_task_from_cache(self):
"""Test that a task is removed."""
task_key = "task_key"
_cache = WorkerCache()
_cache.add_task_to_cache(task_key)
self.assertEqual(_cache.worker_cache, [task_key])
_cache.remove_task_from_cache(task_key)
self.assertEqual(_cache.worker_cache, [])
def test_remove_task_from_cache_value_not_in_cache(self):
"""Test that a task is removed."""
task_list = [1, 2, 3, 4]
_cache = WorkerCache()
for task in task_list:
_cache.add_task_to_cache(task)
self.assertEqual(_cache.worker_cache, task_list)
_cache.remove_task_from_cache(5)
self.assertEqual(_cache.worker_cache, task_list)
def test_remove_task_from_cache(self):
"""Test that a task is removed."""
task_list = [1, 2, 3, 4]
expected = [1, 2, 3]
_cache = WorkerCache()
_cache.set_host_specific_task_list(task_list)
self.assertEqual(_cache.host_specific_worker_cache, task_list)
_cache.remove_task_from_cache(4)
self.assertEqual(_cache.host_specific_worker_cache, expected)
