updating tests which were mocking task_status_manager since TaskStatusManager is deleted now