mirror of
https://github.com/apache/superset.git
synced 2026-04-19 08:04:53 +00:00
refactor(test): add login_as_admin in global conftest (#20703)
This commit is contained in:
@@ -50,33 +50,31 @@ from tests.integration_tests.fixtures.world_bank_dashboard import (
|
||||
from .base_tests import SupersetTestCase
|
||||
|
||||
|
||||
def delete_imports():
|
||||
with app.app_context():
|
||||
# Imported data clean up
|
||||
session = db.session
|
||||
for slc in session.query(Slice):
|
||||
if "remote_id" in slc.params_dict:
|
||||
session.delete(slc)
|
||||
for dash in session.query(Dashboard):
|
||||
if "remote_id" in dash.params_dict:
|
||||
session.delete(dash)
|
||||
for table in session.query(SqlaTable):
|
||||
if "remote_id" in table.params_dict:
|
||||
session.delete(table)
|
||||
session.commit()
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True, scope="module")
|
||||
def clean_imports():
|
||||
yield
|
||||
delete_imports()
|
||||
|
||||
|
||||
class TestImportExport(SupersetTestCase):
|
||||
"""Testing export import functionality for dashboards"""
|
||||
|
||||
@classmethod
|
||||
def delete_imports(cls):
|
||||
with app.app_context():
|
||||
# Imported data clean up
|
||||
session = db.session
|
||||
for slc in session.query(Slice):
|
||||
if "remote_id" in slc.params_dict:
|
||||
session.delete(slc)
|
||||
for dash in session.query(Dashboard):
|
||||
if "remote_id" in dash.params_dict:
|
||||
session.delete(dash)
|
||||
for table in session.query(SqlaTable):
|
||||
if "remote_id" in table.params_dict:
|
||||
session.delete(table)
|
||||
session.commit()
|
||||
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
cls.delete_imports()
|
||||
|
||||
@classmethod
|
||||
def tearDownClass(cls):
|
||||
cls.delete_imports()
|
||||
|
||||
def create_slice(
|
||||
self,
|
||||
name,
|
||||
|
||||
Reference in New Issue
Block a user