Fixed integration tests

Signed-off-by: Kenrick Yap <[email protected]>
opensearch-project · Dec 18, 2024 · daf11fa · daf11fa
1 parent f8d243d
commit daf11fa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/integ-test/src/integration/scala/org/opensearch/flint/spark/FlintSparkSuite.scala b/integ-test/src/integration/scala/org/opensearch/flint/spark/FlintSparkSuite.scala
@@ -806,8 +806,8 @@ trait FlintSparkSuite extends QueryTest with FlintSuite with OpenSearchSuite wit
          |   city_name STRING,
          |   time_zone STRING,
          |   location STRING,
-         |   ip_range_start BIGINT,
-         |   ip_range_end BIGINT,
+         |   ip_range_start DECIMAL(38,0),
+         |   ip_range_end DECIMAL(38,0),
          |   ipv4 BOOLEAN
          | )
          | USING $tableType $tableOptions
@@ -830,9 +830,9 @@ trait FlintSparkSuite extends QueryTest with FlintSuite with OpenSearchSuite wit
          |  1123655167,
          |  true
          | ),
-         | VALUES (
+         | (
          |  '2a09:bac2:19f8::/45',
-         |  `'CA',
+         |  'CA',
          |  'Canada',
          |  'North America',
          |  'PE',

diff --git a/...test/src/integration/scala/org/opensearch/flint/spark/ppl/FlintSparkPPLGeoipITSuite.scala b/...test/src/integration/scala/org/opensearch/flint/spark/ppl/FlintSparkPPLGeoipITSuite.scala
@@ -41,10 +41,11 @@ class FlintSparkPPLGeoipITSuite
 
     // Retrieve the results
     val results: Array[Row] = frame.collect()
+
     // Define the expected results
     val expectedResults: Array[Row] = Array(
-      Row("66.249.157.90", Row("JM", "Jamaica", "North America", 14, "Saint Catherine Parish", "Portmore", "America/Jamaica", "17.9686, -76.8827")),
-      Row("2a09:bac2:19f8:2ac3::", Row("CA", "Canada", "North America", "PE", "Prince Edward Island", "Charlottetown", "America/Halifax", "46.2396, -63.1355"))
+      Row("66.249.157.90", Row("JM", "Jamaica", "North America", "14", "Saint Catherine Parish", "Portmore", "America/Jamaica", "17.9686,-76.8827")),
+      Row("2a09:bac2:19f8:2ac3::", Row("CA", "Canada", "North America", "PE", "Prince Edward Island", "Charlottetown", "America/Halifax", "46.2396,-63.1355"))
     )
 
     // Compare the results