def test_fetch_data(self, mock_pyodbc_rows_to_tuples): data = [(1, "foo")] with mock.patch.object(BaseEngineSpec, "fetch_data", return_value=data) as mock_fetch: result = MssqlEngineSpec.fetch_data(None, 0) mock_pyodbc_rows_to_tuples.assert_called_once_with(data) self.assertEqual(result, "converted")
def test_fetch_data(app_context: AppContext) -> None: from superset.db_engine_specs.base import BaseEngineSpec from superset.db_engine_specs.mssql import MssqlEngineSpec with mock.patch.object( MssqlEngineSpec, "pyodbc_rows_to_tuples", return_value="converted", ) as mock_pyodbc_rows_to_tuples: data = [(1, "foo")] with mock.patch.object(BaseEngineSpec, "fetch_data", return_value=data): result = MssqlEngineSpec.fetch_data(None, 0) mock_pyodbc_rows_to_tuples.assert_called_once_with(data) assert result == "converted"