test(unit): Extend unit tests