diff --git a/integ-test/src/integration/scala/org/opensearch/flint/spark/ppl/FlintSparkPPLBasicITSuite.scala b/integ-test/src/integration/scala/org/opensearch/flint/spark/ppl/FlintSparkPPLBasicITSuite.scala index 3b381ce49..58ce488a7 100644 --- a/integ-test/src/integration/scala/org/opensearch/flint/spark/ppl/FlintSparkPPLBasicITSuite.scala +++ b/integ-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/ppl-spark-integration/src/test/scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanBasicQueriesTranslatorTestSuite.scala b/ppl-spark-integration/src/test/scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanBasicQueriesTranslatorTestSuite.scala index 67a4c7563..95c18449a 100644 --- a/ppl-spark-integration/src/test/scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanBasicQueriesTranslatorTestSuite.scala +++ b/ppl-spark-integration/src/test/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) }