Added some tests

Signed-off-by: Anton Tanasenko <atg.sleepless@gmail.com>