[flink] 02/04: [hotfix][test] Adds helper method to testingLeaderElectionDriver for specifying session ID

[mapohl]

This is an automated email from the ASF dual-hosted git repository.

mapohl pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/flink.git

commit 1d208f13d8954c68a10c1b380b3b0dfd00921f63
Author: Matthias Pohl <matthias.p...@aiven.io>
AuthorDate: Tue Apr 18 18:26:09 2023 +0200

    [hotfix][test] Adds helper method to testingLeaderElectionDriver for 
specifying session ID
    
    Signed-off-by: Matthias Pohl <matthias.p...@aiven.io>
---
 .../flink/runtime/leaderelection/TestingLeaderElectionDriver.java | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git 
a/flink-runtime/src/test/java/org/apache/flink/runtime/leaderelection/TestingLeaderElectionDriver.java
 
b/flink-runtime/src/test/java/org/apache/flink/runtime/leaderelection/TestingLeaderElectionDriver.java
index 5352e498641..9f344b30aa3 100644
--- 
a/flink-runtime/src/test/java/org/apache/flink/runtime/leaderelection/TestingLeaderElectionDriver.java
+++ 
b/flink-runtime/src/test/java/org/apache/flink/runtime/leaderelection/TestingLeaderElectionDriver.java
@@ -69,13 +69,17 @@ public class TestingLeaderElectionDriver implements 
LeaderElectionDriver {
         return leaderInformation;
     }
 
-    public void isLeader() {
+    public void isLeader(UUID newSessionID) {
         synchronized (lock) {
             isLeader.set(true);
-            leaderElectionEventHandler.onGrantLeadership(UUID.randomUUID());
+            leaderElectionEventHandler.onGrantLeadership(newSessionID);
         }
     }
 
+    public void isLeader() {
+        isLeader(UUID.randomUUID());
+    }
+
     public void notLeader() {
         synchronized (lock) {
             isLeader.set(false);