def test__get_subjects(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
assert len(doc._get_subjects(resource, ["fast-forms"])) == 0
assert len(doc._get_subjects(resource, ["fast-topic"])) == 2
def test__get_classifications(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
assert len(doc._get_classifications(resource, "rt-ppt")) == 0
assert len(doc._get_classifications(resource, "rt-tt")) == 0
assert len(doc._get_classifications(resource, "rt-pt")) == 0
def test__get_resource(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
assert doc._get_resource(resource) == resource
paratext = Resource.paratext_from_gsx_entry(entry_original, resource)
assert doc._get_resource(paratext) == resource
def test_prepare_date_display(self, resource):
doc = ResourceDocument()
assert doc.prepare_date_display(resource) is None
resource.date = Date(date_display="1971")
resource.date.save()
assert doc.prepare_date_display(resource) is not None
def test_prepare_summary(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
assert len(doc.prepare_summary(resource)) == 1
resource.summary = "resource summary"
assert len(doc.prepare_summary(resource)) == 2
def test_prepare_subjects(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
paratext = Resource.paratext_from_gsx_entry(entry_original, resource)
assert len(doc.prepare_subjects(resource)) == 2
paratext.subjects.add(search_term_or_none("fast-topic", "Operas"))
assert len(doc.prepare_subjects(resource)) == 4
def test_prepare_classifications_paratext(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
paratext = Resource.paratext_from_gsx_entry(entry_original, resource)
assert len(doc.prepare_classifications_paratext(resource)) == 0
paratext.classifications.add(
Classification(edition=search_term_or_none("rt-pt", "Preface")),
bulk=False,
)
assert len(doc.prepare_classifications_paratext(resource)) == 3
def test_prepare_form_genre(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
paratext = Resource.paratext_from_gsx_entry(entry_original, resource)
assert len(doc.prepare_form_genre(resource)) == 0
resource.subjects.add(search_term_or_none("fast-forms", "History"))
assert len(doc.prepare_form_genre(resource)) == 2
paratext.subjects.add(search_term_or_none("fast-forms", "Periodicals"))
assert len(doc.prepare_form_genre(resource)) == 4
def test_prepare_title(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
paratext = Resource.paratext_from_gsx_entry(entry_original, resource)
assert len(doc.prepare_title(resource)) == 1
paratext.title.main_title = "a different title"
paratext.title.save()
paratext.save()
assert len(doc.prepare_title(resource)) == 2
def test_prepare_places(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
assert "fictional_place" not in doc.prepare_places(resource)[0]
place = resource.places.first()
place.fictional_place = "mordor"
place.save()
prepared = doc.prepare_places(resource)[0]
assert "(" in prepared["place"]["address"]
assert prepared["fictional_place"] is not None
def test_prepare_languages(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
paratext = Resource.paratext_from_gsx_entry(entry_original, resource)
assert len(doc.prepare_languages(resource)) == 2
paratext.languages.add(
ResourceLanguage(
language=search_term_or_none("iso639-2", "english")),
bulk=False,
)
assert len(doc.prepare_languages(resource)) == 4
def test_prepare_classifications_translation(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
paratext = Resource.paratext_from_gsx_entry(entry_original, resource)
assert len(doc.prepare_classifications_translation(resource)) == 0
resource.classifications.add(
Classification(edition=search_term_or_none("rt-tt", "Integral")),
bulk=False,
)
assert len(doc.prepare_classifications_translation(resource)) == 2
paratext.classifications.add(
Classification(edition=search_term_or_none("rt-tt", "Partial")),
bulk=False,
)
assert len(doc.prepare_classifications_translation(resource)) == 4
def test_prepare_year(self, resource):
doc = ResourceDocument()
prepared_data = doc.prepare_year(resource)
assert prepared_data is None
resource.date = Date(date_display="1971")
resource.date.save()
prepared_data = doc.prepare_year(resource)
assert prepared_data is not None
assert prepared_data == [1971]
resource.date = Date(date_display="1971/1972")
resource.date.save()
prepared_data = doc.prepare_year(resource)
assert prepared_data is not None
assert len(prepared_data) == 2
def test_prepare_contributions(self, entry_original, person):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
paratext = Resource.paratext_from_gsx_entry(entry_original, resource)
assert len(doc.prepare_contributions(resource)) == 3
person.name = "Anonymous Badger"
person.save()
contribution = Contribution(resource=resource, agent=person)
contribution.save()
contribution.roles.add(search_term_or_none("wikidata", "bookseller"))
paratext.contributions.add(contribution, bulk=False)
assert len(doc.prepare_contributions(resource)) == 4
contribution = doc.prepare_contributions(resource)[2]
assert contribution["agent"]["name"] == "Anonymous"
def test_prepare_classifications_printing_publishing(self, entry_original):
doc = ResourceDocument()
resource = Resource.from_gsx_entry(entry_original)
paratext = Resource.paratext_from_gsx_entry(entry_original, resource)
assert len(
doc.prepare_classifications_printing_publishing(resource)) == 0
resource.classifications.add(
Classification(edition=search_term_or_none("rt-ppt", "Forgeries")),
bulk=False,
)
assert len(
doc.prepare_classifications_printing_publishing(resource)) == 2
paratext.classifications.add(
Classification(edition=search_term_or_none("rt-ppt", "Piracies")),
bulk=False,
)
assert len(
doc.prepare_classifications_printing_publishing(resource)) == 4
def test_get_instances_from_related(self, resource_for_search):
resource = resource_for_search
date_display = "2025"
search = ResourceDocument.search().query("match",
date_display=date_display)
assert len(search.execute()) == 0
resource.date.date_display = date_display
resource.date.save()
search = ResourceDocument.search().query("match",
date_display=date_display)
assert len(search.execute()) == 1
label = "pytest"
search = ResourceDocument.search().query("term", subjects__label=label)
assert len(search.execute()) == 0
subject = resource.subjects.first()
subject.label = label
subject.save()
search = ResourceDocument.search().query("term", subjects__label=label)
assert len(search.execute()) == subject.resources.count()
search = ResourceDocument.search().query(
"match_phrase", title=resource.title.main_title)
assert len(search.execute()) == resource.title.resources.count()
title = resource.title
title.main_title = "radical translations"
title.save()
search = ResourceDocument.search().query(
"match_phrase", title=resource.title.main_title)
assert len(search.execute()) == title.resources.count()
label = "pytest"
search = ResourceDocument.search().query(
"term", classifications__edition__label=label)
assert len(search.execute()) == 0
edition = resource.classifications.first().edition
edition.label = label
edition.save()
search = ResourceDocument.search().query(
"term", classifications__edition__label=label)
assert len(search.execute()) == edition.resources.count()
contribution = resource.contributions.first()
agent = contribution.agent
search = ResourceDocument.search().query(
"match", contributions__agent__name=agent.name)
assert len(search.execute()) >= agent.contributed_to.count()
agent.name = "change agent display name"
agent.save()
contribution.save()
search = ResourceDocument.search().query(
"match", contributions__agent__name=agent.name)
assert len(search.execute()) >= agent.contributed_to.count()
label = "flemish"
search = ResourceDocument.search().query(
"term", languages_language__label=label)
assert len(search.execute()) == 0
language = resource.languages.first().language
language.label = label
language.save()
search = ResourceDocument.search().query(
"term", languages_language__label=label)
assert len(search.execute()) == language.resources.count()
label = "nowhere"
search = ResourceDocument.search().query("match",
places__fictional_place=label)
assert len(search.execute()) == 0
rp = resource.places.first()
rp.fictional_place = label
rp.save()
search = ResourceDocument.search().query("match",
places__fictional_place=label)
assert len(search.execute()) == 1
def test_get_queryset(self):
qs = ResourceDocument().get_queryset()
assert qs.model == Resource
