Added unit tests mockups