Unit tests for some part of functionality