Finish switching to mock from mox.

This cleans up a few tests to finish the mox -> mock transition. Fixes #102.

PTAL @dhermes