Исходя из вашего описания ожидаемого поведения, это звучит как довольно надежный способ модульного тестирования метода. По «истинному пути Мокито» я полагаю, что вы хотите что-то вроде:
Set mockBSet = mock(Set.class);
stub(mockBSet.size()).toReturn(2);
A testObjectA = new A(mockBSet);
Assert.assertEquals(testObjectA.getSomethingAmount(), 2);
Это при условии, что инициализация Set и тестовых объектов не имеет побочных эффектов, которые вы хотите, чтобы тест принимал во внимание. (Например, всегда добавляются некоторые «что-то» по умолчанию.) В этом случае вы также можете опустить заглушку и подтвердить, что тестируемый метод возвращает значение, равное размеру набора.