Setup mock with expected method in setup