def test_upsert_content_file(mocker):
"""
Tests that upsert_content_file calls the correct celery task with parameters
"""
patched_task = mocker.patch("search.tasks.upsert_content_file")
content_file = ContentFileFactory.create()
upsert_content_file(content_file.id)
patched_task.delay.assert_called_once_with(content_file.id)
def test_index_run_content_files(mocker):
"""
Tests that index_run_content_files calls the correct celery task w/parameter
"""
patched_task = mocker.patch("search.tasks.index_run_content_files")
content_file = ContentFileFactory.create()
index_run_content_files(content_file.id)
patched_task.delay.assert_called_once_with(content_file.id)
def test_serialize_content_file_for_bulk_deletion():
"""
Test that serialize_content_file_for_bulk_deletion yields a valid ESContentFileSerializer
"""
content_file = ContentFileFactory.create()
assert serialize_content_file_for_bulk_deletion(content_file) == {
"_id": gen_content_file_id(content_file.key),
"_op_type": "delete",
}
def test_serialize_content_file_for_bulk():
"""
Test that serialize_content_file_for_bulk yields a valid ESContentFileSerializer
"""
content_file = ContentFileFactory.create()
assert serialize_content_file_for_bulk(content_file) == {
"_id": gen_content_file_id(content_file.key),
**ESContentFileSerializer(content_file).data,
}
def test_delete_content_file(mocker):
"""Tests that deleting a content_file triggers the correct ES delete task"""
patched_delete_task = mocker.patch("search.task_helpers.delete_document")
content_file = ContentFileFactory.create()
delete_content_file(content_file)
assert patched_delete_task.delay.called is True
assert patched_delete_task.delay.call_args[0] == (
gen_content_file_id(content_file.key),
COURSE_TYPE,
)
def test_delete_course(mocker):
"""
Tests that delete_course calls the delete tasks for the course and its content files
"""
patched_delete_task = mocker.patch("search.task_helpers.delete_document")
course = CourseFactory.create()
course_es_id = gen_course_id(course.platform, course.course_id)
content_files = [ContentFileFactory.create(run=run) for run in course.runs.all()]
delete_course(course)
patched_delete_task.delay.assert_any_call(course_es_id, COURSE_TYPE)
for content_file in content_files:
patched_delete_task.delay.assert_any_call(
gen_content_file_id(content_file.key), COURSE_TYPE, routing=course_es_id
)
def test_es_content_file_serializer():
""" Verify that the ESContentFileSerializer has the correct data"""
content_kwargs = {
"content": "Some text",
"content_author": "MIT",
"content_language": "en",
"content_title": "test title",
}
content_file = ContentFileFactory.create(**content_kwargs)
serialized = ESContentFileSerializer(content_file).data
assert_json_equal(
serialized,
{
"object_type": RESOURCE_FILE_TYPE,
"run_id": content_file.run.run_id,
"run_title": content_file.run.title,
"semester": content_file.run.semester,
"year": int(content_file.run.year),
"topics": list(content_file.run.topics.values_list("name", flat=True)),
"key": content_file.key,
"uid": content_file.uid,
"resource_relations": {
"name": "resourcefile",
"parent": gen_course_id(
content_file.run.content_object.platform,
content_file.run.content_object.course_id,
),
},
"title": content_file.title,
"short_description": content_file.description,
"file_type": content_file.file_type,
"content_type": content_file.content_type,
"url": content_file.url,
"section": content_file.section,
"content": content_kwargs["content"],
"content_title": content_kwargs["content_title"],
"content_author": content_kwargs["content_author"],
"content_language": content_kwargs["content_language"],
},
)
def test_delete_run_content_files(mocker):
"""Tests that delete_run_content_files triggers the correct ES delete task"""
patched_task = mocker.patch("search.tasks.delete_run_content_files")
content_file = ContentFileFactory.create()
delete_run_content_files(content_file.id)
patched_task.delay.assert_called_once_with(content_file.id)