Add unittests for logger