Use mocked interface - phase 1

TODO: Use mocks for all other tests.