Revert "updating tests which were mocking task_status_manager since TaskStatusManager is deleted now"

This reverts commit 13ed0e62aa3ca049b5119383ac71668021073730.
adding mock patches for TaskStatus model instead of task_status_manager which doesn't exist anymore