def test_import_ocw2hugo_course_noncourse(settings):
""" Website should not be created for a non-course """
setup_s3(settings)
name = "biology"
s3_key = f"{TEST_OCW2HUGO_PREFIX}{name}/data/course_legacy.json"
import_ocw2hugo_course(MOCK_BUCKET_NAME, TEST_OCW2HUGO_PREFIX, s3_key)
assert Website.objects.filter(name=name).count() == 0
def test_import_ocw2hugo_course_bad_date(mocker, settings):
""" Website publish date should be null if the JSON date can't be parsed """
setup_s3(settings)
name = "1-050-engineering-mechanics-i-fall-2007"
s3_key = f"{TEST_OCW2HUGO_PREFIX}{name}/data/course_legacy.json"
mocker.patch("ocw_import.api.parse_date", side_effect=ValueError())
import_ocw2hugo_course(MOCK_BUCKET_NAME, TEST_OCW2HUGO_PREFIX, s3_key)
website = Website.objects.get(name=name)
assert website.publish_date is None
def test_import_ocw2hugo_video_gallery(mocker, settings):
""" Website publish date should be null if the JSON date can't be parsed """
setup_s3(settings)
name = "es-s41-speak-italian-with-your-mouth-full-spring-2012"
s3_key = f"{TEST_OCW2HUGO_PREFIX}{name}/data/course_legacy.json"
mocker.patch("ocw_import.api.parse_date", side_effect=ValueError())
import_ocw2hugo_course(MOCK_BUCKET_NAME, TEST_OCW2HUGO_PREFIX, s3_key)
video_lectures = WebsiteContent.objects.get(
text_id="383b8d1c-30df-d781-cc05-c1c648453997")
assert video_lectures.type == "video_gallery"
def test_import_ocw2hugo_content_log_exception(mocker, settings):
""" Log an exception if the website content cannot be saved/updated """
setup_s3(settings)
name = "1-201j-transportation-systems-analysis-demand-and-economics-fall-2008"
s3_key = f"{TEST_OCW2HUGO_PREFIX}{name}/data/course_legacy.json"
mock_log = mocker.patch("ocw_import.api.log.exception")
import_ocw2hugo_course(MOCK_BUCKET_NAME, TEST_OCW2HUGO_PREFIX, s3_key)
assert mock_log.call_count == 1
mock_log.assert_called_once_with("Error saving WebsiteContent for %s",
f"{name}/content/pages/test_no_uid.md")
def test_import_ocw2hugo_course_log_exception(mocker, settings):
""" Log an exception if the website cannot be saved/updated """
setup_s3(settings)
name = "1-050-engineering-mechanics-i-fall-2007"
s3_key = f"{TEST_OCW2HUGO_PREFIX}{name}/data/course_legacy.json"
mocker.patch("ocw_import.api.parse_date", return_value="Invalid date")
mock_log = mocker.patch("ocw_import.api.log.exception")
import_ocw2hugo_course(MOCK_BUCKET_NAME, TEST_OCW2HUGO_PREFIX, s3_key)
assert Website.objects.filter(name=name).first() is None
mock_log.assert_called_once_with("Error saving website %s", s3_key)
def test_import_ocw2hugo_course_gdrive(mocker, settings, gdrive_enabled):
"""Google drive folders should be created if integration is enabled"""
mocker.patch("ocw_import.api.is_gdrive_enabled",
return_value=gdrive_enabled)
mock_sync_gdrive = mocker.patch("ocw_import.api.create_gdrive_folders", )
setup_s3(settings)
name = "1-050-engineering-mechanics-i-fall-2007"
s3_key = f"{TEST_OCW2HUGO_PREFIX}{name}/data/course_legacy.json"
import_ocw2hugo_course(MOCK_BUCKET_NAME, TEST_OCW2HUGO_PREFIX, s3_key)
website = Website.objects.get(name=name)
if gdrive_enabled:
mock_sync_gdrive.assert_called_once_with(website.short_id)
else:
mock_sync_gdrive.assert_not_called()
def test_import_ocw2hugo_menu(settings, mocker):
""" Website publish date should be null if the JSON date can't be parsed """
uuid4_hex = "a" * 32
mocker.patch("uuid.uuid4").return_value.hex = uuid4_hex
setup_s3(settings)
name = "1-050-engineering-mechanics-i-fall-2007"
s3_key = f"{TEST_OCW2HUGO_PREFIX}{name}/data/course_legacy.json"
import_ocw2hugo_course(MOCK_BUCKET_NAME, TEST_OCW2HUGO_PREFIX, s3_key)
website = Website.objects.get(name=name)
navmenu = WebsiteContent.objects.get(website=website, type="navmenu")
assert navmenu.metadata == {
"leftnav": [
{
"url": "/pages/syllabus",
"name": "Syllabus",
"weight": 10,
"identifier": "96c31e82-69f0-6d67-daee-8272526ac56a",
},
{
"url": "/pages/calendar",
"name": "Calendar",
"weight": 20,
"identifier": "ff5e415d-cded-bcfc-d6b2-c4a96377207c",
},
{
"url": "/pages/lecture-notes",
"name": "Lecture Notes",
"weight": 30,
"identifier": "dec40ff4-e8ca-636f-c6db-d88880914a96",
},
{
"url": "/pages/assignments",
"name": "Assignments",
"weight": 40,
"identifier": "8e344ad5-a553-4368-9048-9e95e736657a",
},
{
"url": "/pages/related-resources",
"name": "Related Resources",
"weight": 50,
"identifier": "4f5c3926-e4d5-6974-7f16-131a6f692568",
},
{
"url": "https://openlearning.mit.edu/",
"name": "Open Learning",
"identifier": f"external-{uuid4_hex}",
},
]
}
def test_delete_unpublished_courses(settings, course_starter):
""" delete_unpublished_courses should remove Website objects when they aren't present in the passed in list of paths """
setup_s3(settings)
settings.OCW_IMPORT_STARTER_SLUG = "course"
for course_path in ALL_COURSES_PATHS:
import_ocw2hugo_course(
MOCK_BUCKET_NAME,
TEST_OCW2HUGO_PREFIX,
course_path,
starter_id=course_starter.id,
)
assert Website.objects.all().count() == 4
delete_unpublished_courses(paths=ALL_COURSES_PATHS)
assert Website.objects.all().count() == 4
delete_unpublished_courses(paths=SINGLE_COURSE_PATHS)
assert Website.objects.all().count() == 2
def import_ocw2hugo_course_paths(paths=None, bucket_name=None, prefix=None):
"""
Import all ocw2hugo courses & content
Args:
paths (list of str): list of course data template paths
bucket_name (str): S3 bucket name
prefix (str): S3 prefix before start of course_id path
"""
if not paths:
return
course_site_starter_id = (WebsiteStarter.objects.filter(
slug=settings.OCW_IMPORT_STARTER_SLUG).values_list("id",
flat=True).get())
for path in paths:
log.info("Importing course: '%s'", path)
import_ocw2hugo_course(bucket_name,
prefix,
path,
starter_id=course_site_starter_id)
def test_import_ocw2hugo_course_content(mocker, settings):
""" import_ocw2hugo_course should create a new website plus content"""
setup_s3(settings)
name = "1-050-engineering-mechanics-i-fall-2007"
s3_key = f"{TEST_OCW2HUGO_PREFIX}{name}/data/course_legacy.json"
filenames = [f"file-{i}" for i in range(100)]
get_valid_new_filename_mock = mocker.patch(
"ocw_import.api.get_valid_new_filename", side_effect=filenames)
website_starter = WebsiteStarterFactory.create()
import_ocw2hugo_course(MOCK_BUCKET_NAME,
TEST_OCW2HUGO_PREFIX,
s3_key,
starter_id=website_starter.id)
website = Website.objects.get(name=name)
assert website.starter == website_starter
assert website.source == WEBSITE_SOURCE_OCW_IMPORT
assert website.short_id == "1.050-fall-2007"
with open(f"{TEST_OCW2HUGO_PATH}/{name}/data/course_legacy.json",
"r") as infile:
assert json.dumps(website.metadata,
sort_keys=True) == json.dumps(json.load(infile),
sort_keys=True)
assert (WebsiteContent.objects.filter(website=website,
type=CONTENT_TYPE_PAGE).count() == 7)
assert (WebsiteContent.objects.filter(
website=website, type=CONTENT_TYPE_RESOURCE).count() == 67)
related_page = WebsiteContent.objects.get(
text_id="4f5c3926-e4d5-6974-7f16-131a6f692568")
assert related_page.type == CONTENT_TYPE_PAGE
assert related_page.metadata.get("title") == "Related Resources"
assert related_page.filename == "file-3"
assert related_page.dirpath == "content/pages"
get_valid_new_filename_mock.assert_any_call(
website.pk,
related_page.dirpath,
"related-resources",
related_page.text_id,
)
child_page = WebsiteContent.objects.get(
text_id="6a79c92a-7b81-44f5-b23e-870c73367065")
assert child_page.parent == WebsiteContent.objects.get(
text_id="a38d0e39-8dc8-4a90-b38f-96f349e73c26")
lecture_pdf = WebsiteContent.objects.get(
text_id="7f91d524-57aa-ef80-93c5-8a43f10a099b")
assert lecture_pdf.type == CONTENT_TYPE_RESOURCE
assert lecture_pdf.metadata.get("file_type") == "application/pdf"
assert lecture_pdf.filename == "file-20"
assert lecture_pdf.dirpath == "content/resources"
assert lecture_pdf.file == re.sub(r"^/?coursemedia", "courses",
lecture_pdf.metadata.get("file"))
assert lecture_pdf.metadata.get("learning_resource_types") == [
"Lecture Notes"
]
get_valid_new_filename_mock.assert_any_call(
website.pk,
lecture_pdf.dirpath,
"lec1",
lecture_pdf.text_id,
)
# Any existing content not imported should be deleted
obsolete_id = "testing123"
WebsiteContentFactory.create(website=website, text_id=obsolete_id)
assert WebsiteContent.objects.filter(website=website,
text_id=obsolete_id).exists()
import_ocw2hugo_course(MOCK_BUCKET_NAME,
TEST_OCW2HUGO_PREFIX,
s3_key,
starter_id=website_starter.id)
assert not WebsiteContent.objects.filter(website=website,
text_id=obsolete_id).exists()
def test_import_ocw2hugo_course_metadata(settings, root_website):
""" import_ocw2hugo_course should also populate site metadata"""
setup_s3(settings)
name = "1-050-engineering-mechanics-i-fall-2007"
s3_key = f"{TEST_OCW2HUGO_PREFIX}{name}/data/course_legacy.json"
website_starter = WebsiteStarterFactory.create()
assert (WebsiteContent.objects.filter(
website=root_website, type=CONTENT_TYPE_INSTRUCTOR).count() == 0)
import_ocw2hugo_course(MOCK_BUCKET_NAME,
TEST_OCW2HUGO_PREFIX,
s3_key,
starter_id=website_starter.id)
assert list(
WebsiteContent.objects.filter(type=CONTENT_TYPE_INSTRUCTOR).values(
"title", "dirpath", "is_page_content",
"metadata").order_by("title")) == [
{
"title": "Prof. Franz-Josef Ulm",
"dirpath": "content/instructors",
"is_page_content": True,
"metadata": {
"first_name": "Franz-Josef",
"middle_initial": "",
"last_name": "Ulm",
"salutation": "Prof.",
"headless": True,
},
},
{
"title": "Prof. Markus Buehler",
"dirpath": "content/instructors",
"is_page_content": True,
"metadata": {
"first_name": "Markus",
"middle_initial": "",
"last_name": "Buehler",
"salutation": "Prof.",
"headless": True,
},
},
]
website = Website.objects.get(name=name)
metadata = WebsiteContent.objects.get(website=website,
type=CONTENT_TYPE_METADATA)
assert metadata.metadata == {
"level": ["Undergraduate"],
"topics": [
["Engineering", "Mechanical Engineering", "Solid Mechanics"],
["Engineering", "Aerospace Engineering", "Structural Mechanics"],
["Engineering", "Civil Engineering", "Structural Engineering"],
],
"instructors": {
"content": [
"0b39fff4-81fb-b968-8e2d-a0ce16ece1d4",
"95041ae9-ab5b-75af-f711-13fcd917f464",
],
"website":
"ocw-www",
},
"course_title":
"Engineering Mechanics I",
"course_description":
"This subject provides an introduction to the mechanics of materials and structures. You will be introduced to and become familiar with all relevant physical properties and fundamental laws governing the behavior of materials and structures and you will learn how to solve a variety of problems of interest to civil and environmental engineers. While there will be a chance for you to put your mathematical skills obtained in 18.01, 18.02, and eventually 18.03 to use in this subject, the emphasis is on the physical understanding of why a material or structure behaves the way it does in the engineering design of materials and structures.\n",
"department_numbers": ["1"],
"extra_course_numbers":
"",
"primary_course_number":
"1.050",
"course_image": {
"content": "ba36b428-9898-8e45-81d4-2067ac439546",
"website": "1-050-engineering-mechanics-i-fall-2007",
},
"course_image_thumbnail": {
"content": "4cdfb4e3-32fa-9fdf-a166-c337e35fc009",
"website": "1-050-engineering-mechanics-i-fall-2007",
},
"learning_resource_types": ["Problem Sets", "Lecture Notes"],
"term":
"Fall",
"year":
"2007",
"legacy_uid":
"95f204a1-7715-8120-c7c9-66014bee40dd",
}