Add more unit tests.

Signed-off-by: Chris Lalancette <clalancette@gmail.com>