def testUserToDict(self):
"""Whether the necessary fields match in a dictionary representation."""
user = models.User()
user.is_key_revoked = False
user_dict = user.to_dict()
self.assertEqual(user_dict['id'], user.id)
self.assertEqual(user_dict['email'], user.email)
self.assertEqual(user_dict['name'], user.name)
self.assertEqual(user_dict['private_key'], user.private_key)
self.assertEqual(user_dict['public_key'], user.public_key)
self.assertEqual(user_dict['access'], models.NOT_REVOKED_TEXT)
self.assertEqual(user_dict['accessChange'], models.DISABLE_TEXT)
user.is_key_revoked = True
user_dict = user.to_dict()
self.assertEqual(user_dict['access'], models.REVOKED_TEXT)
self.assertEqual(user_dict['accessChange'], models.ENABLE_TEXT)
def testUserToDict(self):
"""Whether the necessary fields match in a dictionary representation."""
user = models.User()
user.is_key_revoked = False
user_dict = user.to_dict()
self.assertEqual(user_dict['id'], user.id)
self.assertEqual(user_dict['email'], user.email)
self.assertEqual(user_dict['name'], user.name)
self.assertEqual(user_dict['private_key'], user.private_key)
self.assertEqual(user_dict['public_key'], user.public_key)
self.assertEqual(user_dict['access'], models.NOT_REVOKED_TEXT)
self.assertEqual(user_dict['accessChange'], models.DISABLE_TEXT)
user.is_key_revoked = True
user_dict = user.to_dict()
self.assertEqual(user_dict['access'], models.REVOKED_TEXT)
self.assertEqual(user_dict['accessChange'], models.ENABLE_TEXT)
def testListUsersHandler(self):
"""Test the list user handler gets users from the database."""
users = []
for fake_email_and_name in base_test.FAKE_EMAILS_AND_NAMES:
user = models.User(email=fake_email_and_name['email'],
name=fake_email_and_name['name'],
private_key=fake_email_and_name['pri'],
public_key=fake_email_and_name['pub'])
user.save()
users.append(user)
resp = self.client.get(flask.url_for('user_list'))
user_list_output = json.loads(resp.data[len(ufo.XSSI_PREFIX):])['items']
self.assertEquals(len(user_list_output),
len(base_test.FAKE_EMAILS_AND_NAMES))
for user in users:
self.assertIn(user.to_dict(), user_list_output)
