def test_db_connect_exception(self):
db = MySQLdb.connect(server, user, password, database)
cursor = db.cursor()
call(["sudo", "service", "mysql", "stop"], stdout=stdout_log)
with self.assertRaises(MySQLdb.Error):
qwifi.main()
with self.assertRaises(MySQLdb.OperationalError):
qwifi.update_radcheck(db, cursor)
with self.assertRaises(MySQLdb.Error):
qwifi.cull(db, cursor)
call(["sudo", "service", "mysql", "start"], stdout=stdout_log)
def test_db_connect_exception(self):
db = MySQLdb.connect(server, user, password, database)
cursor = db.cursor()
call(["sudo", "service", "mysql", "stop"], stdout=stdout_log)
with self.assertRaises(MySQLdb.Error):
qwifi.main()
with self.assertRaises(MySQLdb.OperationalError):
qwifi.update_radcheck(db, cursor)
with self.assertRaises(MySQLdb.Error):
qwifi.cull(db, cursor)
call(["sudo", "service", "mysql", "start"], stdout=stdout_log)
def test_update_radcheck_device(self):
qwifi.session_mode = qwifi.session_modes.DEVICE
os.system("mysql -u " + user +" -p" + password + " -h " + server + " " + database + " < " + "test.sql" )
db = MySQLdb.connect(server, user, password, database)
cursor = db.cursor()
#let updateRadcheck do it's job
qwifi.update_radcheck(db, cursor)
#three new entries should be added to radcheck 6 (original) + 3 (reject) == 6 entries
cursor.execute("SELECT * from radcheck;")
self.assertEqual(len(cursor.fetchall()), 9)
db.close()
def test_update_radcheck_device(self):
qwifi.session_mode = qwifi.session_modes.DEVICE
os.system("mysql -u " + user + " -p" + password + " -h " + server +
" " + database + " < " + "test.sql")
db = MySQLdb.connect(server, user, password, database)
cursor = db.cursor()
#let updateRadcheck do it's job
qwifi.update_radcheck(db, cursor)
#three new entries should be added to radcheck 6 (original) + 3 (reject) == 6 entries
cursor.execute("SELECT * from radcheck;")
self.assertEqual(len(cursor.fetchall()), 9)
db.close()
def test_update_radcheck_ap(self):
qwifi.session_mode = qwifi.session_modes.AP
qwifi.config.set("session", "timeout", '10')
os.system("mysql -u " + user +" -p" + password + " -h " + server + " " + database + " < " + "test.sql" )
db = MySQLdb.connect(server, user, password, database)
cursor = db.cursor()
#let updateRadcheck do it's job
qwifi.update_radcheck(db, cursor)
#five new entries should be added to radcheck 6 (original) + 3 (reject) + 2 (regen) == 11 entries
cursor.execute("SELECT * from radcheck;")
self.assertEqual(len(cursor.fetchall()), 11)
cursor.execute("DELETE from radcheck;")
qwifi.update_radcheck(db, cursor)
cursor.execute("SELECT * FROM radcheck;")
self.assertEqual(len(cursor.fetchall()), 2)
db.close()
def test_update_radcheck_ap(self):
qwifi.session_mode = qwifi.session_modes.AP
qwifi.config.set("session", "timeout", '10')
os.system("mysql -u " + user + " -p" + password + " -h " + server +
" " + database + " < " + "test.sql")
db = MySQLdb.connect(server, user, password, database)
cursor = db.cursor()
#let updateRadcheck do it's job
qwifi.update_radcheck(db, cursor)
#five new entries should be added to radcheck 6 (original) + 3 (reject) + 2 (regen) == 11 entries
cursor.execute("SELECT * from radcheck;")
self.assertEqual(len(cursor.fetchall()), 11)
cursor.execute("DELETE from radcheck;")
qwifi.update_radcheck(db, cursor)
cursor.execute("SELECT * FROM radcheck;")
self.assertEqual(len(cursor.fetchall()), 2)
db.close()
