Skip to content

Commit

Permalink
chore: s/MockFixture/MockerFixture/g (#29160)
Browse files Browse the repository at this point in the history
  • Loading branch information
betodealmeida authored and eschutho committed Jul 24, 2024
1 parent 56a40d3 commit 12aabe7
Show file tree
Hide file tree
Showing 32 changed files with 235 additions and 213 deletions.
8 changes: 4 additions & 4 deletions tests/integration_tests/access_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@

import pytest
from flask.ctx import AppContext
from pytest_mock import MockFixture
from pytest_mock import MockerFixture
from sqlalchemy import inspect # noqa: F401

from tests.integration_tests.fixtures.birth_names_dashboard import (
Expand Down Expand Up @@ -90,7 +90,7 @@
)
def test_get_user_id(
app_context: AppContext,
mocker: MockFixture,
mocker: MockerFixture,
username: Optional[str],
user_id: Optional[int],
) -> None:
Expand All @@ -109,7 +109,7 @@ def test_get_user_id(
)
def test_get_username(
app_context: AppContext,
mocker: MockFixture,
mocker: MockerFixture,
username: Optional[str],
) -> None:
mock_g = mocker.patch("superset.utils.core.g", spec={})
Expand All @@ -121,7 +121,7 @@ def test_get_username(
@pytest.mark.parametrize("force", [False, True])
def test_override_user(
app_context: AppContext,
mocker: MockFixture,
mocker: MockerFixture,
username: str,
force: bool,
) -> None:
Expand Down
14 changes: 8 additions & 6 deletions tests/integration_tests/reports/alert_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
import pandas as pd
import pytest
from flask.ctx import AppContext
from pytest_mock import MockFixture
from pytest_mock import MockerFixture

from superset.commands.report.exceptions import AlertQueryError
from superset.reports.models import ReportCreationMethod, ReportScheduleType
Expand Down Expand Up @@ -61,7 +61,7 @@ def test_execute_query_as_report_executor(
creator_name: Optional[str],
config: list[ExecutorType],
expected_result: Union[tuple[ExecutorType, str], Exception],
mocker: MockFixture,
mocker: MockerFixture,
app_context: AppContext,
get_user,
) -> None:
Expand Down Expand Up @@ -99,7 +99,7 @@ def test_execute_query_as_report_executor(


def test_execute_query_succeeded_no_retry(
mocker: MockFixture, app_context: None
mocker: MockerFixture, app_context: None
) -> None:
from superset.commands.report.alert import AlertCommand

Expand All @@ -116,7 +116,7 @@ def test_execute_query_succeeded_no_retry(


def test_execute_query_succeeded_with_retries(
mocker: MockFixture, app_context: None
mocker: MockerFixture, app_context: None
) -> None:
from superset.commands.report.alert import AlertCommand, AlertQueryError

Expand Down Expand Up @@ -147,7 +147,9 @@ def _mocked_execute_query() -> pd.DataFrame:
assert execute_query_mock.call_count == expected_max_retries


def test_execute_query_failed_no_retry(mocker: MockFixture, app_context: None) -> None:
def test_execute_query_failed_no_retry(
mocker: MockerFixture, app_context: None
) -> None:
from superset.commands.report.alert import AlertCommand, AlertQueryTimeout

execute_query_mock = mocker.patch(
Expand All @@ -168,7 +170,7 @@ def _mocked_execute_query() -> None:


def test_execute_query_failed_max_retries(
mocker: MockFixture, app_context: None
mocker: MockerFixture, app_context: None
) -> None:
from superset.commands.report.alert import AlertCommand, AlertQueryError

Expand Down
4 changes: 2 additions & 2 deletions tests/integration_tests/sqla_models_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
import numpy as np
import pandas as pd
from flask.ctx import AppContext
from pytest_mock import MockFixture
from pytest_mock import MockerFixture
from sqlalchemy.sql import text
from sqlalchemy.sql.elements import TextClause

Expand Down Expand Up @@ -856,7 +856,7 @@ def test_none_operand_in_filter(login_as_admin, physical_dataset):
],
)
def test__normalize_prequery_result_type(
mocker: MockFixture,
mocker: MockerFixture,
row: pd.Series,
dimension: str,
result: Any,
Expand Down
16 changes: 10 additions & 6 deletions tests/integration_tests/test_jinja_context.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@

import pytest
from flask.ctx import AppContext
from pytest_mock import MockFixture
from pytest_mock import MockerFixture

import superset.utils.database
from superset.exceptions import SupersetTemplateException
Expand Down Expand Up @@ -109,7 +109,7 @@ def test_template_kwarg_nested_module(app_context: AppContext) -> None:
tp.process_template(template, foo={"bar": datetime})


def test_template_hive(app_context: AppContext, mocker: MockFixture) -> None:
def test_template_hive(app_context: AppContext, mocker: MockerFixture) -> None:
lp_mock = mocker.patch(
"superset.jinja_context.HiveTemplateProcessor.latest_partition"
)
Expand All @@ -121,7 +121,7 @@ def test_template_hive(app_context: AppContext, mocker: MockFixture) -> None:
assert tp.process_template(template) == "the_latest"


def test_template_spark(app_context: AppContext, mocker: MockFixture) -> None:
def test_template_spark(app_context: AppContext, mocker: MockerFixture) -> None:
lp_mock = mocker.patch(
"superset.jinja_context.SparkTemplateProcessor.latest_partition"
)
Expand All @@ -138,7 +138,7 @@ def test_template_spark(app_context: AppContext, mocker: MockFixture) -> None:
assert tp.process_template(template) == "the_latest"


def test_template_trino(app_context: AppContext, mocker: MockFixture) -> None:
def test_template_trino(app_context: AppContext, mocker: MockerFixture) -> None:
lp_mock = mocker.patch(
"superset.jinja_context.TrinoTemplateProcessor.latest_partition"
)
Expand All @@ -155,7 +155,9 @@ def test_template_trino(app_context: AppContext, mocker: MockFixture) -> None:
assert tp.process_template(template) == "the_latest"


def test_template_context_addons(app_context: AppContext, mocker: MockFixture) -> None:
def test_template_context_addons(
app_context: AppContext, mocker: MockerFixture
) -> None:
addons_mock = mocker.patch("superset.jinja_context.context_addons")
addons_mock.return_value = {"datetime": datetime}
maindb = superset.utils.database.get_example_database()
Expand All @@ -164,7 +166,9 @@ def test_template_context_addons(app_context: AppContext, mocker: MockFixture) -
assert tp.process_template(template) == "SELECT '2017-01-01T00:00:00'"


def test_custom_process_template(app_context: AppContext, mocker: MockFixture) -> None:
def test_custom_process_template(
app_context: AppContext, mocker: MockerFixture
) -> None:
"""Test macro defined in custom template processor works."""

mock_dt = mocker.patch(
Expand Down
12 changes: 6 additions & 6 deletions tests/unit_tests/charts/commands/importers/v1/import_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@

import pytest
from flask_appbuilder.security.sqla.models import Role, User
from pytest_mock import MockFixture
from pytest_mock import MockerFixture
from sqlalchemy.orm.session import Session

from superset import security_manager
Expand Down Expand Up @@ -71,7 +71,7 @@ def session_with_schema(session: Session) -> Generator[Session, None, None]:
yield session


def test_import_chart(mocker: MockFixture, session_with_schema: Session) -> None:
def test_import_chart(mocker: MockerFixture, session_with_schema: Session) -> None:
"""
Test importing a chart.
"""
Expand All @@ -93,7 +93,7 @@ def test_import_chart(mocker: MockFixture, session_with_schema: Session) -> None


def test_import_chart_managed_externally(
mocker: MockFixture, session_with_schema: Session
mocker: MockerFixture, session_with_schema: Session
) -> None:
"""
Test importing a chart that is managed externally.
Expand All @@ -115,7 +115,7 @@ def test_import_chart_managed_externally(


def test_import_chart_without_permission(
mocker: MockFixture,
mocker: MockerFixture,
session_with_schema: Session,
) -> None:
"""
Expand Down Expand Up @@ -156,7 +156,7 @@ def test_filter_chart_annotations(session: Session) -> None:


def test_import_existing_chart_without_permission(
mocker: MockFixture,
mocker: MockerFixture,
session_with_data: Session,
) -> None:
"""
Expand Down Expand Up @@ -185,7 +185,7 @@ def test_import_existing_chart_without_permission(


def test_import_existing_chart_with_permission(
mocker: MockFixture,
mocker: MockerFixture,
session_with_data: Session,
) -> None:
"""
Expand Down
4 changes: 2 additions & 2 deletions tests/unit_tests/commands/dataset/test_update.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
from unittest.mock import MagicMock

import pytest
from pytest_mock import MockFixture
from pytest_mock import MockerFixture

from superset import db
from superset.commands.dataset.exceptions import DatasetInvalidError
Expand All @@ -27,7 +27,7 @@


@pytest.mark.usefixture("session")
def test_update_uniqueness_error(mocker: MockFixture) -> None:
def test_update_uniqueness_error(mocker: MockerFixture) -> None:
SqlaTable.metadata.create_all(db.session.get_bind())
database = Database(database_name="my_db", sqlalchemy_uri="sqlite://")
bar = SqlaTable(table_name="bar", schema="foo", database=database)
Expand Down
4 changes: 2 additions & 2 deletions tests/unit_tests/commands/export_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,10 @@
# pylint: disable=invalid-name, unused-argument, import-outside-toplevel

from freezegun import freeze_time
from pytest_mock import MockFixture
from pytest_mock import MockerFixture


def test_export_assets_command(mocker: MockFixture) -> None:
def test_export_assets_command(mocker: MockerFixture) -> None:
"""
Test that all assets are exported correctly.
"""
Expand Down
10 changes: 6 additions & 4 deletions tests/unit_tests/commands/importers/v1/assets_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@

import copy

from pytest_mock import MockFixture
from pytest_mock import MockerFixture
from sqlalchemy.orm.session import Session
from sqlalchemy.sql import select

Expand All @@ -31,7 +31,7 @@
)


def test_import_new_assets(mocker: MockFixture, session: Session) -> None:
def test_import_new_assets(mocker: MockerFixture, session: Session) -> None:
"""
Test that all new assets are imported correctly.
"""
Expand Down Expand Up @@ -63,7 +63,7 @@ def test_import_new_assets(mocker: MockFixture, session: Session) -> None:
assert len(dashboard_ids) == expected_number_of_dashboards


def test_import_adds_dashboard_charts(mocker: MockFixture, session: Session) -> None:
def test_import_adds_dashboard_charts(mocker: MockerFixture, session: Session) -> None:
"""
Test that existing dashboards are updated with new charts.
"""
Expand Down Expand Up @@ -102,7 +102,9 @@ def test_import_adds_dashboard_charts(mocker: MockFixture, session: Session) ->
assert len(dashboard_ids) == expected_number_of_dashboards


def test_import_removes_dashboard_charts(mocker: MockFixture, session: Session) -> None:
def test_import_removes_dashboard_charts(
mocker: MockerFixture, session: Session
) -> None:
"""
Test that existing dashboards are updated without old charts.
"""
Expand Down
6 changes: 3 additions & 3 deletions tests/unit_tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@

import pytest
from _pytest.fixtures import SubRequest
from pytest_mock import MockFixture
from pytest_mock import MockerFixture
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.orm.session import Session
Expand All @@ -39,7 +39,7 @@


@pytest.fixture
def get_session(mocker: MockFixture) -> Callable[[], Session]:
def get_session(mocker: MockerFixture) -> Callable[[], Session]:
"""
Create an in-memory SQLite db.session.to test models.
"""
Expand Down Expand Up @@ -141,7 +141,7 @@ def app_context(app: SupersetApp) -> Iterator[None]:


@pytest.fixture
def full_api_access(mocker: MockFixture) -> Iterator[None]:
def full_api_access(mocker: MockerFixture) -> Iterator[None]:
"""
Allow full access to the API.
Expand Down
12 changes: 7 additions & 5 deletions tests/unit_tests/dao/queries_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from typing import Any

import pytest
from pytest_mock import MockFixture
from pytest_mock import MockerFixture
from sqlalchemy.orm.session import Session

from superset.exceptions import QueryNotFoundException, SupersetCancelQueryException
Expand Down Expand Up @@ -116,7 +116,7 @@ def test_query_dao_get_queries_changed_after(session: Session) -> None:


def test_query_dao_stop_query_not_found(
mocker: MockFixture, app: Any, session: Session
mocker: MockerFixture, app: Any, session: Session
) -> None:
from superset import db
from superset.common.db_query_status import QueryStatus
Expand Down Expand Up @@ -159,7 +159,7 @@ def test_query_dao_stop_query_not_found(


def test_query_dao_stop_query_not_running(
mocker: MockFixture, app: Any, session: Session
mocker: MockerFixture, app: Any, session: Session
) -> None:
from superset import db
from superset.common.db_query_status import QueryStatus
Expand Down Expand Up @@ -198,7 +198,7 @@ def test_query_dao_stop_query_not_running(


def test_query_dao_stop_query_failed(
mocker: MockFixture, app: Any, session: Session
mocker: MockerFixture, app: Any, session: Session
) -> None:
from superset import db
from superset.common.db_query_status import QueryStatus
Expand Down Expand Up @@ -240,7 +240,9 @@ def test_query_dao_stop_query_failed(
assert query.status == QueryStatus.RUNNING


def test_query_dao_stop_query(mocker: MockFixture, app: Any, session: Session) -> None:
def test_query_dao_stop_query(
mocker: MockerFixture, app: Any, session: Session
) -> None:
from superset import db
from superset.common.db_query_status import QueryStatus
from superset.models.core import Database
Expand Down
Loading

0 comments on commit 12aabe7

Please sign in to comment.