Use pure mock instead of partial mock.