class TestUserModel(unittest.TestCase):
def setUp(self):
from models.users import Users
self.model = Users()
self.model.collection = test_mongo.get_collection("users")
def test_create_user(self):
self.model.collection.remove()
self.model.create_user("*****@*****.**", "password")
eq_(1, self.model.collection.count())
def test_check_user(self):
self.model.collection.remove()
self.model.create_user("*****@*****.**", "password")
assert {} != self.model.check_user("*****@*****.**", "password")
def test_count_users(self):
self.model.collection.remove()
self.model.create_user("*****@*****.**", "one")
self.model.create_user("*****@*****.**", "two")
eq_(2, self.model.count_users())
def test_existence(self):
self.model.collection.remove()
self.model.create_user("*****@*****.**", "one")
self.model.create_user("*****@*****.**", "two")
eq_(1, self.model.user_exists("*****@*****.**"))
