доделал unit тесты