Add mock tests for task controller