Use mock for tests, PLAYGROUND-233