More tests for the functionality

Signed-off-by: license2e <ilya@shindyapin.com>