BUG#35212199: Check for identifier quotes in the database name

Connector/Python raises an error when setting the database name on a
connection that includes non-alphanumeric characters and using the pure
Python implementation.

This patch fixes the issue by using `connection.cmd_init_db()` to set
the database name.

Change-Id: I7f597238104fd79b324fe67be8f428f5e4c18b6c

Author: nmariz

CHANGES.txt

@@ -12,6 +12,7 @@ v8.1.0
 ======
 
 - WL#15591: Improve the network module
+- BUG#35212199: Check for identifier quotes in the database name
 
 v8.0.33
 =======

lib/mysql/connector/connection.py

@@ -1279,7 +1279,7 @@ def database(self) -> str:
     @database.setter
     def database(self, value: str) -> None:
         """Set the current database"""
-        self.cmd_query(f"USE {value}")
+        self.cmd_init_db(value)
 
     def is_connected(self) -> bool:
         """Reports whether the connection to MySQL Server is available

tests/test_bugs.py

@@ -7484,3 +7484,42 @@ def compressed_packet_incomplete_at_last(self):
                 res = cur.fetchone()
                 self.assertEqual(data[:-5000], res[0][:-5000])
                 self.assertEqual(len(data), len(res[0]))
+
+
+class BugOra35212199(tests.MySQLConnectorTests):
+    """BUG#35212199: Check for identifier quotes in the database name
+
+    Connector/Python raises an error when setting the database name on a
+    connection that includes non-alphanumeric characters and using the pure
+    Python implementation.
+
+    This patch fixes the issue by using `connection.cmd_init_db()` to set
+    the database name.
+    """
+
+    database = "abc-abc|$%"
+
+    def setUp(self):
+        with mysql.connector.connect(**tests.get_mysql_config()) as cnx:
+            try:
+                cnx.cmd_query(f"DROP DATABASE `{self.database}`")
+            except errors.DatabaseError:  # Database already exists
+                pass
+            cnx.cmd_query(f"CREATE DATABASE `{self.database}`")
+
+    def tearDown(self):
+        with mysql.connector.connect(**tests.get_mysql_config()) as cnx:
+            cnx.cmd_query(f"DROP DATABASE `{self.database}`")
+
+    @foreach_cnx()
+    def test_set_database_name(self):
+        """Test setting the database on an established connection."""
+        self.cnx.database = self.database
+
+    @foreach_cnx()
+    def test_connect_database_name(self):
+        """Test setting the database when establishing a connection."""
+        config = tests.get_mysql_config()
+        config["database"] = self.database
+        with self.cnx.__class__(**config) as cnx:
+            self.assertTrue(cnx.is_connected())