Updated tests to check for waitlist

bmun · Dec 19, 2023 · d5fe61a · d5fe61a
1 parent 1c6af2a
commit d5fe61a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/huxley/core/tests/admin/test_registration.py b/huxley/core/tests/admin/test_registration.py
@@ -41,6 +41,7 @@ def test_preference_export(self):
             cp.name for cp in registration.committee_preferences.all())]
 
         payment_type_string = ['Credit Card' if registration.payment_type == 1 else 'Check']
+        is_waitlisted_string = ['Yes' if registration.is_waitlisted == 1 else 'No']
 
         fields = [
             registration.registered_at,
@@ -64,6 +65,7 @@ def test_preference_export(self):
         fields.extend(country_preferences)
         fields.extend(committee_preferences)
         fields.extend(registration.registration_comments)
+        fields.extend(is_waitlisted_string)
 
         fields_csv += ','.join(map(str, fields))
         self.assertEquals(fields_csv, response.content[:-3].decode('utf-8'))