Commit019c7fb

committed

reduce diff of test_closing_connection_closes_commands

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

1 parentfa15730 commit019c7fbCopy full SHA for 019c7fb

File tree

-3

lines changed

-3

lines changed

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -102,7 +102,7 @@ def test_closing_connection_closes_commands(self, mock_thrift_client_class):`
`102`	`102`
`103`	`103`	`forclosedin (True,False):`
`104`	`104`	`withself.subTest(closed=closed):`
`105`		`-#set initial state based on whether the command is already closed`
	`105`	`+#Set initial state based on whether the command is already closed`
`106`	`106`	`initial_state= (`
`107`	`107`	`CommandState.CLOSEDifclosedelseCommandState.SUCCEEDED`
`108`	`108`	`)`
`@@ -114,7 +114,7 @@ def test_closing_connection_closes_commands(self, mock_thrift_client_class):`
`114`	`114`	`mock_execute_response.is_staging_operation=False`
`115`	`115`	`mock_execute_response.command_id=Mock(spec=CommandId)`
`116`	`116`
`117`		`-# Mock the backend that will be used by the real ThriftResultSet`
	`117`	`+# Mock the backend that will be used`
`118`	`118`	`mock_backend=Mock(spec=ThriftDatabricksClient)`
`119`	`119`	`mock_backend.staging_allowed_local_path=None`
`120`	`120`	`mock_thrift_client_class.return_value=mock_backend`
`@@ -132,7 +132,7 @@ def test_closing_connection_closes_commands(self, mock_thrift_client_class):`
`132`	`132`	`# Mock execute_command to return our real result set`
`133`	`133`	`cursor.backend.execute_command=Mock(return_value=real_result_set)`
`134`	`134`
`135`		`-# Execute a command - this should set cursor.active_result_set to our real result set`
	`135`	`+# Execute a command`
`136`	`136`	`cursor.execute("SELECT 1")`
`137`	`137`
`138`	`138`	`# Close the connection`

Comments

(0)