def test_get_user(self):
username = "******"
dbconn = Db()
now = datetime.now()
now = now.replace(microsecond=0)
formatted_date = now.strftime('%Y-%m-%d %H:%M:%S')
user = User(username, "userpassword", "testchannel", "testbot",
"testbotOAuth", "channelToken", formatted_date,
formatted_date)
dbconn.create_user(user)
exists = dbconn.get_user(user.username)
self.assertEqual(exists.created, now)
self.cleanup()
def test_verify_user_wrong_password(self):
username = "******"
password = "******"
dbconn = Db()
now = datetime.now()
formatted_date = now.strftime('%Y-%m-%d %H:%M:%S')
user = User(username, password, "testchannel", "testbot",
"testbotOAuth", "channelToken", formatted_date,
formatted_date)
dbconn.create_user(user)
verified_user = dbconn.verify_user(user.username, "wRoNgPaSSw0rd")
dbconn.remove_user(user)
self.assertEqual(None, verified_user)
self.cleanup()
def test_remove_user(self):
username = "******"
dbconn = Db()
now = datetime.now()
formatted_date = now.strftime('%Y-%m-%d %H:%M:%S')
user = User(username, "userpassword", "testchannel", "testbot",
"testbotOAuth", "channelToken", formatted_date,
formatted_date)
exists = dbconn.get_user(user.username)
self.assertEqual(exists, None)
dbconn.create_user(user)
exists_now = dbconn.get_user(user.username)
self.assertEqual(exists_now.username, user.username)
dbconn.remove_user(user)
exists_after_delete = dbconn.get_user(user.username)
self.assertEqual(exists_after_delete, None)
