def test_username_taken(self):
try:
database = Database()
database.add_user('username', 'password')
database.add_user('username', 'asdf')
except sqlite3.IntegrityError:
self.fail()
def test_create_user(self):
try:
database = Database()
database.add_user('username', 'password')
sql = 'SELECT COUNT(*) FROM Users WHERE Name="username"'
row = database.execute(sql).fetchone()
self.assertEqual(row[0], 1)
except sqlite3.IntegrityError as e:
self.assertEqual(str(e), 'UNIQUE constraint failed: Users.Name')
def test_password_verify(self):
database = Database()
database.add_user('username', 'password')
self.assertTrue(database.password_verify('username', 'password'))
