test: fix flaky Python unit tests (#12253)

This commit is contained in:
Karol Kostrzewa
2021-01-06 01:40:01 +01:00
committed by GitHub
parent 1b908ab9f1
commit 1a2680d4a1
6 changed files with 24 additions and 2 deletions

View File

@@ -179,13 +179,16 @@ class TestQueryApi(SupersetTestCase):
"""
admin = self.get_user("admin")
client_id = self.get_random_string()
self.insert_query(get_example_database().id, admin.id, client_id)
query = self.insert_query(get_example_database().id, admin.id, client_id)
max_id = db.session.query(func.max(Query.id)).scalar()
self.login(username="admin")
uri = f"api/v1/query/{max_id + 1}"
rv = self.client.get(uri)
self.assertEqual(rv.status_code, 404)
db.session.delete(query)
db.session.commit()
def test_get_query_no_data_access(self):
"""
Query API: Test get query without data access

View File

@@ -534,6 +534,8 @@ class TestSavedQueryApi(SupersetTestCase):
uri = f"api/v1/saved_query/{max_id + 1}"
rv = self.client.get(uri)
assert rv.status_code == 404
db.session.delete(query)
db.session.commit()
def test_create_saved_query(self):
"""