Refs #34899 -- Extracted Field.flatchoices to flatten_choices helper function.

Co-authored-by: Natalia Bidart <124304+nessita@users.noreply.github.com>

1 files changed, 42 insertions, 0 deletions

diff --git a/tests/utils_tests/test_choices.py b/tests/utils_tests/test_choices.py
index a2ad5541a4..e3e3766ea9 100644
--- a/tests/utils_tests/test_choices.py
+++ b/tests/utils_tests/test_choices.py
@@ -1,3 +1,4 @@
+import collections.abc
 from unittest import mock
 
 from django.db.models import TextChoices
@@ -5,6 +6,7 @@ from django.test import SimpleTestCase
 from django.utils.choices import (
     BaseChoiceIterator,
     CallableChoiceIterator,
+    flatten_choices,
     normalize_choices,
 )
 from django.utils.translation import gettext_lazy as _
@@ -56,6 +58,46 @@ class ChoiceIteratorTests(SimpleTestCase):
                 self.assertTrue(str(ctx.exception).endswith("index out of range"))
 
 
+class FlattenChoicesTests(SimpleTestCase):
+    def test_empty(self):
+        def generator():
+            yield from ()
+
+        for choices in ({}, [], (), set(), frozenset(), generator(), None, ""):
+            with self.subTest(choices=choices):
+                result = flatten_choices(choices)
+                self.assertIsInstance(result, collections.abc.Generator)
+                self.assertEqual(list(result), [])
+
+    def test_non_empty(self):
+        choices = [
+            ("C", _("Club")),
+            ("D", _("Diamond")),
+            ("H", _("Heart")),
+            ("S", _("Spade")),
+        ]
+        result = flatten_choices(choices)
+        self.assertIsInstance(result, collections.abc.Generator)
+        self.assertEqual(list(result), choices)
+
+    def test_nested_choices(self):
+        choices = [
+            ("Audio", [("vinyl", _("Vinyl")), ("cd", _("CD"))]),
+            ("Video", [("vhs", _("VHS Tape")), ("dvd", _("DVD"))]),
+            ("unknown", _("Unknown")),
+        ]
+        expected = [
+            ("vinyl", _("Vinyl")),
+            ("cd", _("CD")),
+            ("vhs", _("VHS Tape")),
+            ("dvd", _("DVD")),
+            ("unknown", _("Unknown")),
+        ]
+        result = flatten_choices(choices)
+        self.assertIsInstance(result, collections.abc.Generator)
+        self.assertEqual(list(result), expected)
+
+
 class NormalizeFieldChoicesTests(SimpleTestCase):
     expected = [
         ("C", _("Club")),