Extending tests to cover other methods

Signed-off-by: Aris Kemper <aris.github@gmail.com>