Add unit tests

[alexandear]

Cover at least 80% of code statements with unit tests.

[AntonyAsatiani]

Hello @alexandear,

Could you please assign the following issue to me?

Thank you!

[alexandear]

@AntonyAsatiani assigned, thank you