def test_signals_persons_unpublish(self, mock_bulk, *_):
"""
Unpublishing a person in a language should update its document in the Elasticsearch
persons index or delete it if there is no language published anymore.
It should also reindex the documents for published courses to which it is related,
excluding snapshots.
"""
person = PersonFactory(page_languages=["en", "fr"],
should_publish=True)
published_course, _unpublished_course = CourseFactory.create_batch(
2, fill_team=[person])
self.assertTrue(published_course.extended_object.publish("en"))
published_course.refresh_from_db()
self.run_commit_hooks()
mock_bulk.reset_mock()
# - Unpublish the first language
self.assertTrue(person.extended_object.unpublish("en"))
person.refresh_from_db()
# Elasticsearch should not be called before the db transaction is successful
self.assertFalse(mock_bulk.called)
self.run_commit_hooks()
self.assertEqual(mock_bulk.call_count, 1)
self.assertEqual(len(mock_bulk.call_args[1]["actions"]), 2)
actions = list(mock_bulk.call_args[1]["actions"])
self.assertEqual(actions[0]["_id"], published_course.get_es_id())
self.assertEqual(actions[0]["_op_type"], "index")
self.assertEqual(actions[0]["_index"], "test_courses")
self.assertEqual(actions[1]["_id"], person.get_es_id())
self.assertEqual(actions[1]["_op_type"], "index")
self.assertEqual(actions[1]["_index"], "richie_persons")
mock_bulk.reset_mock()
# - Unpublish the second language
self.assertTrue(person.extended_object.unpublish("fr"))
person.refresh_from_db()
# Elasticsearch should not be called before the db transaction is successful
self.assertFalse(mock_bulk.called)
self.run_commit_hooks()
self.assertEqual(mock_bulk.call_count, 1)
self.assertEqual(len(mock_bulk.call_args[1]["actions"]), 2)
actions = list(mock_bulk.call_args[1]["actions"])
self.assertEqual(actions[0]["_id"], published_course.get_es_id())
self.assertEqual(actions[0]["_op_type"], "index")
self.assertEqual(actions[0]["_index"], "test_courses")
self.assertEqual(actions[1]["_id"], person.get_es_id())
self.assertEqual(actions[1]["_op_type"], "delete")
self.assertEqual(actions[1]["_index"], "richie_persons")
def test_signals_persons_publish(self, mock_bulk, *_):
"""
Publishing a person should update its document in the Elasticsearch persons
index, and the documents for published courses to which it is related, excluding snapshots.
"""
person = PersonFactory()
published_course, _unpublished_course = CourseFactory.create_batch(
2, fill_team=[person])
self.assertTrue(published_course.extended_object.publish("en"))
published_course.refresh_from_db()
self.run_commit_hooks()
mock_bulk.reset_mock()
self.assertTrue(person.extended_object.publish("en"))
person.refresh_from_db()
# Elasticsearch should not be called before the db transaction is successful
self.assertFalse(mock_bulk.called)
self.run_commit_hooks()
self.assertEqual(mock_bulk.call_count, 1)
self.assertEqual(len(mock_bulk.call_args[1]["actions"]), 2)
actions = list(mock_bulk.call_args[1]["actions"])
self.assertEqual(actions[0]["_id"], published_course.get_es_id())
self.assertEqual(actions[0]["_op_type"], "index")
self.assertEqual(actions[0]["_index"], "test_courses")
self.assertEqual(actions[1]["_id"], person.get_es_id())
self.assertEqual(actions[1]["_op_type"], "index")
self.assertEqual(actions[1]["_index"], "richie_persons")
def test_get_es_id_for_draft_person_with_public_extension(self):
"""
A draft person with a public extension. Its ES ID is the ID of the page linked to the
public extension.
"""
person = PersonFactory(should_publish=True)
self.assertEqual(person.get_es_id(),
str(person.public_extension.extended_object_id))
def test_indexers_courses_get_es_documents_from_models(self, _mock_picture):
"""
Happy path: the data is retrieved from the models properly formatted
"""
# Create a course with a page in both english and french
published_categories = [
CategoryFactory(
fill_icon=True,
page_title={"en": "Title cat 1", "fr": "Titre cat 1"},
should_publish=True,
),
CategoryFactory(
fill_icon=True,
page_title={"en": "Title cat 2", "fr": "Titre cat 2"},
should_publish=True,
),
]
draft_category = CategoryFactory(fill_icon=True)
main_organization = OrganizationFactory(
page_title={
"en": "english main organization title",
"fr": "titre organisation principale français",
},
should_publish=True,
)
other_draft_organization = OrganizationFactory(
page_title={
"en": "english other organization title",
"fr": "titre autre organisation français",
}
)
other_published_organization = OrganizationFactory(
page_title={
"en": "english other organization title",
"fr": "titre autre organisation français",
},
should_publish=True,
)
person1 = PersonFactory(
page_title={"en": "Eugène Delacroix", "fr": "Eugène Delacroix"},
should_publish=True,
)
person2 = PersonFactory(
page_title={"en": "Comte de Saint-Germain", "fr": "Earl of Saint-Germain"},
should_publish=True,
)
person_draft = PersonFactory(
page_title={"en": "Jules de Polignac", "fr": "Jules de Polignac"}
)
course = CourseFactory(
duration=[3, WEEK],
effort=[2, HOUR],
fill_categories=published_categories + [draft_category],
fill_cover=True,
fill_icons=published_categories + [draft_category],
fill_organizations=[
main_organization,
other_draft_organization,
other_published_organization,
],
fill_team=[person1, person_draft, person2],
page_title={
"en": "an english course title",
"fr": "un titre cours français",
},
)
CourseRunFactory.create_batch(2, direct_course=course)
course.extended_object.publish("en")
course.extended_object.publish("fr")
course.refresh_from_db()
# Add a description in several languages
placeholder = course.public_extension.extended_object.placeholders.get(
slot="course_description"
)
plugin_params = {"placeholder": placeholder, "plugin_type": "CKEditorPlugin"}
add_plugin(body="english description line 1.", language="en", **plugin_params)
add_plugin(body="english description line 2.", language="en", **plugin_params)
add_plugin(body="a propos français ligne 1.", language="fr", **plugin_params)
add_plugin(body="a propos français ligne 2.", language="fr", **plugin_params)
# Add an introduction in several languages
placeholder = course.public_extension.extended_object.placeholders.get(
slot="course_introduction"
)
plugin_params = {"placeholder": placeholder, "plugin_type": "PlainTextPlugin"}
add_plugin(body="english introduction.", language="en", **plugin_params)
add_plugin(body="introduction française.", language="fr", **plugin_params)
# The results were properly formatted and passed to the consumer
expected_course = {
"_id": course.get_es_id(),
"_index": "some_index",
"_op_type": "some_action",
"absolute_url": {
"en": "/en/an-english-course-title/",
"fr": "/fr/un-titre-cours-francais/",
},
"categories": [
published_categories[0].get_es_id(),
published_categories[1].get_es_id(),
],
"categories_names": {
"en": ["Title cat 1", "Title cat 2"],
"fr": ["Titre cat 1", "Titre cat 2"],
},
"code": course.code,
"complete": {
"en": [
"an english course title",
"english course title",
"course title",
"title",
],
"fr": [
"un titre cours français",
"titre cours français",
"cours français",
"français",
],
},
"course_runs": [
{
"start": course_run.public_course_run.start,
"end": course_run.public_course_run.end,
"enrollment_start": course_run.public_course_run.enrollment_start,
"enrollment_end": course_run.public_course_run.enrollment_end,
"languages": course_run.public_course_run.languages,
}
for course_run in course.course_runs.order_by("-end")
],
"cover_image": {
"en": {"info": "picture info"},
"fr": {"info": "picture info"},
},
"description": {
"en": "english description line 1. english description line 2.",
"fr": "a propos français ligne 1. a propos français ligne 2.",
},
"duration": {"en": "3 weeks", "fr": "3 semaines"},
"effort": {"en": "2 hours", "fr": "2 heures"},
"icon": {
"en": {
"color": published_categories[0].color,
"info": "picture info",
"title": "Title cat 1",
},
"fr": {
"color": published_categories[0].color,
"info": "picture info",
"title": "Titre cat 1",
},
},
"introduction": {
"en": "english introduction.",
"fr": "introduction française.",
},
"is_new": False,
"is_listed": True,
"organization_highlighted": {
"en": "english main organization title",
"fr": "titre organisation principale français",
},
"organization_highlighted_cover_image": {},
"organizations": [
main_organization.get_es_id(),
other_published_organization.get_es_id(),
],
"organizations_names": {
"en": [
"english main organization title",
"english other organization title",
],
"fr": [
"titre organisation principale français",
"titre autre organisation français",
],
},
"persons": [
person1.get_es_id(),
person2.get_es_id(),
],
"persons_names": {
"en": ["Eugène Delacroix", "Comte de Saint-Germain"],
"fr": ["Eugène Delacroix", "Earl of Saint-Germain"],
},
"title": {"fr": "un titre cours français", "en": "an english course title"},
}
indexed_courses = list(
CoursesIndexer.get_es_documents(index="some_index", action="some_action")
)
self.assertEqual(len(indexed_courses), 1)
self.assertEqual(indexed_courses[0], expected_course)
def test_get_es_id_for_draft_person_with_no_public_extension(self):
"""
A draft person with no public extension. It has no ES ID.
"""
person = PersonFactory()
self.assertEqual(person.get_es_id(), None)
