change names of ThriftBackend -> ThriftDatabricksClient in tests

varun-edachali-dbx · varun-edachali-dbx · commit 6ecb6bf97cd9 · 2025-05-28T03:47:03.000Z
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/tests/unit/test_client.py b/tests/unit/test_client.py
@@ -81,7 +81,7 @@ class ClientTestSuite(unittest.TestCase):
         "access_token": "tok",
     }
 
-    @patch("%s.session.ThriftBackend" % PACKAGE_NAME, ThriftBackendMockFactory.new())
+    @patch("%s.session.ThriftDatabricksClient" % PACKAGE_NAME, ThriftDatabricksClientMockFactory.new())
     @patch("%s.client.ResultSet" % PACKAGE_NAME)
     def test_closing_connection_closes_commands(self, mock_result_set_class):
         # Test once with has_been_closed_server side, once without

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ class ClientTestSuite(unittest.TestCase):`
`81`	`81`	`"access_token": "tok",`
`82`	`82`	`}`
`83`	`83`
`84`		`- @patch("%s.session.ThriftBackend" % PACKAGE_NAME, ThriftBackendMockFactory.new())`
	`84`	`+ @patch("%s.session.ThriftDatabricksClient" % PACKAGE_NAME, ThriftDatabricksClientMockFactory.new())`
`85`	`85`	`@patch("%s.client.ResultSet" % PACKAGE_NAME)`
`86`	`86`	`def test_closing_connection_closes_commands(self, mock_result_set_class):`
`87`	`87`	`# Test once with has_been_closed_server side, once without`