Add tests for logger