From c66e4a33e1a07c439f03fe47f146a6c6482bf6df Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Thu, 8 Feb 2024 06:38:42 +0100
Subject: Merging upstream version 21.0.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 tests/dialects/test_mysql.py | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

(limited to 'tests/dialects/test_mysql.py')

diff --git a/tests/dialects/test_mysql.py b/tests/dialects/test_mysql.py
index 3a3e49e..fd27a1e 100644
--- a/tests/dialects/test_mysql.py
+++ b/tests/dialects/test_mysql.py
@@ -386,7 +386,6 @@ class TestMySQL(Validator):
             "snowflake": "SELECT 11",
             "spark": "SELECT 11",
             "sqlite": "SELECT 11",
-            "mysql": "SELECT b'1011'",
             "tableau": "SELECT 11",
             "teradata": "SELECT 11",
             "trino": "SELECT 11",
@@ -590,6 +589,26 @@ class TestMySQL(Validator):
         )
 
     def test_mysql(self):
+        self.validate_all(
+            "SELECT JSON_EXTRACT('[10, 20, [30, 40]]', '$[1]')",
+            read={
+                "sqlite": "SELECT JSON_EXTRACT('[10, 20, [30, 40]]', '$[1]')",
+            },
+            write={
+                "mysql": "SELECT JSON_EXTRACT('[10, 20, [30, 40]]', '$[1]')",
+                "sqlite": "SELECT '[10, 20, [30, 40]]' -> '$[1]'",
+            },
+        )
+        self.validate_all(
+            "SELECT JSON_EXTRACT('[10, 20, [30, 40]]', '$[1]', '$[0]')",
+            read={
+                "sqlite": "SELECT JSON_EXTRACT('[10, 20, [30, 40]]', '$[1]', '$[0]')",
+            },
+            write={
+                "mysql": "SELECT JSON_EXTRACT('[10, 20, [30, 40]]', '$[1]', '$[0]')",
+                "sqlite": "SELECT JSON_EXTRACT('[10, 20, [30, 40]]', '$[1]', '$[0]')",
+            },
+        )
         self.validate_all(
             "SELECT * FROM x LEFT JOIN y ON x.id = y.id UNION SELECT * FROM x RIGHT JOIN y ON x.id = y.id LIMIT 0",
             read={
@@ -790,6 +809,7 @@ COMMENT='客户账户表'"""
             ("CHARACTER SET", "CHARACTER SET"),
             ("COLLATION", "COLLATION"),
             ("DATABASES", "DATABASES"),
+            ("SCHEMAS", "DATABASES"),
             ("FUNCTION STATUS", "FUNCTION STATUS"),
             ("PROCEDURE STATUS", "PROCEDURE STATUS"),
             ("GLOBAL STATUS", "GLOBAL STATUS"),
@@ -850,7 +870,7 @@ COMMENT='客户账户表'"""
             self.assertEqual(show.text("target"), "foo")
 
     def test_show_grants(self):
-        show = self.validate_identity(f"SHOW GRANTS FOR foo")
+        show = self.validate_identity("SHOW GRANTS FOR foo")
         self.assertIsInstance(show, exp.Show)
         self.assertEqual(show.name, "GRANTS")
         self.assertEqual(show.text("target"), "foo")
-- 
cgit v1.2.3