Add more unit tests covering success/status