Started using the mocks in tests.
Replaced an old mock with testify mock.