[Testing] implement unit tests