adding unit test for task context