[3048] Added unit tests.

Gist: https://gist.github.com/mjelonek92/494f26f5c346fadc2614