Refactoring Mock and separating in other file.