From c73906ac9b943894e9b81a19f5ed23d3e815d2c4 Mon Sep 17 00:00:00 2001
From: Chen Dai <daichen@amazon.com>
Date: Wed, 8 Nov 2023 18:43:35 -0800
Subject: [PATCH] Fix recover index IT

Signed-off-by: Chen Dai <daichen@amazon.com>
---
 .../opensearch/flint/spark/FlintSparkIndexJobSqlITSuite.scala | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkIndexJobSqlITSuite.scala b/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkIndexJobSqlITSuite.scala
index ddbfeeb16..b8d3028de 100644
--- a/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkIndexJobSqlITSuite.scala
+++ b/integ-test/src/test/scala/org/opensearch/flint/spark/FlintSparkIndexJobSqlITSuite.scala
@@ -67,9 +67,10 @@ class FlintSparkIndexJobSqlITSuite extends FlintSparkSuite with Matchers {
       assertion
         .run { checkpointDir =>
           s""" CREATE INDEX $testIndex ON $testTable
-             | (time, name)
+             | (timestamp, name)
              | WITH (
              |   auto_refresh = true,
+             |   id_expression = 'timestamp',
              |   checkpoint_location = '${checkpointDir.getAbsolutePath}'
              | )
              |""".stripMargin
@@ -94,6 +95,7 @@ class FlintSparkIndexJobSqlITSuite extends FlintSparkSuite with Matchers {
              | $testMvQuery
              | WITH (
              |   auto_refresh = true,
+             |   id_expression = 'name',
              |   checkpoint_location = '${checkpointDir.getAbsolutePath}'
              | )
              |""".stripMargin