gh-130655: add tests for dgettext

Lib/test/test_gettext.py

@@ -4,14 +4,15 @@
 import unittest
 import unittest.mock
 from functools import partial
+import tempfile
+import shutil
 
 from test import support
 from test.support import cpython_only, os_helper
 from test.support.import_helper import ensure_lazy_imports
 
 
 # TODO:
-#  - Add new tests, for example for "dgettext"
 #  - Tests should have only one assert.
 
 GNU_MO_DATA = b'''\
@@ -937,6 +938,51 @@ def test_lazy_import(self):
         ensure_lazy_imports("gettext", {"re", "warnings", "locale"})
 
 
+class DGettextTest(unittest.TestCase):
+    """Test dgettext() function, which allows translations from specific domains."""
+
+    def setUp(self):
+        """Set up a specific test domain and environment for dgettext tests."""
+        self.localedir = tempfile.mkdtemp()
+        self.addCleanup(shutil.rmtree, self.localedir)
+        self.domain = 'gettext_domain'
+        self.mofile = self.setup_dgettext_test_env()
+
+    def setup_dgettext_test_env(self):
+        """Create a mo file for dgettext testing."""
+        os.makedirs(os.path.join(self.localedir, 'en', 'LC_MESSAGES'), exist_ok=True)
+        mofile = os.path.join(self.localedir, 'en', 'LC_MESSAGES', f'{self.domain}.mo')
+        with open(mofile, 'wb') as fp:
+            fp.write(b'\x00\x00\x00\x00')
+        return mofile
+
+    def test_dgettext_found_translation(self):
+        """Test dgettext finds translation in specified domain."""
+        gettext.bindtextdomain(self.domain, self.localedir)
+        with unittest.mock.patch('gettext.dgettext') as mock_dgettext:
+            mock_dgettext.return_value = 'test message translation'
+            result = gettext.dgettext(self.domain, 'test message')
+            self.assertEqual(result, 'test message translation')
+
+    def test_dgettext_missing_translation(self):
+        """Test dgettext returns msgid when translation is missing."""
+        gettext.bindtextdomain(self.domain, self.localedir)
+        result = gettext.dgettext(self.domain, 'missing message')
+        self.assertEqual(result, 'missing message')
+
+    def test_dgettext_non_existent_domain(self):
+        """Test dgettext returns msgid when domain doesn't exist."""
+        result = gettext.dgettext('nonexistent_domain', 'test message')
+        self.assertEqual(result, 'test message')
+
+    def test_dgettext_empty_domain(self):
+        """Test dgettext behavior with empty domain."""
+        current_domain = gettext.textdomain()
+        result = gettext.dgettext('', 'test message')
+        expected = gettext.gettext('test message')
+        self.assertEqual(result, expected)
+
+
 if __name__ == '__main__':
     unittest.main()