openrewrite-migrate-python 0.20.0__tar.gz → 0.22.0__tar.gz

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrewrite-migrate-python
-Version: 0.20.0
+Version: 0.22.0
 Summary: OpenRewrite recipes for migrating Python codebases to newer Python versions.
 Author-email: "Moderne Inc." <support@moderne.io>
 License: Moderne Proprietary

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "openrewrite-migrate-python"
 description = "OpenRewrite recipes for migrating Python codebases to newer Python versions."
-version = "0.20.0"  # Updated dynamically during release
+version = "0.22.0"  # Updated dynamically during release
 authors = [{ name = "Moderne Inc.", email = "support@moderne.io" }]
 license = { text = "Moderne Proprietary" }
 readme = "README.md"

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/src/openrewrite_migrate_python/migrate/__init__.py

@@ -73,6 +73,7 @@ from .locale_deprecations import ReplaceLocaleResetlocale
 from .locale_getdefaultlocale_deprecation import FindLocaleGetdefaultlocale
 from .macpath_deprecations import FindMacpathModule
 from .mailcap_migrations import FindMailcapModule
+from .mock_assertions import AddMockAssertPrefix
 from .module_cached_attribute import ReplaceModuleCachedAttribute
 from .nntplib_migrations import FindNntplibModule
 from .os_deprecations import FindOsPopen, FindOsSpawn
@@ -85,7 +86,7 @@ from .pep594_system_migrations import (
     FindOssaudiodevModule,
     FindSpwdModule,
 )
-from .pipes_migrations import FindPipesModule
+from .pipes_migrations import FindPipesModule, ReplacePipesQuote
 from .pkgutil_deprecations import ReplacePkgutilFindLoader, ReplacePkgutilGetLoader
 from .profile_deprecations import FindProfileModule
 from .pydantic_deprecations import (
@@ -192,6 +193,7 @@ __all__ = [
     "FindInspectGetargspec",
     "FindMacpathModule",
     "FindMailcapModule",
+    "AddMockAssertPrefix",
     "FindMsilibModule",
     "FindNisModule",
     "FindNntplibModule",
@@ -201,6 +203,7 @@ __all__ = [
     "FindOssaudiodevModule",
     "FindPathlibLinkTo",
     "FindPipesModule",
+    "ReplacePipesQuote",
     "ReplacePlatformPopen",
     "ReplaceReTemplate",
     "FindRemovedModules312",

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/src/openrewrite_migrate_python/migrate/datetime_utc.py

@@ -2,13 +2,14 @@
 Recipes for migrating deprecated datetime.utcnow() and datetime.utcfromtimestamp() calls.
 
 These methods are deprecated in Python 3.12+ because they return naive datetime objects
-that don't carry timezone information. The recommended replacements are:
+that don't carry timezone information. The replacements return the same naive value:
 
-- datetime.utcnow() -> datetime.now(UTC)
-- datetime.utcfromtimestamp(ts) -> datetime.fromtimestamp(ts, UTC)
+- datetime.utcnow() -> datetime.now(UTC).replace(tzinfo=None)
+- datetime.utcfromtimestamp(ts) -> datetime.fromtimestamp(ts, UTC).replace(tzinfo=None)
 
-The replacements are timezone-aware where the deprecated calls were naive, so values already
-stored or compared as naive UTC need migrating alongside the call sites.
+An aware result never equals a naive datetime, raises ``TypeError`` when ordered against or
+subtracted from one, and prints with a ``+00:00`` offset. Making the values aware is a change
+of its own, not part of the upgrade.
 
 See: https://docs.python.org/3/library/datetime.html#datetime.datetime.utcnow
 """
@@ -56,18 +57,18 @@ _DATETIME_MODULE = ("datetime", None)
 
 @lru_cache(maxsize=None)
 def _now_template(utc: str) -> Template:
-    return template(f"{_sel}.now({raw(utc)})")
+    return template(f"{_sel}.now({raw(utc)}).replace(tzinfo=None)")
 
 
 @lru_cache(maxsize=None)
 def _fromtimestamp_template(utc: str) -> Template:
-    return template(f"{_sel}.fromtimestamp({_ts}, {raw(utc)})")
+    return template(f"{_sel}.fromtimestamp({_ts}, {raw(utc)}).replace(tzinfo=None)")
 
 
 class _DatetimeUtcVisitor(PythonVisitor[ExecutionContext]):
-    """Rewrites a naive UTC constructor to its aware counterpart, spelling UTC the way the
-    file's own ``datetime`` imports allow. Those imports also identify the receiver, which a
-    parse that attributes no types still supplies.
+    """Rewrites a deprecated naive UTC constructor to its aware counterpart made naive again,
+    spelling UTC the way the file's own ``datetime`` imports allow. Those imports also
+    identify the receiver, which a parse that attributes no types still supplies.
     """
 
     _target = ""
@@ -131,7 +132,7 @@ class _DatetimeUtcVisitor(PythonVisitor[ExecutionContext]):
 @categorize(_Python312)
 class ReplaceDatetimeUtcNow(Recipe):
     """
-    Replace ``datetime.utcnow()`` with ``datetime.now(UTC)``.
+    Replace ``datetime.utcnow()`` with ``datetime.now(UTC).replace(tzinfo=None)``.
 
     The ``datetime.utcnow()`` method is deprecated in Python 3.12 because it returns
     a naive datetime object that doesn't carry timezone information. This can lead
@@ -144,7 +145,7 @@ class ReplaceDatetimeUtcNow(Recipe):
 
         After:
             from datetime import datetime, UTC
-            now = datetime.now(UTC)
+            now = datetime.now(UTC).replace(tzinfo=None)
     """
 
     @property
@@ -153,15 +154,15 @@ class ReplaceDatetimeUtcNow(Recipe):
 
     @property
     def display_name(self) -> str:
-        return "Replace `datetime.utcnow()` with `datetime.now(UTC)`"
+        return "Replace `datetime.utcnow()` with `datetime.now(UTC).replace(tzinfo=None)`"
 
     @property
     def description(self) -> str:
         return (
             "The `datetime.utcnow()` method is deprecated in Python 3.12. "
-            "Replace it with `datetime.now(UTC)`, importing `UTC` from `datetime` where the "
-            "name is not already in scope. The result is timezone-aware where `utcnow()` was "
-            "naive, so stored or compared values may need migrating alongside the call sites."
+            "Replace it with `datetime.now(UTC).replace(tzinfo=None)`, importing `UTC` from "
+            "`datetime` where the name is not already in scope. The result is the naive UTC "
+            "datetime `utcnow()` returned, so it still compares with other naive values."
         )
 
     @property
@@ -193,7 +194,8 @@ class ReplaceDatetimeUtcNow(Recipe):
 @categorize(_Python312)
 class ReplaceDatetimeUtcFromTimestamp(Recipe):
     """
-    Replace ``datetime.utcfromtimestamp(ts)`` with ``datetime.fromtimestamp(ts, UTC)``.
+    Replace ``datetime.utcfromtimestamp(ts)`` with
+    ``datetime.fromtimestamp(ts, UTC).replace(tzinfo=None)``.
 
     The ``datetime.utcfromtimestamp()`` method is deprecated in Python 3.12 because it
     returns a naive datetime object.
@@ -205,7 +207,7 @@ class ReplaceDatetimeUtcFromTimestamp(Recipe):
 
         After:
             from datetime import datetime, UTC
-            dt = datetime.fromtimestamp(1234567890, UTC)
+            dt = datetime.fromtimestamp(1234567890, UTC).replace(tzinfo=None)
     """
 
     @property
@@ -214,16 +216,16 @@ class ReplaceDatetimeUtcFromTimestamp(Recipe):
 
     @property
     def display_name(self) -> str:
-        return "Replace `datetime.utcfromtimestamp()` with `datetime.fromtimestamp(ts, UTC)`"
+        return "Replace `datetime.utcfromtimestamp()` with `datetime.fromtimestamp(ts, UTC).replace(tzinfo=None)`"
 
     @property
     def description(self) -> str:
         return (
             "The `datetime.utcfromtimestamp()` method is deprecated in Python 3.12. "
-            "Replace it with `datetime.fromtimestamp(ts, UTC)`, importing `UTC` from "
-            "`datetime` where the name is not already in scope. The result is timezone-aware "
-            "where `utcfromtimestamp()` was naive, so stored or compared values may need "
-            "migrating alongside the call sites."
+            "Replace it with `datetime.fromtimestamp(ts, UTC).replace(tzinfo=None)`, importing "
+            "`UTC` from `datetime` where the name is not already in scope. The result is the "
+            "naive UTC datetime `utcfromtimestamp()` returned, so it still compares with other "
+            "naive values."
         )
 
     @property

openrewrite_migrate_python-0.22.0/src/openrewrite_migrate_python/migrate/mock_assertions.py

@@ -0,0 +1,117 @@
+"""
+Recipes for mock assertions that Python 3.12 rejects.
+
+See: https://docs.python.org/3/whatsnew/3.12.html
+"""
+
+from typing import Any, List
+
+from rewrite import ExecutionContext, Preconditions, Recipe, TreeVisitor
+from rewrite.category import CategoryDescriptor
+from rewrite.decorators import categorize
+from rewrite.marketplace import Python
+from rewrite.java.support_types import JavaType, Space
+from rewrite.java.tree import Assert, Block, J, MethodInvocation
+from rewrite.python.preconditions import uses_method
+from rewrite.python.type_utils import is_of_type_with_name
+from rewrite.python.visitor import PythonVisitor
+
+from openrewrite_code_quality_python.codequality._trivia import has_comments
+
+_Python312 = [
+    *Python,
+    CategoryDescriptor(display_name="Migrate"),
+    CategoryDescriptor(display_name="Python 3.12"),
+]
+
+# `called` is a property, so a mock still answers it.
+_UNPREFIXED = ("called_once", "called_with", "called_once_with", "any_call", "has_calls", "not_called")
+
+
+def _is_unprefixed_mock_call(tree: Any) -> bool:
+    if not isinstance(tree, MethodInvocation) or tree.select is None:
+        return False
+    if tree.name.simple_name not in _UNPREFIXED:
+        return False
+    # A mock stored on `self` in `setUp()` has no type the parser sees.
+    receiver = tree.select.type
+    return isinstance(receiver, JavaType.Unknown) or is_of_type_with_name(
+        receiver, True, lambda n: n.startswith(("unittest.mock.", "mock."))
+    )
+
+
+def _prefixed(call: MethodInvocation, prefix: Space) -> MethodInvocation:
+    return call.replace(
+        _prefix=prefix,
+        _name=call.name.replace(_simple_name=f"assert_{call.name.simple_name}", _type=None, _field_type=None),
+        _method_type=None,
+    )
+
+
+@categorize(_Python312)
+class AddMockAssertPrefix(Recipe):
+    """
+    Add the ``assert_`` prefix to mock assertions that lack it.
+
+    Example:
+        Before:
+            def test_cleanup(self):
+                self.callback.called_once_with(42)
+
+        After:
+            def test_cleanup(self):
+                self.callback.assert_called_once_with(42)
+    """
+
+    @property
+    def name(self) -> str:
+        return "org.openrewrite.python.migrate.AddMockAssertPrefix"
+
+    @property
+    def display_name(self) -> str:
+        return "Add the `assert_` prefix to mock assertions that lack it"
+
+    @property
+    def description(self) -> str:
+        return (
+            "Since Python 3.12 a mock raises `AttributeError` for `called_once_with()` and the "
+            "other assertion names without their `assert_` prefix. Before, such a call asserted "
+            "nothing. The prefix is added where the call is a statement of its own inside a "
+            "function, the whole condition of an `assert` without a message, or the only argument "
+            "of `assertTrue()`. The recipe matches the method name, and leaves a receiver whose "
+            "type resolves to a class other than a mock. A test that passed only because nothing "
+            "was asserted fails once it asserts."
+        )
+
+    @property
+    def tags(self) -> List[str]:
+        return ["python", "migration", "3.12", "unittest"]
+
+    def editor(self) -> TreeVisitor[Any, ExecutionContext]:
+        class Visitor(PythonVisitor[ExecutionContext]):
+            def visit_method_invocation(self, method: MethodInvocation, p: ExecutionContext) -> J:
+                m = super().visit_method_invocation(method, p)
+                if not isinstance(m, MethodInvocation) or not isinstance(
+                    self.cursor.parent_tree_cursor().value, Block
+                ):
+                    return m
+                if _is_unprefixed_mock_call(m):
+                    return _prefixed(m, m.prefix)
+                if m.name.simple_name != "assertTrue" or len(m.arguments) != 1:
+                    return m
+                return self._replace_check(m, m.arguments[0])
+
+            def visit_assert(self, assert_: Assert, p: ExecutionContext) -> J:
+                a = super().visit_assert(assert_, p)
+                if not isinstance(a, Assert) or a.detail is not None:
+                    return a
+                return self._replace_check(a, a.condition)
+
+            def _replace_check(self, check: J, condition: Any) -> J:
+                if not _is_unprefixed_mock_call(condition) or has_comments(check.replace(_prefix=Space.EMPTY)):
+                    return check
+                return _prefixed(condition, check.prefix)
+
+        # A gate naming a mock type would drop the unknown receivers the visitor accepts.
+        gate = Preconditions.or_(*(uses_method(f"*..* {name}(..)") for name in _UNPREFIXED))
+        return Preconditions.check(gate, Visitor())

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/src/openrewrite_migrate_python/migrate/pipes_migrations.py

@@ -17,6 +17,7 @@ from rewrite import ExecutionContext, Recipe, TreeVisitor
 from rewrite.category import CategoryDescriptor
 from rewrite.decorators import categorize
 from rewrite.marketplace import Python
+from rewrite.python.recipes import ChangeImport
 from ._module_imports import find_module_imports
 
 # Define category path: Python > Migrate > Python 3.13
@@ -27,6 +28,46 @@ _Python313 = [
 ]
 
 
+@categorize(_Python313)
+class ReplacePipesQuote(Recipe):
+    """
+    Replace ``pipes.quote()`` with ``shlex.quote()``.
+
+    ``pipes.quote`` is ``shlex.quote`` re-exported, and the ``pipes`` module was removed in
+    Python 3.13.
+
+    Example:
+        Before:
+            import pipes
+            quoted = pipes.quote(path)
+
+        After:
+            import shlex
+            quoted = shlex.quote(path)
+    """
+
+    @property
+    def name(self) -> str:
+        return "org.openrewrite.python.migrate.ReplacePipesQuote"
+
+    @property
+    def display_name(self) -> str:
+        return "Replace `pipes.quote()` with `shlex.quote()`"
+
+    @property
+    def description(self) -> str:
+        return (
+            "`pipes.quote()` is `shlex.quote()` under another name, and the `pipes` module was "
+            "removed in Python 3.13. The `pipes` import stays where the file uses another of its "
+            "members."
+        )
+
+    @property
+    def tags(self) -> List[str]:
+        return ["python", "migration", "3.13", "PEP 594", "pipes"]
+
+    def recipe_list(self) -> List[Recipe]:
+        return [ChangeImport(old_module="pipes", old_name="quote", new_module="shlex")]
 
 
 @categorize(_Python313)

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/src/openrewrite_migrate_python/migrate/upgrade_to_python312.py

@@ -62,6 +62,7 @@ class UpgradeToPython312(Recipe):
         from .distutils_deprecations import FindDistutilsUsage
         from .distutils_migrations import ReplaceDistutilsVersion
         from .imp_migrations import FindImpUsage
+        from .mock_assertions import AddMockAssertPrefix
         from .os_deprecations import FindOsPopen, FindOsSpawn
         from .pathlib_deprecations import FindPathlibLinkTo
         from .pkgutil_deprecations import ReplacePkgutilFindLoader, ReplacePkgutilGetLoader
@@ -101,6 +102,8 @@ class UpgradeToPython312(Recipe):
             FindShutilRmtreeOnerror(),
             # sys.last_type/last_value/last_traceback -> sys.last_exc
             ReplaceSysLastExcInfo(),
+            # Mocks reject `called_once_with()` and the other unprefixed assertion names since 3.12
+            AddMockAssertPrefix(),
             # Deprecated in 3.12, removed in 3.16 and 3.17
             ReplaceBitwiseInversionOnBool(),
             FindBitwiseInversionOnBool(),

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/src/openrewrite_migrate_python/migrate/upgrade_to_python313.py

@@ -80,7 +80,7 @@ class UpgradeToPython313(Recipe):
             FindOssaudiodevModule,
             FindSpwdModule,
         )
-        from .pipes_migrations import FindPipesModule
+        from .pipes_migrations import FindPipesModule, ReplacePipesQuote
         from .telnetlib_migrations import FindTelnetlibModule
         from .upgrade_to_python312 import UpgradeToPython312
         from .uu_migrations import FindUuModule
@@ -91,6 +91,8 @@ class UpgradeToPython313(Recipe):
             UpgradeToPython312(),
             # locale.resetlocale() removed in 3.13 (deprecated in 3.11)
             ReplaceLocaleResetlocale(),
+            # Before FindPipesModule, which then marks only the imports still needed
+            ReplacePipesQuote(),
             # PEP 594 "dead batteries" - removed modules (detection only)
             FindAifcModule(),
             FindAudioopModule(),

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/src/openrewrite_migrate_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrewrite-migrate-python
-Version: 0.20.0
+Version: 0.22.0
 Summary: OpenRewrite recipes for migrating Python codebases to newer Python versions.
 Author-email: "Moderne Inc." <support@moderne.io>
 License: Moderne Proprietary

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/src/openrewrite_migrate_python.egg-info/SOURCES.txt

@@ -101,6 +101,7 @@ src/openrewrite_migrate_python/migrate/locale_deprecations.py
 src/openrewrite_migrate_python/migrate/locale_getdefaultlocale_deprecation.py
 src/openrewrite_migrate_python/migrate/macpath_deprecations.py
 src/openrewrite_migrate_python/migrate/mailcap_migrations.py
+src/openrewrite_migrate_python/migrate/mock_assertions.py
 src/openrewrite_migrate_python/migrate/module_cached_attribute.py
 src/openrewrite_migrate_python/migrate/nntplib_migrations.py
 src/openrewrite_migrate_python/migrate/os_deprecations.py
@@ -187,11 +188,13 @@ tests/test_locale_deprecations.py
 tests/test_locale_getdefaultlocale_deprecation.py
 tests/test_macpath_deprecations.py
 tests/test_mailcap_migrations.py
+tests/test_mock_assertions.py
 tests/test_module_cached_attribute.py
 tests/test_os_deprecations.py
 tests/test_pathlib_deprecations.py
 tests/test_pep594_migrations.py
 tests/test_pep594_system_migrations.py
+tests/test_pipes_migrations.py
 tests/test_pkgutil_deprecations.py
 tests/test_platform_deprecations.py
 tests/test_profile_deprecations.py

{openrewrite_migrate_python-0.20.0 → openrewrite_migrate_python-0.22.0}/tests/test_datetime_utc.py

@@ -22,7 +22,7 @@ class TestReplaceDatetimeUtcNow:
                 """,
                 """
                 from datetime import datetime, UTC
-                now = datetime.now(UTC)
+                now = datetime.now(UTC).replace(tzinfo=None)
                 """,
             )
         )
@@ -41,7 +41,7 @@ class TestReplaceDatetimeUtcNow:
                 from datetime import datetime, UTC
                 now = datetime.utcnow(  # wall clock
                 )
-                later = datetime.now(UTC)
+                later = datetime.now(UTC).replace(tzinfo=None)
                 """,
             )
         )
@@ -57,7 +57,7 @@ class TestReplaceDatetimeUtcNow:
                 """,
                 """
                 from datetime import datetime, timezone
-                now = datetime.now(timezone.utc)
+                now = datetime.now(timezone.utc).replace(tzinfo=None)
                 """,
             )
         )
@@ -73,7 +73,7 @@ class TestReplaceDatetimeUtcNow:
                 """,
                 """
                 import datetime
-                now = datetime.datetime.now(datetime.UTC)
+                now = datetime.datetime.now(datetime.UTC).replace(tzinfo=None)
                 """,
             )
         )
@@ -96,7 +96,7 @@ class TestReplaceDatetimeUtcNow:
                 class Job:
                     started: datetime.datetime
                     def restart(self):
-                        return self.started.now(UTC)
+                        return self.started.now(UTC).replace(tzinfo=None)
                 """,
             )
         )
@@ -111,7 +111,7 @@ class TestReplaceDatetimeUtcNow:
                 """,
                 """
                 from datetime import datetime as dt, UTC
-                now = dt.now(UTC)
+                now = dt.now(UTC).replace(tzinfo=None)
                 """,
             )
         )
@@ -131,7 +131,7 @@ class TestReplaceDatetimeUtcNow:
                 from datetime import datetime, timezone
                 from pytz import UTC
 
-                now = datetime.now(timezone.utc)
+                now = datetime.now(timezone.utc).replace(tzinfo=None)
                 """,
             )
         )
@@ -155,7 +155,7 @@ class TestReplaceDatetimeUtcNow:
 
                 def stamp():
                     UTC = "Z"
-                    return UTC, datetime.now(timezone.utc)
+                    return UTC, datetime.now(timezone.utc).replace(tzinfo=None)
                 """,
             )
         )
@@ -191,7 +191,7 @@ class TestReplaceDatetimeUtcNow:
                 def cutoff(max_age):
                     from datetime import datetime, timedelta
 
-                    return datetime.now(UTC) - timedelta(seconds=max_age)
+                    return datetime.now(UTC).replace(tzinfo=None) - timedelta(seconds=max_age)
                 """,
             )
         )
@@ -225,7 +225,7 @@ class TestReplaceDatetimeUtcNow:
                 from datetime import datetime, UTC
                 class Stamp(datetime):
                     pass
-                now = Stamp.now(UTC)
+                now = Stamp.now(UTC).replace(tzinfo=None)
                 """,
             )
         )
@@ -275,7 +275,7 @@ class TestReplaceDatetimeUtcNow:
 
                 obj = MyClass()
                 result = obj.utcnow()
-                now = datetime.now(UTC)
+                now = datetime.now(UTC).replace(tzinfo=None)
                 """,
             )
         )
@@ -283,7 +283,7 @@ class TestReplaceDatetimeUtcNow:
 class TestReplaceDatetimeUtcFromTimestamp:
     """Tests for the ReplaceDatetimeUtcFromTimestamp recipe."""
 
-    def test_appends_utc_argument(self):
+    def test_appends_utc_and_keeps_the_result_naive(self):
         spec = RecipeSpec(recipe=ReplaceDatetimeUtcFromTimestamp())
         spec.rewrite_run(
             python(
@@ -293,7 +293,7 @@ class TestReplaceDatetimeUtcFromTimestamp:
                 """,
                 """
                 from datetime import datetime, UTC
-                dt = datetime.fromtimestamp(1234567890, UTC)
+                dt = datetime.fromtimestamp(1234567890, UTC).replace(tzinfo=None)
                 """,
             )
         )
@@ -312,7 +312,7 @@ class TestReplaceDatetimeUtcFromTimestamp:
                 from datetime import datetime, UTC
                 dt = datetime.utcfromtimestamp(ts  # epoch
                 )
-                other = datetime.fromtimestamp(ts, UTC)
+                other = datetime.fromtimestamp(ts, UTC).replace(tzinfo=None)
                 """,
             )
         )

openrewrite_migrate_python-0.22.0/tests/test_mock_assertions.py

@@ -0,0 +1,86 @@
+from rewrite.test import RecipeSpec, python
+
+from openrewrite_migrate_python.migrate.mock_assertions import AddMockAssertPrefix
+from tests._after_recipe import assert_idempotent_and_parse_shape
+
+_check = assert_idempotent_and_parse_shape(AddMockAssertPrefix())
+
+
+class TestAddMockAssertPrefix:
+
+    def test_prefixes_a_call_whose_result_is_discarded(self):
+        RecipeSpec(recipe=AddMockAssertPrefix()).rewrite_run(
+            python(
+                """
+                from unittest.mock import Mock
+
+                def test_writes(self):
+                    m = Mock()
+                    m.called_once_with(1)
+                    self.popen.stdin.has_calls([])
+                """,
+                """
+                from unittest.mock import Mock
+
+                def test_writes(self):
+                    m = Mock()
+                    m.assert_called_once_with(1)
+                    self.popen.stdin.assert_has_calls([])
+                """,
+                after_recipe=_check,
+            )
+        )
+
+    def test_replaces_a_truth_assertion_of_the_call(self):
+        RecipeSpec(recipe=AddMockAssertPrefix()).rewrite_run(
+            python(
+                """
+                import unittest
+
+                class T(unittest.TestCase):
+                    def test_cleanup(self):
+                        assert self.callback.called_with(42, keyword='value')
+                        self.assertTrue(self.publisher.called_once())
+                """,
+                """
+                import unittest
+
+                class T(unittest.TestCase):
+                    def test_cleanup(self):
+                        self.callback.assert_called_with(42, keyword='value')
+                        self.publisher.assert_called_once()
+                """,
+                after_recipe=_check,
+            )
+        )
+
+    def test_leaves_other_receivers_and_values_that_are_read(self):
+        RecipeSpec(recipe=AddMockAssertPrefix()).rewrite_run(
+            python(
+                """
+                class Recorder:
+                    def has_calls(self, calls): ...
+
+                def test_kept(self):
+                    Recorder().has_calls([])
+                    results.append(self.m.any_call(1))
+                    assert self.adapter.called_once  # a requests_mock property
+                """
+            )
+        )
+
+    def test_keeps_a_check_whose_message_or_comment_would_be_lost(self):
+        RecipeSpec(recipe=AddMockAssertPrefix()).rewrite_run(
+            python(
+                """
+                import unittest
+
+                class T(unittest.TestCase):
+                    def test_kept(self):
+                        assert self.m.called_with(1), "not cleaned up"
+                        self.assertTrue(self.m.called_once(), "not published")
+                        self.assertTrue(  # published once
+                            self.m.called_once())
+                """
+            )
+        )