replace mock with unittest.mock (#25774)

> mock is now part of the Python standard library, available as
unittest.mock in Python 3.3 onwards.

## How I Tested These Changes

bk

docs/content/concepts/assets/asset-checks/define-execute-asset-checks.mdx

@@ -143,8 +143,7 @@ To define multiple, similar asset checks, use a factory pattern. In the followin
 
 ```python file=/concepts/assets/asset_checks/factory.py
 from typing import Any, Mapping, Sequence
-
-from mock import MagicMock
+from unittest.mock import MagicMock
 
 from dagster import (
     AssetCheckResult,

docs/content/concepts/testing.mdx

@@ -245,7 +245,7 @@ Mock resources can be provided directly using `materialize_to_memory`:
 
 ```python file=/concepts/ops_jobs_graphs/unit_tests.py startafter=start_materialize_resources endbefore=end_materialize_resources
 from dagster import asset, materialize_to_memory, ConfigurableResource
-import mock
+from unittest import mock
 
 
 class MyServiceResource(ConfigurableResource): ...
@@ -384,7 +384,7 @@ If your ops rely on more complex resources, such as those that build separate cl
 
 ```python file=/concepts/resources/pythonic_resources.py startafter=start_new_resource_testing_with_state_ops endbefore=end_new_resource_testing_with_state_ops dedent=4
 from dagster import ConfigurableResource, op
-import mock
+from unittest import mock
 
 class MyClient:
     ...

examples/docs_beta_snippets/docs_beta_snippets/guides/quality-testing/unit-testing-assets-and-ops/asset-resource.py

@@ -1,4 +1,5 @@
-import mock
+from unittest import mock
+
 from dagster_aws.s3 import S3FileHandle, S3FileManager
 
 import dagster as dg

examples/docs_beta_snippets/docs_beta_snippets/guides/quality-testing/unit-testing-assets-and-ops/op-resource.py

@@ -1,4 +1,5 @@
-import mock
+from unittest import mock
+
 from dagster_aws.s3 import S3FileHandle, S3FileManager
 
 import dagster as dg

examples/docs_snippets/docs_snippets/concepts/assets/asset_checks/factory.py

@@ -1,6 +1,5 @@
 from typing import Any, Mapping, Sequence
-
-from mock import MagicMock
+from unittest.mock import MagicMock
 
 from dagster import (
     AssetCheckResult,

examples/docs_snippets/docs_snippets/concepts/assets/graph_backed_asset.py

@@ -9,7 +9,7 @@
     Definitions,
     OpExecutionContext,
 )
-from mock import MagicMock
+from unittest.mock import MagicMock
 
 
 def create_db_connection():

examples/docs_snippets/docs_snippets/concepts/ops_jobs_graphs/unit_tests.py

@@ -172,7 +172,7 @@ def context_op(context: OpExecutionContext):
 # start_test_op_with_resource_client
 
 from dagster import ConfigurableResource
-import mock
+from unittest import mock
 
 
 class Client:
@@ -407,7 +407,7 @@ def test_data_assets():
 
 # start_materialize_resources
 from dagster import asset, materialize_to_memory, ConfigurableResource
-import mock
+from unittest import mock
 
 
 class MyServiceResource(ConfigurableResource): ...

examples/docs_snippets/docs_snippets/concepts/resources/pythonic_resources.py

@@ -700,7 +700,7 @@ def new_resource_testing_with_state_ops() -> None:
     # start_new_resource_testing_with_state_ops
 
     from dagster import ConfigurableResource, op
-    import mock
+    from unittest import mock
 
     class MyClient:
         ...

examples/docs_snippets/docs_snippets/integrations/airbyte/airbyte.py

@@ -110,7 +110,7 @@ def scope_airbyte_cloud_manual_config():
 
 
 def scope_add_downstream_assets():
-    import mock
+    from unittest import mock
 
     with mock.patch("dagster_snowflake_pandas.SnowflakePandasIOManager"):
         # start_add_downstream_assets
@@ -158,7 +158,7 @@ def stargazers_file(stargazers: pd.DataFrame):
 
 
 def scope_add_downstream_assets_w_deps():
-    import mock
+    from unittest import mock
 
     with mock.patch("dagster_snowflake.SnowflakeResource"):
         # start_with_deps_add_downstream_assets
@@ -209,7 +209,7 @@ def stargazers_file(snowflake: SnowflakeResource):
 
 
 def scope_add_downstream_assets_cloud():
-    import mock
+    from unittest import mock
 
     with mock.patch("dagster_snowflake_pandas.SnowflakePandasIOManager"):
         # start_add_downstream_assets_cloud
@@ -263,7 +263,7 @@ def stargazers_file(stargazers: pd.DataFrame):
 
 
 def scope_add_downstream_assets_cloud_with_deps():
-    import mock
+    from unittest import mock
 
     with mock.patch("dagster_snowflake.SnowflakeResource"):
         # start_with_deps_add_downstream_assets_cloud

examples/docs_snippets/docs_snippets/integrations/fivetran/fivetran.py

@@ -107,7 +107,7 @@ def scope_schedule_assets():
 
 
 def scope_add_downstream_assets():
-    import mock
+    from unittest import mock
 
     with mock.patch("dagster_snowflake_pandas.SnowflakePandasIOManager"):
         # start_add_downstream_assets

examples/docs_snippets/docs_snippets_tests/concepts_tests/assets_tests/test_asset_checks.py

@@ -1,10 +1,10 @@
 from datetime import datetime
 from importlib import resources
+from unittest import mock
+from unittest.mock import MagicMock
 
-import mock
 import pytest
 from dagster_tests.cli_tests.command_tests.assets import fail_asset
-from mock import MagicMock
 
 from dagster import Definitions
 from dagster._core.definitions.asset_check_spec import AssetCheckKey

examples/docs_snippets/docs_snippets_tests/concepts_tests/assets_tests/test_asset_config.py

@@ -1,4 +1,4 @@
-import mock
+from unittest import mock
 
 from docs_snippets.concepts.assets.asset_config import (
     MyDownstreamAssetConfig,

examples/docs_snippets/docs_snippets_tests/concepts_tests/assets_tests/test_asset_testing.py

@@ -1,4 +1,4 @@
-import mock
+from unittest import mock
 
 
 def test_uses_resource() -> None:

examples/docs_snippets/docs_snippets_tests/concepts_tests/resources_tests/test_pythonic_resources.py

@@ -1,6 +1,6 @@
 from typing import Any
+from unittest import mock
 
-import mock
 import pytest
 
 from dagster._core.definitions.run_config import RunConfig

examples/experimental/dagster-airlift/dagster_airlift_tests/unit_tests/core_tests/test_load_defs.py

@@ -1,7 +1,7 @@
 from pathlib import Path
 from typing import cast
+from unittest import mock
 
-import mock
 from dagster import (
     AssetKey,
     AssetsDefinition,

examples/experimental/dagster-airlift/dagster_airlift_tests/unit_tests/core_tests/test_sensor.py

@@ -1,7 +1,7 @@
 from datetime import datetime, timedelta, timezone
 from typing import Optional, Sequence
+from unittest import mock
 
-import mock
 import pytest
 from dagster import (
     AssetCheckKey,

examples/experimental/dagster-airlift/dagster_airlift_tests/unit_tests/mwaa_tests/test_mwaa_auth.py

@@ -1,5 +1,6 @@
+from unittest import mock
+
 import boto3
-import mock
 from dagster_airlift.mwaa import MwaaSessionAuthBackend
 
 

examples/experimental/dagster-blueprints/dagster_blueprints_tests/test_cli/test_blueprint_command.py

@@ -3,8 +3,8 @@
 from pathlib import Path
 from tempfile import TemporaryDirectory
 from typing import List
+from unittest import mock
 
-import mock
 import pytest
 from click.testing import CliRunner
 

examples/project_fully_featured/setup.py

@@ -24,7 +24,6 @@
         "dbt-duckdb",
         "dbt-snowflake",
         "duckdb!=0.3.3, <= 6.0.0",  # missing wheels
-        "mock",
         "pandas",
         "pyarrow>=4.0.0",
         "pyspark",

python_modules/dagster-graphql/dagster_graphql_tests/graphql/test_asset_backfill.py

@@ -1,6 +1,6 @@
 from typing import Optional, Tuple
+from unittest import mock
 
-import mock
 from dagster import (
     AssetKey,
     DailyPartitionsDefinition,

python_modules/dagster-graphql/dagster_graphql_tests/graphql/test_nux.py

@@ -1,4 +1,5 @@
-import mock
+from unittest import mock
+
 from dagster_graphql.test.utils import execute_dagster_graphql
 
 SET_NUX_SEEN_MUTATION = """

python_modules/dagster-test/dagster_test_tests/test_toys.py

@@ -1,4 +1,5 @@
-import mock
+from unittest import mock
+
 import pytest
 from dagster import (
     DagsterEvent,

python_modules/dagster/dagster_tests/asset_defs_tests/test_asset_deps_error_informative.py

@@ -2,8 +2,8 @@
 import sys
 import time
 from typing import List
+from unittest import mock
 
-import mock
 import pytest
 from dagster import AssetIn, AssetKey, Definitions, asset
 from dagster._core.definitions.resolved_asset_deps import resolve_similar_asset_names

python_modules/dagster/dagster_tests/asset_defs_tests/test_asset_selection_error_informative.py

@@ -1,7 +1,7 @@
 import re
 import sys
+from unittest import mock
 
-import mock
 import pytest
 from dagster import AssetSelection, Definitions, asset, define_asset_job
 from dagster._core.errors import DagsterInvalidSubsetError

python_modules/dagster/dagster_tests/cli_tests/command_tests/test_cli_commands.py

@@ -4,8 +4,8 @@
 import tempfile
 from contextlib import contextmanager
 from typing import ContextManager, NoReturn, Optional, Tuple
+from unittest import mock
 
-import mock
 import pytest
 from click.testing import CliRunner
 from dagster import (

python_modules/dagster/dagster_tests/cli_tests/command_tests/test_schedule_commands.py

@@ -1,7 +1,7 @@
 import re
+from unittest import mock
 
 import click
-import mock
 import pytest
 from click.testing import CliRunner
 from dagster._cli.schedule import (

python_modules/dagster/dagster_tests/cli_tests/command_tests/test_sensor_commands.py

@@ -1,7 +1,7 @@
 import re
+from unittest import mock
 
 import click
-import mock
 import pytest
 from click.testing import CliRunner
 from dagster._cli.sensor import (

python_modules/dagster/dagster_tests/cli_tests/test_api_commands.py

@@ -1,6 +1,6 @@
 import os
+from unittest import mock
 
-import mock
 import pytest
 from click.testing import CliRunner
 from dagster import DagsterEventType, job, op, reconstructable

python_modules/dagster/dagster_tests/core_tests/execution_tests/test_instance_concurrency_context.py

@@ -1,7 +1,7 @@
 import time
 from contextlib import ExitStack
+from unittest import mock
 
-import mock
 import pytest
 from dagster import job, op
 from dagster._core.execution.plan.instance_concurrency_context import (

python_modules/dagster/dagster_tests/core_tests/hook_tests/test_hook_invocation.py

@@ -1,6 +1,6 @@
 import re
+from unittest import mock
 
-import mock
 import pytest
 from dagster import (
     DagsterInstance,

python_modules/dagster/dagster_tests/core_tests/resource_tests/pythonic_resources/test_general_pythonic_resources.py

@@ -1,8 +1,8 @@
 import enum
 from abc import ABC, abstractmethod
 from typing import List, Mapping, Optional
+from unittest import mock
 
-import mock
 import pytest
 from dagster import (
     AssetExecutionContext,

python_modules/dagster/dagster_tests/core_tests/test_data_time.py

@@ -2,8 +2,8 @@
 import random
 from collections import defaultdict
 from typing import List, NamedTuple, Optional
+from unittest import mock
 
-import mock
 import pytest
 from dagster import (
     AssetKey,

python_modules/dagster/dagster_tests/core_tests/test_telemetry_upload.py

@@ -1,7 +1,7 @@
 import logging
 import os
+from unittest import mock
 
-import mock
 import pytest
 import responses
 from click.testing import CliRunner

python_modules/dagster/dagster_tests/daemon_sensor_tests/test_sensor_failure_recovery.py

@@ -1,4 +1,5 @@
 import multiprocessing
+from unittest.mock import patch
 
 import pytest
 from dagster._core.definitions.run_request import InstigatorType
@@ -19,7 +20,6 @@
 from dagster._seven import IS_WINDOWS
 from dagster._time import create_datetime, get_timezone
 from dagster._vendored.dateutil.relativedelta import relativedelta
-from mock import patch
 
 from dagster_tests.daemon_sensor_tests.conftest import create_workspace_load_target
 from dagster_tests.daemon_sensor_tests.test_sensor_run import wait_for_all_runs_to_start

python_modules/dagster/dagster_tests/daemon_sensor_tests/test_sensor_run.py

@@ -8,6 +8,7 @@
 from contextlib import ExitStack
 from typing import Any
 from unittest import mock
+from unittest.mock import patch
 
 import pytest
 from dagster import (
@@ -94,7 +95,6 @@
 from dagster._record import copy
 from dagster._time import create_datetime, get_current_datetime
 from dagster._vendored.dateutil.relativedelta import relativedelta
-from mock import patch
 
 from dagster_tests.daemon_sensor_tests.conftest import create_workspace_load_target
 

python_modules/dagster/dagster_tests/daemon_tests/test_backfill.py

@@ -4,9 +4,9 @@
 import string
 import sys
 import time
+from unittest import mock
 
 import dagster._check as check
-import mock
 import pytest
 from dagster import (
     AllPartitionMapping,

python_modules/dagster/dagster_tests/definitions_tests/declarative_automation_tests/scenario_utils/base_scenario.py

@@ -18,9 +18,9 @@
     Tuple,
     Union,
 )
+from unittest import mock
 
 import dagster._check as check
-import mock
 import pytest
 from dagster import (
     AssetIn,

python_modules/dagster/dagster_tests/definitions_tests/declarative_automation_tests/scenario_utils/scenario_state.py

@@ -7,8 +7,8 @@
 from contextlib import contextmanager
 from dataclasses import dataclass, field
 from typing import AbstractSet, Iterable, NamedTuple, Optional, Sequence, Union, cast
+from unittest import mock
 
-import mock
 from dagster import (
     AssetExecutionContext,
     AssetKey,