(flink) branch master updated: [FLINK-37137][runtime] Fix unstable test testGenerateStreamGraphJson (#25990)

[junrui]

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

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


The following commit(s) were added to refs/heads/master by this push:
     new 07ea97cd1ae [FLINK-37137][runtime] Fix unstable test 
testGenerateStreamGraphJson (#25990)
07ea97cd1ae is described below

commit 07ea97cd1ae71251abd4b0da57803850fe11bc33
Author: Yu Chen <yuchen.e...@gmail.com>
AuthorDate: Thu Jan 16 10:04:37 2025 +0800

    [FLINK-37137][runtime] Fix unstable test testGenerateStreamGraphJson 
(#25990)
---
 .../apache/flink/runtime/jobgraph/jsonplan/JsonGeneratorTest.java    | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git 
a/flink-runtime/src/test/java/org/apache/flink/runtime/jobgraph/jsonplan/JsonGeneratorTest.java
 
b/flink-runtime/src/test/java/org/apache/flink/runtime/jobgraph/jsonplan/JsonGeneratorTest.java
index b187e997f00..3df0e283bd7 100644
--- 
a/flink-runtime/src/test/java/org/apache/flink/runtime/jobgraph/jsonplan/JsonGeneratorTest.java
+++ 
b/flink-runtime/src/test/java/org/apache/flink/runtime/jobgraph/jsonplan/JsonGeneratorTest.java
@@ -180,8 +180,9 @@ public class JsonGeneratorTest {
         expectedJobVertexIds.add(null);
         validateStreamGraph(streamGraph, parsedStreamGraph, 
expectedJobVertexIds);
 
-        jobVertexIdMap.put(1, new JobVertexID());
-        jobVertexIdMap.put(2, new JobVertexID());
+        for (StreamNode node : streamGraph.getStreamNodes()) {
+            jobVertexIdMap.put(node.getId(), new JobVertexID());
+        }
         streamGraphJson = 
JsonPlanGenerator.generateStreamGraphJson(streamGraph, jobVertexIdMap);
 
         parsedStreamGraph = mapper.readValue(streamGraphJson, 
StreamGraphJsonSchema.class);