|
13 | 13 | _filter_session_configuration, |
14 | 14 | ) |
15 | 15 | fromdatabricks.sql.backend.sea.models.baseimportServiceError,StatementStatus |
| 16 | +fromdatabricks.sql.backend.sea.result_setimportSeaResultSet |
16 | 17 | fromdatabricks.sql.backend.typesimportSessionId,CommandId,CommandState,BackendType |
17 | 18 | fromdatabricks.sql.parameters.nativeimportIntegerParameter,TDbsqlParameter |
18 | 19 | fromdatabricks.sql.thrift_api.TCLIServiceimportttypes |
@@ -703,7 +704,7 @@ def test_results_message_to_execute_response_is_staging_operation(self, sea_clie |
703 | 704 | deftest_get_catalogs(self,sea_client,sea_session_id,mock_cursor): |
704 | 705 | """Test the get_catalogs method.""" |
705 | 706 | # Mock the execute_command method |
706 | | -mock_result_set=Mock() |
| 707 | +mock_result_set=Mock(spec=SeaResultSet) |
707 | 708 | withpatch.object( |
708 | 709 | sea_client,"execute_command",return_value=mock_result_set |
709 | 710 | )asmock_execute: |
@@ -735,7 +736,7 @@ def test_get_catalogs(self, sea_client, sea_session_id, mock_cursor): |
735 | 736 | deftest_get_schemas(self,sea_client,sea_session_id,mock_cursor): |
736 | 737 | """Test the get_schemas method with various parameter combinations.""" |
737 | 738 | # Mock the execute_command method |
738 | | -mock_result_set=Mock() |
| 739 | +mock_result_set=Mock(spec=SeaResultSet) |
739 | 740 | withpatch.object( |
740 | 741 | sea_client,"execute_command",return_value=mock_result_set |
741 | 742 | )asmock_execute: |
@@ -884,7 +885,7 @@ def test_get_tables(self, sea_client, sea_session_id, mock_cursor): |
884 | 885 | deftest_get_columns(self,sea_client,sea_session_id,mock_cursor): |
885 | 886 | """Test the get_columns method with various parameter combinations.""" |
886 | 887 | # Mock the execute_command method |
887 | | -mock_result_set=Mock() |
| 888 | +mock_result_set=Mock(spec=SeaResultSet) |
888 | 889 | withpatch.object( |
889 | 890 | sea_client,"execute_command",return_value=mock_result_set |
890 | 891 | )asmock_execute: |
|