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)