Add mock for unit test.