def test_get_overall_tasks_progress(self, redis_service_mock):
redis_instance_mock = Mock()
redis_instance_mock.get_int.side_effect = [100, 50]
redis_service_mock.return_value = redis_instance_mock
importer = BulkImportParallelRunner(
open(
os.path.join(os.path.dirname(__file__), '..',
'samples/sample_ocldev.json'), 'r').read(),
'ocladmin', True)
self.assertEqual(importer.get_overall_tasks_progress(), 0)
importer.tasks = [Mock(task_id='task1'), Mock(task_id='task2')]
self.assertEqual(importer.get_overall_tasks_progress(), 150)
def test_notify_progress(self, redis_service_mock): # pylint: disable=no-self-use
redis_instance_mock = Mock(set_json=Mock())
redis_service_mock.return_value = redis_instance_mock
importer = BulkImportParallelRunner(
open(
os.path.join(os.path.dirname(__file__), '..',
'samples/sample_ocldev.json'), 'r').read(),
'ocladmin', True, None, 'task-id')
importer.tasks = [Mock(task_id='task-1'), Mock(task_id='task-2')]
now = 1607346541.793877 # datetime.datetime(2020, 12, 7, 13, 09, 1, 793877) UTC
importer.start_time = now
importer.elapsed_seconds = 10.45
importer.notify_progress()
redis_instance_mock.set_json.assert_called_once_with(
'task-id',
dict(
summary=
"Started: 2020-12-07 13:09:01.793877 | Processed: 0/64 | Time: 10.45secs",
#sub_task_ids=['task-1', 'task-2']
))