Add some mock tests