From 42f614a478799300142e5fb21a69b8db0edd5e00 Mon Sep 17 00:00:00 2001
From: Trenton H <797416+stumpylog@users.noreply.github.com>
Date: Sun, 11 Jan 2026 13:46:32 -0800
Subject: [PATCH] Remove coverage for a few just in case places

---
 .../management/commands/document_consumer.py  |  6 +-
 src/documents/tests/factories.py              | 88 +++++++++++++++++++
 2 files changed, 91 insertions(+), 3 deletions(-)

diff --git a/src/documents/management/commands/document_consumer.py b/src/documents/management/commands/document_consumer.py
index 6ae370db2..22c310406 100644
--- a/src/documents/management/commands/document_consumer.py
+++ b/src/documents/management/commands/document_consumer.py
@@ -351,7 +351,7 @@ def _consume_file(
     if subdirs_as_tags:
         try:
             tag_ids = _tags_from_path(filepath, consumption_dir)
-        except Exception:
+        except Exception:  # pragma: nocover
             logger.exception(f"Error creating tags from path for {filepath}")
 
     # Queue for consumption
@@ -561,10 +561,10 @@ class Command(BaseCommand):
                 elif is_testing:
                     # In testing, use short timeout to check stop flag
                     timeout_ms = testing_timeout_ms
-                else:
+                else:  # pragma: nocover
                     # No pending files, wait indefinitely
                     timeout_ms = 0
 
-            except KeyboardInterrupt:
+            except KeyboardInterrupt:  # pragma: nocover
                 logger.info("Received interrupt, stopping consumer")
                 self.stop_flag.set()
diff --git a/src/documents/tests/factories.py b/src/documents/tests/factories.py
index de41bbd02..1e596d36b 100644
--- a/src/documents/tests/factories.py
+++ b/src/documents/tests/factories.py
@@ -1,8 +1,48 @@
+from typing import Any
+
+import factory
+from django.contrib.auth.models import User
 from factory import Faker
 from factory.django import DjangoModelFactory
 
 from documents.models import Correspondent
 from documents.models import Document
+from documents.models import Tag
+
+
+class UserFactory(DjangoModelFactory):
+    """Factory for creating User instances."""
+
+    class Meta:
+        model = User
+        django_get_or_create = ("username",)
+
+    username = Faker("user_name")
+    email = Faker("email")
+    first_name = Faker("first_name")
+    last_name = Faker("last_name")
+    is_active = True
+    is_staff = False
+    is_superuser = False
+
+    @classmethod
+    def _create(
+        cls,
+        model_class: type[User],
+        *args: Any,
+        **kwargs: Any,
+    ) -> User:
+        """Override create to use set_password for proper password handling."""
+        password = kwargs.pop("password", None)
+        user = super()._create(model_class, *args, **kwargs)
+        if password:
+            user.set_password(password)
+            user.save(update_fields=["password"])
+        return user
+
+    class Params:
+        staff = factory.Trait(is_staff=True)
+        superuser = factory.Trait(is_staff=True, is_superuser=True)
 
 
 class CorrespondentFactory(DjangoModelFactory):
@@ -15,3 +55,51 @@ class CorrespondentFactory(DjangoModelFactory):
 class DocumentFactory(DjangoModelFactory):
     class Meta:
         model = Document
+
+
+class TagFactory(DjangoModelFactory):
+    """
+    Factory for creating Tag instances.
+
+    Examples:
+        tag = TagFactory()
+        tag = TagFactory(name="Invoice")
+        tag = TagFactory(owner=user)
+        tag = TagFactory(inbox=True)
+        child = TagFactory(name="Child", parent=parent_tag)
+    """
+
+    class Meta:
+        model = Tag
+        skip_postgeneration_save = True
+
+    name = Faker("catch_phrase")
+    match = ""  # Intentionally empty - matching patterns are opt-in
+    matching_algorithm = Tag.MATCH_ANY
+    is_insensitive = True
+    owner = None
+    color = Faker("hex_color")
+    is_inbox_tag = False
+    tn_parent = None
+
+    @factory.lazy_attribute
+    def tn_priority(self) -> int:
+        return 0
+
+    @classmethod
+    def _create(
+        cls,
+        model_class: type[Tag],
+        *args: Any,
+        **kwargs: Any,
+    ) -> Tag:
+        """Handle TreeNodeModel parent assignment via set_parent()."""
+        parent = kwargs.pop("parent", None) or kwargs.pop("tn_parent", None)
+        instance = super()._create(model_class, *args, **kwargs)
+        if parent is not None:
+            instance.set_parent(parent)
+        return instance
+
+    class Params:
+        inbox = factory.Trait(is_inbox_tag=True)
+        with_owner = factory.Trait(owner=factory.SubFactory(UserFactory))