def test_duplicate_user(self):
"""Duplicate usernames are not permitted."""
user1 = User(username="******", email="*****@*****.**")
user1.save()
user2 = User(username="******", email="*****@*****.**")
self.assertRaises(IntegrityError, user2.save)
def test_insert(self):
db_organization = Organization.get(1)
user = User(
username="******",
firstname="un",
lastname="it",
organization=db_organization,
email="*****@*****.**",
password="******"
)
user.save()
db_user = User.get_by_username("unit")
self.assertEqual(db_user, user)
def test_read(self):
for org in self.entities.get("organizations"):
for user in org.get("users"):
db_user = User.get_by_username(user["username"])
self.assertEqual(db_user.username, user["username"])
self.assertEqual(db_user.firstname, user["firstname"])
self.assertEqual(db_user.lastname, user["lastname"])
self.assertTrue(db_user.check_password(user["password"]))
def test_read(self):
for organization in self.entities.get("organizations"):
org = Organization.get_by_name(organization.get("name"))
self.assertEqual(org.name, organization.get("name"))
self.assertEqual(org.domain, organization.get("domain"))
self.assertEqual(org.address1, organization.get("address1"))
self.assertEqual(org.address2, organization.get("address2"))
self.assertEqual(org.zipcode, str(organization.get("zipcode")))
self.assertEqual(org.country, organization.get("country"))
for user in organization.get("users"):
db_user = User.get_by_username(user.get("username"))
self.assertIsNotNone(db_user)
self.assertEqual(db_user.organization.name,
organization.get("name"))
def test_methods(self):
""""Test model methods."""
user = self.entities.get("organizations")[0].get("users")[0]
assert User.get_by_username(user.get("username"))
assert User.username_exists(user.get("username"))
assert User.get_user_list()
def test_relations(self):
organization = self.entities.get("organizations")[0]
user = organization.get("users")[0]
db_user = User.get_by_username(user.get("username"))
self.assertEqual(db_user.organization.name, organization["name"])