This is an automated email from the ASF dual-hosted git repository. hugh pushed a commit to branch rm-conftest in repository https://gitbox.apache.org/repos/asf/superset.git
commit 573ea0901b8d3980df5d46262939f3da9f04c411 Author: hughhhh <[email protected]> AuthorDate: Mon Feb 22 09:57:33 2021 -0500 remove unneeded conftest --- tests/conftest.py | 93 ------------------------------------------------------- 1 file changed, 93 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py deleted file mode 100644 index b854385..0000000 --- a/tests/conftest.py +++ /dev/null @@ -1,93 +0,0 @@ -# Licensed to the Apache Software Foundation (ASF) under one -# or more contributor license agreements. See the NOTICE file -# distributed with this work for additional information -# regarding copyright ownership. The ASF licenses this file -# to you under the Apache License, Version 2.0 (the -# "License"); you may not use this file except in compliance -# with the License. You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, -# software distributed under the License is distributed on an -# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY -# KIND, either express or implied. See the License for the -# specific language governing permissions and limitations -# under the License. -# isort:skip_file -from typing import Any - -import pytest -from sqlalchemy.engine import Engine - -from tests.test_app import app - -from superset import db -from superset.utils.core import get_example_database - - -CTAS_SCHEMA_NAME = "sqllab_test_db" -ADMIN_SCHEMA_NAME = "admin_database" - - [email protected](autouse=True, scope="session") -def setup_sample_data() -> Any: - with app.app_context(): - setup_presto_if_needed() - - from superset.cli import load_test_users_run - - load_test_users_run() - - from superset import examples - - examples.load_css_templates() - - yield - - with app.app_context(): - engine = get_example_database().get_sqla_engine() - - # drop sqlachemy tables - - db.session.commit() - from sqlalchemy.ext import declarative - - sqla_base = declarative.declarative_base() - # uses sorted_tables to drop in proper order without violating foreign constrains - for table in sqla_base.metadata.sorted_tables: - table.__table__.drop() - db.session.commit() - - -def drop_from_schema(engine: Engine, schema_name: str): - schemas = engine.execute(f"SHOW SCHEMAS").fetchall() - if schema_name not in [s[0] for s in schemas]: - # schema doesn't exist - return - tables_or_views = engine.execute(f"SHOW TABLES in {schema_name}").fetchall() - for tv in tables_or_views: - engine.execute(f"DROP TABLE IF EXISTS {schema_name}.{tv[0]}") - engine.execute(f"DROP VIEW IF EXISTS {schema_name}.{tv[0]}") - - -def setup_presto_if_needed(): - backend = app.config["SQLALCHEMY_EXAMPLES_URI"].split("://")[0] - if backend == "presto": - # decrease poll interval for tests - presto_poll_interval = app.config["PRESTO_POLL_INTERVAL"] - extra = f'{{"engine_params": {{"connect_args": {{"poll_interval": {presto_poll_interval}}}}}}}' - database = get_example_database() - database.extra = extra - db.session.commit() - - if backend in {"presto", "hive"}: - database = get_example_database() - engine = database.get_sqla_engine() - drop_from_schema(engine, CTAS_SCHEMA_NAME) - engine.execute(f"DROP SCHEMA IF EXISTS {CTAS_SCHEMA_NAME}") - engine.execute(f"CREATE SCHEMA {CTAS_SCHEMA_NAME}") - - drop_from_schema(engine, ADMIN_SCHEMA_NAME) - engine.execute(f"DROP SCHEMA IF EXISTS {ADMIN_SCHEMA_NAME}") - engine.execute(f"CREATE SCHEMA {ADMIN_SCHEMA_NAME}")
