Update tests/server/tasks/test_id_generator.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: 2 people

tests/server/tasks/test_id_generator.py

@@ -96,17 +96,20 @@ def test_generate_produces_unique_ids(self):
         # All IDs should be unique
         assert len(ids) == len(set(ids))
 
-    def test_generate_with_none_context(self):
-        """Test that generate works with context set to None."""
+    @pytest.mark.parametrize(
+        'context_arg',
+        [
+            None,
+            IDGeneratorContext(),
+        ],
+        ids=[
+            'none_context',
+            'empty_context',
+        ],
+    )
+    def test_generate_works_with_various_contexts(self, context_arg):
+        """Test that generate works with various context inputs."""
         generator = UUIDGenerator()
-        result = generator.generate(None)
-        assert isinstance(result, str)
-        uuid.UUID(result)
-
-    def test_generate_with_empty_context(self):
-        """Test that generate works with an empty context."""
-        generator = UUIDGenerator()
-        context = IDGeneratorContext()
-        result = generator.generate(context)
+        result = generator.generate(context_arg)
         assert isinstance(result, str)
         uuid.UUID(result)