mirror of
https://github.com/apache/superset.git
synced 2026-04-19 08:04:53 +00:00
chore(sql_lab): Added Unit Test for stop query exception (#17464)
* added unit test * feedback implemented
This commit is contained in:
@@ -1547,6 +1547,28 @@ class TestCore(SupersetTestCase):
|
||||
data = self.get_resp(url)
|
||||
self.assertIn("Error message", data)
|
||||
|
||||
@mock.patch("superset.sql_lab.cancel_query")
|
||||
@mock.patch("superset.views.core.db.session")
|
||||
def test_stop_query_not_implemented(
|
||||
self, mock_superset_db_session, mock_sql_lab_cancel_query
|
||||
):
|
||||
"""
|
||||
Handles stop query when the DB engine spec does not
|
||||
have a cancel query method.
|
||||
"""
|
||||
form_data = {"client_id": "foo"}
|
||||
query_mock = mock.Mock()
|
||||
query_mock.client_id = "foo"
|
||||
query_mock.status = QueryStatus.RUNNING
|
||||
self.login(username="admin")
|
||||
mock_superset_db_session.query().filter_by().one().return_value = query_mock
|
||||
mock_sql_lab_cancel_query.return_value = False
|
||||
rv = self.client.post(
|
||||
"/superset/stop_query/", data={"form_data": json.dumps(form_data)},
|
||||
)
|
||||
|
||||
assert rv.status_code == 422
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user