mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2026-10-02 22:30:31 +00:00
Trenton H
e50bcc2efb
* Chore: Refresh the mypy and pyrefly baselines * Chore: Type factory calls as the model they build for mypy factory-boy leaves Factory() unannotated, so mypy took UserFactory() to be a factory instance rather than a User. A shared TypedModelFactory base declares the call as returning the model for type checking only. * Refresh again after rebase
100 lines
2.7 KiB
Python
100 lines
2.7 KiB
Python
"""
|
|
Factory-boy factories for documents app models.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import factory
|
|
from django.contrib.auth.models import User
|
|
|
|
from documents.models import Correspondent
|
|
from documents.models import Document
|
|
from documents.models import DocumentType
|
|
from documents.models import MatchingModel
|
|
from documents.models import PaperlessTask
|
|
from documents.models import StoragePath
|
|
from documents.models import Tag
|
|
from paperless_testing.typed_factory import TypedModelFactory
|
|
|
|
|
|
class CorrespondentFactory(TypedModelFactory[Correspondent]):
|
|
class Meta:
|
|
model = Correspondent
|
|
|
|
name = factory.Sequence(lambda n: f"{factory.Faker('company')} {n}")
|
|
match = ""
|
|
matching_algorithm = MatchingModel.MATCH_NONE
|
|
|
|
|
|
class DocumentTypeFactory(TypedModelFactory[DocumentType]):
|
|
class Meta:
|
|
model = DocumentType
|
|
|
|
name = factory.Sequence(lambda n: f"{factory.Faker('bs')} {n}")
|
|
match = ""
|
|
matching_algorithm = MatchingModel.MATCH_NONE
|
|
|
|
|
|
class TagFactory(TypedModelFactory[Tag]):
|
|
class Meta:
|
|
model = Tag
|
|
|
|
name = factory.Sequence(lambda n: f"{factory.Faker('word')} {n}")
|
|
match = ""
|
|
matching_algorithm = MatchingModel.MATCH_NONE
|
|
is_inbox_tag = False
|
|
|
|
|
|
class StoragePathFactory(TypedModelFactory[StoragePath]):
|
|
class Meta:
|
|
model = StoragePath
|
|
|
|
name = factory.Sequence(
|
|
lambda n: f"{factory.Faker('file_path', depth=2, extension='')} {n}",
|
|
)
|
|
path = factory.LazyAttribute(lambda o: f"{o.name}/{{title}}")
|
|
match = ""
|
|
matching_algorithm = MatchingModel.MATCH_NONE
|
|
|
|
|
|
class DocumentFactory(TypedModelFactory[Document]):
|
|
class Meta:
|
|
model = Document
|
|
|
|
title = factory.Faker("sentence", nb_words=4)
|
|
checksum = factory.Faker("sha256")
|
|
content = factory.Faker("paragraph")
|
|
correspondent = None
|
|
document_type = None
|
|
storage_path = None
|
|
|
|
|
|
class UserFactory(TypedModelFactory[User]):
|
|
class Meta:
|
|
model = User
|
|
|
|
username = factory.Sequence(lambda n: f"user{n}")
|
|
is_staff = False
|
|
is_superuser = False
|
|
# Hashing a real password costs about half a second per user, and no test
|
|
# authenticates with its password. None gives the unusable password
|
|
# create_user() gives.
|
|
password = factory.django.Password(None)
|
|
|
|
class Params:
|
|
superuser = factory.Trait(is_staff=True, is_superuser=True)
|
|
staff = factory.Trait(is_staff=True)
|
|
|
|
|
|
class PaperlessTaskFactory(TypedModelFactory[PaperlessTask]):
|
|
class Meta:
|
|
model = PaperlessTask
|
|
|
|
task_id = factory.Faker("uuid4")
|
|
task_type = PaperlessTask.TaskType.CONSUME_FILE
|
|
trigger_source = PaperlessTask.TriggerSource.WEB_UI
|
|
status = PaperlessTask.Status.PENDING
|
|
input_data = factory.LazyFunction(dict)
|
|
result_data = None
|
|
acknowledged = False
|