OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # -*- coding: utf-8 -*- | 2 # -*- coding: utf-8 -*- |
3 """This file contains tests for the task manager.""" | 3 """This file contains tests for the task manager.""" |
4 | 4 |
5 from __future__ import unicode_literals | 5 from __future__ import unicode_literals |
6 | 6 |
7 import time | 7 import time |
8 import unittest | 8 import unittest |
9 | 9 |
10 from dfvfs.lib import definitions as dfvfs_definitions | 10 from dfvfs.lib import definitions as dfvfs_definitions |
(...skipping 384 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
395 self.assertEqual(len(manager._tasks_abandoned), 0) | 395 self.assertEqual(len(manager._tasks_abandoned), 0) |
396 self.assertEqual(len(manager._tasks_pending_merge), 3) | 396 self.assertEqual(len(manager._tasks_pending_merge), 3) |
397 | 397 |
398 # Indicate to the task manager that an unknown task is pending merge. | 398 # Indicate to the task manager that an unknown task is pending merge. |
399 task = tasks.Task() | 399 task = tasks.Task() |
400 task.storage_file_size = 10 | 400 task.storage_file_size = 10 |
401 | 401 |
402 with self.assertRaises(KeyError): | 402 with self.assertRaises(KeyError): |
403 manager.UpdateTaskAsProcessingByIdentifier(task.identifier) | 403 manager.UpdateTaskAsProcessingByIdentifier(task.identifier) |
404 | 404 |
405 def testUpdateTasksAsPendingMerge(self): | |
406 """Tests the UpdateTasksAsPendingMerge function.""" | |
407 manager = task_manager.TaskManager() | |
408 | |
409 self.assertEqual(len(manager._tasks_queued), 0) | |
410 self.assertEqual(len(manager._tasks_processing), 0) | |
411 self.assertEqual(len(manager._tasks_abandoned), 0) | |
412 self.assertEqual(len(manager._tasks_pending_merge), 0) | |
413 | |
414 task1 = manager.CreateTask(self._TEST_SESSION_IDENTIFIER) | |
415 task1.storage_file_size = 10 | |
416 | |
417 task2 = manager.CreateTask(self._TEST_SESSION_IDENTIFIER) | |
418 task2.storage_file_size = 10 | |
419 | |
420 self.assertEqual(len(manager._tasks_queued), 2) | |
421 self.assertEqual(len(manager._tasks_processing), 0) | |
422 self.assertEqual(len(manager._tasks_abandoned), 0) | |
423 self.assertEqual(len(manager._tasks_pending_merge), 0) | |
424 | |
425 manager.UpdateTasksAsPendingMerge([task1, task2]) | |
426 | |
427 self.assertEqual(len(manager._tasks_queued), 0) | |
428 self.assertEqual(len(manager._tasks_processing), 0) | |
429 self.assertEqual(len(manager._tasks_abandoned), 0) | |
430 self.assertEqual(len(manager._tasks_pending_merge), 2) | |
431 | |
432 def testUpdateTaskAsProcessingByIdentifier(self): | 405 def testUpdateTaskAsProcessingByIdentifier(self): |
433 """Tests the UpdateTaskAsProcessingByIdentifier function.""" | 406 """Tests the UpdateTaskAsProcessingByIdentifier function.""" |
434 manager = task_manager.TaskManager() | 407 manager = task_manager.TaskManager() |
435 | 408 |
436 self.assertEqual(len(manager._tasks_queued), 0) | 409 self.assertEqual(len(manager._tasks_queued), 0) |
437 self.assertEqual(len(manager._tasks_processing), 0) | 410 self.assertEqual(len(manager._tasks_processing), 0) |
438 self.assertEqual(len(manager._tasks_abandoned), 0) | 411 self.assertEqual(len(manager._tasks_abandoned), 0) |
439 self.assertEqual(len(manager._tasks_pending_merge), 0) | 412 self.assertEqual(len(manager._tasks_pending_merge), 0) |
440 | 413 |
441 task = manager.CreateTask(self._TEST_SESSION_IDENTIFIER) | 414 task = manager.CreateTask(self._TEST_SESSION_IDENTIFIER) |
(...skipping 133 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
575 | 548 |
576 manager.UpdateTaskAsProcessingByIdentifier(task.identifier) | 549 manager.UpdateTaskAsProcessingByIdentifier(task.identifier) |
577 self.assertEqual(manager.GetAbandonedTasks(), []) | 550 self.assertEqual(manager.GetAbandonedTasks(), []) |
578 self.assertTrue(manager.HasPendingTasks()) | 551 self.assertTrue(manager.HasPendingTasks()) |
579 | 552 |
580 # TODO: Add tests for updating tasks. | 553 # TODO: Add tests for updating tasks. |
581 | 554 |
582 | 555 |
583 if __name__ == '__main__': | 556 if __name__ == '__main__': |
584 unittest.main() | 557 unittest.main() |
OLD | NEW |