-add mock test class