Add more unit tests

Signed-off-by: Zack Cerza <zack.cerza@inktank.com>