added back random test, connection debug

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

src/databricks/sql/client.py

@@ -305,7 +305,7 @@ def read(self) -> Optional[OAuthToken]:
         self.use_inline_params = self._set_use_inline_params_with_warning(
             kwargs.get("use_inline_params", False)
         )
-
+        print("Connection init : session_id_hex", self.get_session_id_hex(), flush=True)
         TelemetryClientFactory.initialize_telemetry_client(
             telemetry_enabled=self.telemetry_enabled,
             session_id_hex=self.get_session_id_hex(),
@@ -440,6 +440,7 @@ def cursor(
 
     def close(self) -> None:
         """Close the underlying session and mark all associated cursors as closed."""
+        print("Connection close: session_id_hex: ", self.get_session_id_hex(), flush=True)
         self._close()
 
     def _close(self, close_cursors=True) -> None:

tests/unit/test_client.py

@@ -650,36 +650,36 @@ def test_disable_pandas_respected(self, mock_thrift_backend_class):
 
         mock_table.itercolumns.assert_called_once_with()
 
-    # def test_column_name_api(self):
-    #     ResultRow = Row("first_col", "second_col", "third_col")
-    #     data = [
-    #         ResultRow("val1", 321, 52.32),
-    #         ResultRow("val2", 2321, 252.32),
-    #     ]
-
-    #     expected_values = [["val1", 321, 52.32], ["val2", 2321, 252.32]]
-
-    #     for (row, expected) in zip(data, expected_values):
-    #         self.assertEqual(row.first_col, expected[0])
-    #         self.assertEqual(row.second_col, expected[1])
-    #         self.assertEqual(row.third_col, expected[2])
-
-    #         self.assertEqual(row["first_col"], expected[0])
-    #         self.assertEqual(row["second_col"], expected[1])
-    #         self.assertEqual(row["third_col"], expected[2])
-
-    #         self.assertEqual(row[0], expected[0])
-    #         self.assertEqual(row[1], expected[1])
-    #         self.assertEqual(row[2], expected[2])
-
-    #         self.assertEqual(
-    #             row.asDict(),
-    #             {
-    #                 "first_col": expected[0],
-    #                 "second_col": expected[1],
-    #                 "third_col": expected[2],
-    #             },
-    #         )
+    def test_column_name_api(self):
+        ResultRow = Row("first_col", "second_col", "third_col")
+        data = [
+            ResultRow("val1", 321, 52.32),
+            ResultRow("val2", 2321, 252.32),
+        ]
+
+        expected_values = [["val1", 321, 52.32], ["val2", 2321, 252.32]]
+
+        for (row, expected) in zip(data, expected_values):
+            self.assertEqual(row.first_col, expected[0])
+            self.assertEqual(row.second_col, expected[1])
+            self.assertEqual(row.third_col, expected[2])
+
+            self.assertEqual(row["first_col"], expected[0])
+            self.assertEqual(row["second_col"], expected[1])
+            self.assertEqual(row["third_col"], expected[2])
+
+            self.assertEqual(row[0], expected[0])
+            self.assertEqual(row[1], expected[1])
+            self.assertEqual(row[2], expected[2])
+
+            self.assertEqual(
+                row.asDict(),
+                {
+                    "first_col": expected[0],
+                    "second_col": expected[1],
+                    "third_col": expected[2],
+                },
+            )
 
     @patch("%s.client.ThriftBackend" % PACKAGE_NAME)
     def test_finalizer_closes_abandoned_connection(self, mock_client_class):