Added some mock tests