ITTE-1165 Created unit tests

- Created task unit test.
- Fixed broken unit tests.
- Added UML diagrams.