From 8fc3509895c1364a6981f692f964c834666c85a6 Mon Sep 17 00:00:00 2001
From: Mingi Cho <81455273+ChoMinGi@users.noreply.github.com>
Date: Tue, 24 Mar 2026 15:55:32 +0900
Subject: [PATCH] KAFKA-20323: Refactor deleteRecords stub to use
 partition-based matching for Mockito upgrade

- Change deleteRecords stubs from offset-based to partition-based matching
- Upgrade Mockito from 5.20.0 to 5.23.0
---
 gradle/dependencies.gradle                                | 2 +-
 .../coordinator/share/ShareCoordinatorServiceTest.java    | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/gradle/dependencies.gradle b/gradle/dependencies.gradle
index 50dddc17c049c..05019d9e08729 100644
--- a/gradle/dependencies.gradle
+++ b/gradle/dependencies.gradle
@@ -118,7 +118,7 @@ versions += [
   lz4: "1.10.2",
   mavenArtifact: "3.9.11",
   metrics: "2.2.0",
-  mockito: "5.20.0",
+  mockito: "5.23.0",
   opentelemetryProto: "1.3.2-alpha",
   protobuf: "3.25.5", // a dependency of opentelemetryProto
   pcollections: "4.0.2",
diff --git a/share-coordinator/src/test/java/org/apache/kafka/coordinator/share/ShareCoordinatorServiceTest.java b/share-coordinator/src/test/java/org/apache/kafka/coordinator/share/ShareCoordinatorServiceTest.java
index cf5c51590a581..54ebf9a817d0d 100644
--- a/share-coordinator/src/test/java/org/apache/kafka/coordinator/share/ShareCoordinatorServiceTest.java
+++ b/share-coordinator/src/test/java/org/apache/kafka/coordinator/share/ShareCoordinatorServiceTest.java
@@ -1510,15 +1510,15 @@ public void testRecordPruningTaskPeriodicityWithSomeFailures() throws Exception
             .thenReturn(List.of(tp1, tp2));
 
         when(writer.deleteRecords(
-            any(),
-            eq(10L)
+            eq(tp1),
+            anyLong()
         )).thenReturn(
             CompletableFuture.completedFuture(null)
         );
 
         when(writer.deleteRecords(
-            any(),
-            eq(20L)
+            eq(tp2),
+            anyLong()
         )).thenReturn(
             CompletableFuture.failedFuture(new Exception("bad stuff"))
         );