More unittest coverage

Signed-off-by: Esben Haabendal <esben@haabendal.dk>