update visitRelation with DescribeTableCommand ctor

Signed-off-by: YANGDB <[email protected]>
YANG-DB · Aug 7, 2024 · b021076 · b021076
1 parent e34c0c6
commit b021076
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/...test/src/integration/scala/org/opensearch/flint/spark/ppl/FlintSparkPPLBasicITSuite.scala b/...test/src/integration/scala/org/opensearch/flint/spark/ppl/FlintSparkPPLBasicITSuite.scala
@@ -67,7 +67,8 @@ class FlintSparkPPLBasicITSuite
       val logicalPlan: LogicalPlan = frame.queryExecution.commandExecuted.asInstanceOf[CommandResult].commandLogicalPlan
       // Define the expected logical plan
       val expectedPlan: LogicalPlan =
-          DescribeTableCommand(TableIdentifier("default.flint_ppl_test"), null, isExtended = false, output = Seq())
+           DescribeTableCommand(TableIdentifier("default.flint_ppl_test"), Map.empty, isExtended = false, output = DescribeRelation.getOutputAttrs
+      );
       // Compare the two plans
       comparePlans(logicalPlan, expectedPlan, checkAnalysis = false)
     }

diff --git a/.../scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanBasicQueriesTranslatorTestSuite.scala b/.../scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanBasicQueriesTranslatorTestSuite.scala
@@ -31,7 +31,7 @@ class PPLLogicalPlanBasicQueriesTranslatorTestSuite
     val context = new CatalystPlanContext
     val logPlan = planTransformer.visit(plan(pplParser, "describe t", false), context)
 
-    val expectedPlan = DescribeTableCommand(TableIdentifier("t"), null, isExtended = false, Seq.empty)
+    val expectedPlan = DescribeTableCommand(TableIdentifier("t"), Map.empty, isExtended = false, output = DescribeRelation.getOutputAttrs)
     comparePlans(expectedPlan, logPlan, false)
   }
 
@@ -41,7 +41,7 @@ test("test FQN table describe table clause") {
     val context = new CatalystPlanContext
     val logPlan = planTransformer.visit(plan(pplParser, "describe catalog.schema.t", false), context)
 
-    val expectedPlan = DescribeTableCommand(TableIdentifier("catalog.schema.t"), null, isExtended = false, Seq.empty)
+    val expectedPlan = DescribeTableCommand(TableIdentifier("describe catalog.schema.t"), Map.empty, isExtended = false, output = DescribeRelation.getOutputAttrs)
     comparePlans(expectedPlan, logPlan, false)
   }