ITTE-851 Added unit test for the task and added return  types for some
methods.

Signed-off-by: Simeon Iliev <sisabski23@gmail.com>