def testContains(self):
"""Tests the __contains__ function."""
task = tasks.Task()
task.storage_file_size = 10
heap = task_manager._PendingMergeTaskHeap()
heap.PushTask(task)
self.assertIn(task.identifier, heap)
def testContains(self):
"""Tests the __contains__ function."""
task = tasks.Task()
task.storage_file_size = 10
heap = task_manager._PendingMergeTaskHeap()
heap.PushTask(task)
self.assertIn(task.identifier, heap)
def testLength(self):
"""Tests the __len__ function."""
task = tasks.Task()
task.storage_file_size = 10
heap = task_manager._PendingMergeTaskHeap()
self.assertEqual(len(heap), 0)
heap.PushTask(task)
self.assertEqual(len(heap), 1)
def testLength(self):
"""Tests the __len__ function."""
task = tasks.Task()
task.storage_file_size = 10
heap = task_manager._PendingMergeTaskHeap()
self.assertEqual(len(heap), 0)
heap.PushTask(task)
self.assertEqual(len(heap), 1)
def testPopTask(self):
"""Tests the PopTask function."""
task = tasks.Task()
task.storage_file_size = 10
heap = task_manager._PendingMergeTaskHeap()
result_task = heap.PopTask()
self.assertIsNone(result_task)
heap.PushTask(task)
self.assertEqual(len(heap), 1)
result_task = heap.PopTask()
self.assertEqual(len(heap), 0)
self.assertEqual(result_task, task)
def testPopTask(self):
"""Tests the PopTask function."""
task = tasks.Task()
task.storage_file_size = 10
heap = task_manager._PendingMergeTaskHeap()
result_task = heap.PopTask()
self.assertIsNone(result_task)
heap.PushTask(task)
self.assertEqual(len(heap), 1)
result_task = heap.PopTask()
self.assertEqual(len(heap), 0)
self.assertEqual(result_task, task)
def testPushTask(self):
"""Tests the PushTask function."""
task = tasks.Task()
task.storage_file_size = 10
heap = task_manager._PendingMergeTaskHeap()
self.assertEqual(len(heap), 0)
heap.PushTask(task)
self.assertEqual(len(heap), 1)
task = tasks.Task()
task.storage_file_size = 100
heap.PushTask(task)
self.assertEqual(len(heap), 2)
task = tasks.Task()
with self.assertRaises(ValueError):
heap.PushTask(task)
def testPushTask(self):
"""Tests the PushTask function."""
task = tasks.Task()
task.storage_file_size = 10
heap = task_manager._PendingMergeTaskHeap()
self.assertEqual(len(heap), 0)
heap.PushTask(task)
self.assertEqual(len(heap), 1)
task = tasks.Task()
task.storage_file_size = 100
heap.PushTask(task)
self.assertEqual(len(heap), 2)
task = tasks.Task()
with self.assertRaises(ValueError):
heap.PushTask(task)
