Merge pull request #17 from browniebroke/festure/ungettext_lazy

README.rst

@@ -48,6 +48,9 @@ Currently implemented:
 * ``django_codemod.commands.django_40.UNGetTextToNGetTextCommand``: migrate deprecated
   ``ungettext()`` function to ``ngettext()``.
 
+* ``django_codemod.commands.django_40.UNGetTextLazyToNGetTextLazyCommand``: migrate deprecated
+  ``ungettext_lazy()`` function to ``ngettext_lazy()``.
+
 Credits
 -------
 

django_codemod/commands/django_40.py

@@ -115,3 +115,11 @@ class UNGetTextToNGetTextCommand(UGetTextToGetTextCommand):
     DESCRIPTION: str = "Replaces ungettext() by ngettext()."
     old_name = "ungettext"
     new_name = "ngettext"
+
+
+class UNGetTextLazyToNGetTextLazyCommand(UGetTextToGetTextCommand):
+    """Help resolve deprecation of django.utils.translation.ungettext_lazy."""
+
+    DESCRIPTION: str = "Replaces ungettext_lazy() by ngettext_lazy()."
+    old_name = "ungettext_lazy"
+    new_name = "ngettext_lazy"

tests/commands/test_django_40.py

@@ -7,6 +7,7 @@
     UGetTextLazyToGetTextLazyCommand,
     UGetTextNoopToGetTextNoopCommand,
     UNGetTextToNGetTextCommand,
+    UNGetTextLazyToNGetTextLazyCommand,
 )
 
 
@@ -328,3 +329,53 @@ def test_already_imported_substitution(self) -> None:
             result = ngettext(content, plural_content, count)
         """
         self.assertCodemod(before, after)
+
+
+class TestUNGetTextLazyToNGetTextLazyCommand(CodemodTest):
+
+    TRANSFORM = UNGetTextLazyToNGetTextLazyCommand
+
+    def test_noop(self) -> None:
+        """Test when nothing should change."""
+        before = """
+            from django import conf
+            from django.utils import translation
+
+            foo = ngettext_lazy("bar", "bars", count)
+        """
+        after = """
+            from django import conf
+            from django.utils import translation
+
+            foo = ngettext_lazy("bar", "bars", count)
+        """
+
+        self.assertCodemod(before, after)
+
+    def test_simple_substitution(self) -> None:
+        """Check simple use case."""
+        before = """
+            from django.utils.translation import ungettext_lazy
+
+            result = ungettext_lazy(content, plural_content, count)
+        """
+        after = """
+            from django.utils.translation import ngettext_lazy
+
+            result = ngettext_lazy(content, plural_content, count)
+        """
+        self.assertCodemod(before, after)
+
+    def test_already_imported_substitution(self) -> None:
+        """Test case where ngettext_lazy is already in the imports."""
+        before = """
+            from django.utils.translation import ungettext_lazy, ngettext_lazy
+
+            result = ungettext_lazy(content, plural_content, count)
+        """
+        after = """
+            from django.utils.translation import ngettext_lazy
+
+            result = ngettext_lazy(content, plural_content, count)
+        """
+        self.assertCodemod(before, after)