Add a mock for Unit Test