def test_getUsers(self): db.wipeDatabase() db.addUser("test1", "password", "*****@*****.**") db.addUser("test2", "password", "*****@*****.**") db.addUser("test3", "password", "*****@*****.**") numUsers = db.getUsers() self.assertEqual(len(numUsers), 3)
def tearDown(self): db.wipeDatabase()
def test_returnUser(self): db.wipeDatabase() db.addUser("test1", "password", "*****@*****.**") db.addUser("test2", "password", "*****@*****.**") self.assertEqual(db.returnUser("test1")['username'], "test1")
def test_removeUser(self): db.wipeDatabase() db.addUser("test", "password", "*****@*****.**") db.removeUser("test") self.assertEqual(db.returnUser("test"), None)
def _beforeEach(): wipeDatabase()