Add mock test