def test_remove(self): """ Test removal from the allowlist """ u1 = UserFactory() notes = 'I had a thought....' # Add user create_or_update_certificate_allowlist_entry(u1, self.course_run_key, notes) entry = get_allowlist_entry(u1, self.course_run_key) assert entry.notes == notes # Remove user remove_allowlist_entry(u1, self.course_run_key) entry = get_allowlist_entry(u1, self.course_run_key) assert entry is None
def test_get_allowlist_entry(self): """ Test to verify that we can retrieve an allowlist entry for a learner. """ allowlist_entry = CertificateAllowlistFactory.create(course_id=self.course_run_key, user=self.user) retrieved_entry = get_allowlist_entry(self.user, self.course_run_key) assert retrieved_entry.id == allowlist_entry.id assert retrieved_entry.course_id == allowlist_entry.course_id assert retrieved_entry.user == allowlist_entry.user
def test_add_and_update(self): """ Test add and update of the allowlist """ u1 = UserFactory() notes = 'blah' # Check before adding user entry = get_allowlist_entry(u1, self.course_run_key) assert entry is None # Add user create_or_update_certificate_allowlist_entry(u1, self.course_run_key, notes) entry = get_allowlist_entry(u1, self.course_run_key) assert entry.notes == notes # Update user new_notes = 'really useful info' create_or_update_certificate_allowlist_entry(u1, self.course_run_key, new_notes) entry = get_allowlist_entry(u1, self.course_run_key) assert entry.notes == new_notes
def test_get_allowlist_entry_dne(self): """ Test to verify behavior when an allowlist entry for a user does not exist """ expected_messages = [ f"Attempting to retrieve an allowlist entry for student {self.user.id} in course {self.course_run_key}.", f"No allowlist entry found for student {self.user.id} in course {self.course_run_key}." ] with LogCapture() as log: retrieved_entry = get_allowlist_entry(self.user, self.course_run_key) assert retrieved_entry is None for index, message in enumerate(expected_messages): assert message in log.records[index].getMessage()