From 9c23dc34dbb4c2be64af3952051a50d11ebb7ca4 Mon Sep 17 00:00:00 2001
From: Dongjoon Hyun <dongjoon@apache.org>
Date: Thu, 22 May 2025 14:04:40 -0700
Subject: [PATCH] [SPARK-52268] Add `variant` SQL test and answer file

---
 Tests/SparkConnectTests/Resources/queries/variant.sql        | 5 +++++
 Tests/SparkConnectTests/Resources/queries/variant.sql.answer | 5 +++++
 Tests/SparkConnectTests/SQLTests.swift                       | 1 +
 3 files changed, 11 insertions(+)
 create mode 100644 Tests/SparkConnectTests/Resources/queries/variant.sql
 create mode 100644 Tests/SparkConnectTests/Resources/queries/variant.sql.answer

diff --git a/Tests/SparkConnectTests/Resources/queries/variant.sql b/Tests/SparkConnectTests/Resources/queries/variant.sql
new file mode 100644
index 0000000..ae81f1e
--- /dev/null
+++ b/Tests/SparkConnectTests/Resources/queries/variant.sql
@@ -0,0 +1,5 @@
+FROM VALUES parse_json('{"a": true, "b": 1, "c": "swift"}') T(v)
+|> SELECT v,
+          variant_get(v, '$.a', 'boolean') as a,
+          variant_get(v, '$.b', 'int') as b,
+          variant_get(v, '$.c', 'string') as c
diff --git a/Tests/SparkConnectTests/Resources/queries/variant.sql.answer b/Tests/SparkConnectTests/Resources/queries/variant.sql.answer
new file mode 100644
index 0000000..c50602e
--- /dev/null
+++ b/Tests/SparkConnectTests/Resources/queries/variant.sql.answer
@@ -0,0 +1,5 @@
++----------------------------+----+---+-----+
+|                           v|   a|  b|    c|
++----------------------------+----+---+-----+
+|{"a":true,"b":1,"c":"swift"}|true|  1|swift|
++----------------------------+----+---+-----+
\ No newline at end of file
diff --git a/Tests/SparkConnectTests/SQLTests.swift b/Tests/SparkConnectTests/SQLTests.swift
index a9c6b71..1b9be5f 100644
--- a/Tests/SparkConnectTests/SQLTests.swift
+++ b/Tests/SparkConnectTests/SQLTests.swift
@@ -85,6 +85,7 @@ struct SQLTests {
     "create_table_function.sql",
     "pipesyntax.sql",
     "explain.sql",
+    "variant.sql",
   ]
 
 #if !os(Linux)