Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

chore: s/MockFixture/MockerFixture/g #29160

Merged
merged 2 commits into from
Jun 10, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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
Loading