def test_add_user(self):
self.drop_user()
server = Server(**self.credentials)
user = User(username=self.test_user[0], password=self.test_user[1],
host="localhost")
result = server.add_user(user)
with self.get_db() as dbms:
with dbms as cursor:
cursor.execute("SELECT COUNT(`User`) FROM `mysql`.`user` WHERE `User` = ?",
(self.test_user[0],))
self.assertEqual(1, cursor.fetchone()[0])
self.drop_user()
