Конечно, для вашего count () - метод, получающий приращение, Mockito-Answer - лучший способ.
Кстати: но, тем не менее, напомните, что вы можете «изменить» (определить) поведение поддельного-метод в случае последующих вызовов:
when(mock.someMethod("some arg"))
.thenThrow(new RuntimeException())
.thenReturn("foo");
Здесь вы видите, что первый вызов вызывает исключение RuntimeException, второй вызов возвращает строку "foo" ...