Add mock tests