def test_write_files(self, mock_open):
# Test empty
csv = Collector()
self.assertEquals(csv.has_data(), False)
self.assertEquals(csv.write_files(), None)
# Test with data
csv = Collector()
csv.enrollments.append(1)
self.assertEquals(csv.has_data(), True)
with self.settings(SIS_IMPORT_CSV_DEBUG=False):
path = csv.write_files()
mock_open.assert_called_with(path + '/enrollments.csv', mode='w')
self.assertEquals(csv.has_data(), False)
def test_admins(self):
formatter = AdminCSV('user_id', 'account_id', 'admin', 'active')
csv = Collector()
self.assertEquals(len(csv.admins), 0)
self.assertEquals(csv.add(formatter), True)
self.assertEquals(len(csv.admins), 1)
self.assertEquals(csv.has_data(), True)
def test_users(self):
user = PWS().get_person_by_netid('javerage')
csv = Collector()
self.assertEquals(len(csv.users), 0)
self.assertEquals(csv.add(UserCSV(user, 'active')), True)
self.assertEquals(len(csv.users), 1)
self.assertEquals(csv.add(UserCSV(user, 'active')), False)
self.assertEquals(csv.has_data(), True)
def test_terms(self):
section = get_section_by_label('2013,summer,TRAIN,101/A')
formatter = TermCSV(section)
csv = Collector()
self.assertEquals(len(csv.terms), 0)
self.assertEquals(csv.add(formatter), True)
self.assertEquals(len(csv.terms), 1)
self.assertEquals(csv.add(formatter), False)
self.assertEquals(csv.has_data(), True)
def test_accounts(self):
context = Curriculum(full_name='abc')
formatter = AccountCSV('account_id', 'parent_id', context)
csv = Collector()
self.assertEquals(len(csv.accounts), 0)
self.assertEquals(csv.add(formatter), True)
self.assertEquals(len(csv.accounts), 1)
self.assertEquals(csv.add(formatter), False)
self.assertEquals(csv.has_data(), True)
def test_enrollments(self):
user = PWS().get_person_by_netid('javerage')
csv = Collector()
self.assertEquals(len(csv.enrollments), 0)
self.assertEquals(
csv.add(
EnrollmentCSV(section_id='abc',
person=user,
role='Student',
status='active')), True)
self.assertEquals(len(csv.enrollments), 1)
section = get_section_by_label('2013,winter,DROP_T,100/B')
for registration in get_registrations_by_section(section):
self.assertEquals(
csv.add(EnrollmentCSV(registration=registration)), True)
self.assertEquals(len(csv.enrollments), 3)
section = get_section_by_label('2013,spring,TRAIN,101/A')
for user in section.get_instructors():
self.assertEquals(
csv.add(
EnrollmentCSV(section=section,
instructor=user,
status='active')), True)
# Duplicate
self.assertEquals(
csv.add(
EnrollmentCSV(section=section,
instructor=user,
status='active')), False)
self.assertEquals(len(csv.enrollments), 5)
self.assertEquals(csv.has_data(), True)
# Ad-hoc enrollment
self.assertEquals(
csv.add(
EnrollmentCSV(course_id='course_123',
section_id='section_123',
person=user,
role='Observer',
status='active')), True)
self.assertEquals(len(csv.enrollments), 6)
# Duplicate
self.assertEquals(
csv.add(
EnrollmentCSV(course_id='course_123',
section_id='section_123',
person=user,
role='Observer',
status='active')), False)
def test_courses(self):
section = get_section_by_label('2013,spring,TRAIN,101/A')
section.course_campus = 'PCE'
formatter = CourseCSV(section=section)
csv = Collector()
self.assertEquals(len(csv.courses), 0)
self.assertEquals(csv.add(formatter), True)
self.assertEquals(len(csv.courses), 1)
self.assertEquals(csv.add(formatter), False)
self.assertEquals(csv.has_data(), True)
def handle(self, *args, **options):
file_path = options.get('file_path')
workshop_name = options.get('workshop_name')
term_sis_id = options.get('term_sis_id')
account_sis_id = options.get('account_sis_id')
with open(file_path, 'r') as infile:
file_data = infile.read()
netids = file_data.splitlines()
csvdata = Collector()
for netid in netids:
try:
person = get_person_by_netid(netid.strip())
except UserPolicyException as err:
print("Skipped user '{}': {}".format(netid, err))
continue
if not csvdata.add(UserCSV(person)):
continue
course_sis_id = '-'.join([
term_sis_id,
re.sub(r'[^\w]', '-', workshop_name.lower()), person.uwnetid
])
short_name = '{} {}'.format(date.today().year, workshop_name)
long_name = '{} Sandbox'.format(short_name)
csvdata.add(
CourseCSV(course_id=course_sis_id,
short_name=short_name,
long_name=long_name,
account_id=account_sis_id,
term_id=term_sis_id,
status='active'))
csvdata.add(
EnrollmentCSV(course_id=course_sis_id,
person=person,
role=get_instructor_sis_import_role(),
status=ENROLLMENT_ACTIVE))
csv_path = csvdata.write_files()
if csv_path:
imp = Import(priority=Course.PRIORITY_DEFAULT,
csv_type='course',
csv_path=csv_path)
imp.save()
imp.import_csv()
def test_sections(self):
section = get_section_by_label('2013,spring,TRAIN,101/A')
formatter = SectionCSV(section=section)
csv = Collector()
self.assertEquals(len(csv.sections), 0)
self.assertEquals(csv.add(formatter), True)
self.assertEquals(len(csv.sections), 1)
self.assertEquals(csv.add(formatter), False)
self.assertEquals(
csv.add(
SectionCSV(section_id='abc',
course_id='abc',
name='abc',
status='active')), True)
self.assertEquals(len(csv.sections), 2)
self.assertEquals(csv.has_data(), True)
def __init__(self, items=[]):
self.data = Collector()
self.queue_id = None
self.invalid_users = {}
self.items = items
self.logger = getLogger(__name__)
def test_invalid_format(self):
csv = Collector()
self.assertRaises(TypeError, csv.add, InvalidFormat)
self.assertEquals(csv.has_data(), False)
def test_xlists(self):
csv = Collector()
self.assertEquals(len(csv.xlists), 0)
self.assertEquals(csv.add(XlistCSV('abc', 'def')), True)
self.assertEquals(len(csv.xlists), 1)
self.assertEquals(csv.has_data(), True)
