Switch to using @Rule instead of @ClassRule so we don't need to reset the mock on every run