Add a mock for Unit Test
(cherry picked from commit b50c9c5a2969ced618b97caf7552e1272d1601a5)