def GetStatusInformation(self):
"""Retrieves status information about the tasks.
Returns:
TasksStatus: tasks status information.
"""
status = processing_status.TasksStatus()
with self._lock:
status.number_of_abandoned_tasks = len(self._tasks_abandoned)
status.number_of_queued_tasks = len(self._tasks_queued)
status.number_of_tasks_pending_merge = len(self._tasks_pending_merge)
status.number_of_tasks_processing = len(self._tasks_processing)
status.total_number_of_tasks = self._total_number_of_tasks
return status
def testSample(self):
"""Tests the Sample function."""
task_status = processing_status.TasksStatus()
profiling_configuration = configurations.ProfilingConfiguration()
with shared_test_lib.TempDirectory() as temp_directory:
profiling_configuration.directory = temp_directory
test_profiler = profilers.TaskQueueProfiler(
'test', profiling_configuration)
test_profiler.Start()
for _ in range(5):
test_profiler.Sample(task_status)
time.sleep(0.01)
test_profiler.Stop()
def testUpdateTasksStatus(self):
"""Tests the UpdateTasksStatus function."""
task_status = processing_status.TasksStatus()
status = processing_status.ProcessingStatus()
status.UpdateTasksStatus(task_status)
def testInitialization(self):
"""Tests the __init__ function."""
task_status = processing_status.TasksStatus()
self.assertIsNotNone(task_status)
