Refactor: extract duplication

mockito · Mar 2, 2024 · f5f2718 · f5f2718
1 parent 107bdfe
commit f5f2718
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/main/java/org/mockito/internal/verification/Only.java b/src/main/java/org/mockito/internal/verification/Only.java
@@ -25,15 +25,16 @@ public void verify(VerificationData data) {
         MatchableInvocation target = data.getTarget();
         List<Invocation> invocations = data.getAllInvocations();
         List<Invocation> chunk = findInvocations(invocations, target);
-        if (invocations.size() != 1 && !chunk.isEmpty()) {
+        if(!chunk.isEmpty()) {
             markVerified(chunk.get(0), target);
+        }
+        if (invocations.size() != 1 && !chunk.isEmpty()) {
             Invocation unverified = findFirstUnverified(invocations);
             throw noMoreInteractionsWanted(unverified, (List) invocations);
         }
         if (invocations.size() != 1 || chunk.isEmpty()) {
             throw wantedButNotInvoked(target);
         }
-        markVerified(chunk.get(0), target);
     }
 
     @Override