fixed bug on eligibility

helios/models.py

@@ -266,8 +266,11 @@ class Election(HeliosModel):
       return_val = "<ul>"
       
       for constraint in self.eligibility:
+        if constraint.has_key('constraint'):
           for one_constraint in constraint['constraint']:
             return_val += "<li>%s</li>" % AUTH_SYSTEMS[constraint['auth_system']].pretty_eligibility(one_constraint)
+        else:
+          return_val += "<li> any %s user</li>" % constraint['auth_system']
 
       return_val += "</ul>"
 

helios/tests.py

@@ -148,6 +148,9 @@ class ElectionModelTests(TestCase):
         # without openreg, and now true
         self.assertTrue(self.election.user_eligible_p(self.user))
 
+        # try getting pretty eligibility, make sure it doesn't throw an exception
+        assert self.user.user_type in self.election.pretty_eligibility
+
     def test_facebook_eligibility(self):
         self.election.eligibility = [{'auth_system': 'facebook', 'constraint':[{'group': {'id': '123', 'name':'Fake Group'}}]}]