def test_process_node(self):
# Set the end date in the future
data = mock_data.UNIQUE_MARKETING_SITE_API_COURSE_BODIES[0]
data['field_course_end_date'] = datetime.datetime.max.strftime('%s')
OrganizationFactory.create(
uuid=data.get('field_course_school_node', {})[0].get('uuid'))
config = DrupalLoaderConfigFactory.create(
course_run_ids=data.get('field_course_id'),
partner_code=self.partner.short_code,
load_unpublished_course_runs=False)
data_loader = DrupalCourseMarketingSiteDataLoader(
self.partner,
self.partner.marketing_site_url_root,
ACCESS_TOKEN,
'JWT',
1, # Make this a constant of 1 for no concurrency
False,
set(config.course_run_ids.split(',')),
config.load_unpublished_course_runs)
# Need to mock this method so that the GET isn't sent out to the test data server
with mock.patch(
'course_discovery.apps.publisher.dataloader.create_courses.'
'transfer_course_image'):
data_loader.process_node(
mock_data.UNIQUE_MARKETING_SITE_API_COURSE_BODIES[0])
course_metadata_course_run = CourseMetadataCourseRun.objects.get(
key=data.get('field_course_id'))
self.assertIsNotNone(course_metadata_course_run)
self.assertIsNotNone(course_metadata_course_run.course)
publisher_course_run = PublisherCourseRun.objects.get(
lms_course_id=course_metadata_course_run.key)
self.assertIsNotNone(publisher_course_run)
self.assertIsNotNone(publisher_course_run.course)
def test_process_node_not_whitelisted(self):
# Set the end date in the future
data = mock_data.UNIQUE_MARKETING_SITE_API_COURSE_BODIES[0]
data['field_course_end_date'] = datetime.datetime.max.strftime('%s')
OrganizationFactory.create(
uuid=data.get('field_course_school_node', {})[0].get('uuid'))
config = DrupalLoaderConfigFactory.create(
course_run_ids='SomeFakeCourseRunId',
partner_code=self.partner.short_code,
load_unpublished_course_runs=False)
data_loader = DrupalCourseMarketingSiteDataLoader(
self.partner,
self.partner.marketing_site_url_root,
ACCESS_TOKEN,
'JWT',
1, # Make this a constant of 1 for no concurrency
False,
set(config.course_run_ids.split(',')),
config.load_unpublished_course_runs)
# Need to mock this method so that the GET isn't sent out to the test data server
with mock.patch(
'course_discovery.apps.publisher.dataloader.create_courses.'
'transfer_course_image'):
for body in mock_data.UNIQUE_MARKETING_SITE_API_COURSE_BODIES:
data_loader.process_node(body)
# Even after looping through all course bodies no new rows should be created
course_metadata_course_run = CourseMetadataCourseRun.objects.filter(
key=data.get('field_course_id'))
self.assertEqual(course_metadata_course_run.count(), 0)
def test_load_unpublished_course_runs_with_flag_enabled_no_course_found(
self):
# Set the end date in the future
data = mock_data.UNIQUE_MARKETING_SITE_API_COURSE_BODIES[0]
data['field_course_end_date'] = datetime.datetime.max.strftime('%s')
# Set the status to unpublished
data['status'] = '0'
OrganizationFactory.create(
uuid=data.get('field_course_school_node', {})[0].get('uuid'))
self.course_run.key = data.get('field_course_id')
self.course_run.save()
load_unpublished_course_runs = True
config = DrupalLoaderConfigFactory.create(
course_run_ids=data.get('field_course_id'),
partner_code=self.partner.short_code,
load_unpublished_course_runs=load_unpublished_course_runs)
data_loader = DrupalCourseMarketingSiteDataLoader(
self.partner,
self.partner.marketing_site_url_root,
ACCESS_TOKEN,
'JWT',
1, # Make this a constant of 1 for no concurrency
False,
set(config.course_run_ids.split(',')),
load_unpublished_course_runs)
logger_target = 'course_discovery.apps.publisher.management.commands.load_drupal_data.logger'
with mock.patch(logger_target) as mock_logger:
# Need to mock this method so that the GET isn't sent out to the test data server
with mock.patch(
'course_discovery.apps.publisher.dataloader.create_courses.'
'transfer_course_image'):
data_loader.process_node(
mock_data.UNIQUE_MARKETING_SITE_API_COURSE_BODIES[0])
expected_calls = [
mock.call('No Publisher Course found for Course Run [%s]',
self.course_run.key)
]
mock_logger.info.assert_has_calls(expected_calls)