def test_small_cohort(self):
cohort_upload = CohortUpload()
cohort_upload.name.data = 'small_cohort'
cohort_upload.project.data = mediawiki_project
cohort_upload.records = [
# two existing users
{'raw_id_or_name': 'Editor test-specific-0', 'project': mediawiki_project},
{'raw_id_or_name': 'Editor test-specific-1', 'project': mediawiki_project},
# one invalid username
{'raw_id_or_name': 'Nonexisting', 'project': mediawiki_project},
# one user with invalid project
{'raw_id_or_name': 'Nonexisting2', 'project': 'Nonexisting'},
]
v = ValidateCohort.from_upload(cohort_upload, self.owner_user_id)
v.task.delay(v).get()
self.session.commit()
assert_equal(self.session.query(WikiUserStore).filter(
WikiUserStore.raw_id_or_name == 'Editor test-specific-0').one().valid,
True
)
assert_equal(self.session.query(WikiUserStore).filter(
WikiUserStore.raw_id_or_name == 'Editor test-specific-1').one().valid,
True
)
assert_equal(self.session.query(WikiUserStore).filter(
WikiUserStore.raw_id_or_name == 'Nonexisting').one().valid, False)
assert_equal(self.session.query(WikiUserStore).filter(
WikiUserStore.raw_id_or_name == 'Nonexisting2').one().valid, False)
def test_from_upload_exception(self):
cohort_upload = CohortUpload()
cohort_upload.name.data = 'small_cohort'
cohort_upload.project.data = 'wiki'
cohort_upload.records = [{'fake': 'dict'}]
v = ValidateCohort.from_upload(cohort_upload, self.owner_user_id)
assert_equal(v, None)
def test_from_upload_exception(self):
cohort_upload = CohortUpload()
cohort_upload.name.data = 'small_cohort'
cohort_upload.project.data = 'wiki'
cohort_upload.records = [{'fake': 'dict'}]
v = ValidateCohort.from_upload(cohort_upload, self.owner_user_id)
assert_equal(v, None)
def test_small_cohort(self):
cohort_upload = CohortUpload()
cohort_upload.name.data = 'small_cohort'
cohort_upload.project.data = mediawiki_project
cohort_upload.records = [
# two existing users
{
'raw_id_or_name': 'Editor test-specific-0',
'project': mediawiki_project
},
{
'raw_id_or_name': 'Editor test-specific-1',
'project': mediawiki_project
},
# one invalid username
{
'raw_id_or_name': 'Nonexisting',
'project': mediawiki_project
},
# one user with invalid project
{
'raw_id_or_name': 'Nonexisting2',
'project': 'Nonexisting'
},
]
v = ValidateCohort.from_upload(cohort_upload, self.owner_user_id)
v.task.delay(v).get()
self.session.commit()
assert_equal(
self.session.query(WikiUserStore).filter(
WikiUserStore.raw_id_or_name ==
'Editor test-specific-0').one().valid, True)
assert_equal(
self.session.query(WikiUserStore).filter(
WikiUserStore.raw_id_or_name ==
'Editor test-specific-1').one().valid, True)
assert_equal(
self.session.query(WikiUserStore).filter(
WikiUserStore.raw_id_or_name == 'Nonexisting').one().valid,
False)
assert_equal(
self.session.query(WikiUserStore).filter(
WikiUserStore.raw_id_or_name == 'Nonexisting2').one().valid,
False)