fix test

Texera · Dec 20, 2024 · b603b8d · b603b8d
1 parent 69d8290
commit b603b8d
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/core/amber/src/test/scala/edu/uci/ics/amber/engine/e2e/DataProcessingSpec.scala b/core/amber/src/test/scala/edu/uci/ics/amber/engine/e2e/DataProcessingSpec.scala
@@ -69,16 +69,8 @@ class DataProcessingSpec
       .registerCallback[ExecutionStateUpdate](evt => {
         if (evt.state == COMPLETED) {
           results = workflow.logicalPlan.getTerminalOperatorIds
-            .map(sinkOpId =>
-              (sinkOpId, workflow.logicalPlan.getUpstreamOps(sinkOpId).head.operatorIdentifier)
-            )
-            .filter {
-              case (_, upstreamOpId) => resultStorage.contains(upstreamOpId)
-            }
-            .map {
-              case (sinkOpId, upstreamOpId) =>
-                (sinkOpId, resultStorage.get(upstreamOpId).get().toList)
-            }
+            .filter(terminalOpId => resultStorage.contains(terminalOpId))
+            .map(terminalOpId => terminalOpId -> resultStorage.get(terminalOpId).get().toList)
             .toMap
           completion.setDone()
         }