44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
import os
|
|
import sys
|
|
import unittest
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'payload', 'user', 'remote_access', 'pager-webui'))
|
|
import server
|
|
|
|
|
|
def setUpModule():
|
|
__import__('importlib').reload(server)
|
|
|
|
|
|
class EnterpriseTest(unittest.TestCase):
|
|
def test_basic_rows(self):
|
|
server._db_rows = lambda db, sql: [{'time': 1, 'username': 'a', 'password': 'b'}]
|
|
status, payload = server.h_enterprise_data(type('C', (), {'args': ('basic',)})())
|
|
self.assertEqual(status, 200)
|
|
self.assertEqual(payload['rows'][0]['username'], 'a')
|
|
|
|
def test_challenge_rows(self):
|
|
server._db_rows = lambda db, sql: []
|
|
status, payload = server.h_enterprise_data(type('C', (), {'args': ('challenge',)})())
|
|
self.assertEqual(status, 200)
|
|
self.assertEqual(payload['rows'], [])
|
|
|
|
def test_unknown_table(self):
|
|
status, payload = server.h_enterprise_data(type('C', (), {'args': ('nope',)})())
|
|
self.assertEqual(status, 400)
|
|
|
|
def test_clear(self):
|
|
calls = []
|
|
server._db_write = lambda db, sql: calls.append(sql)
|
|
status, payload = server.h_enterprise_clear(type('C', (), {'body': {'table': 'challenge'}})())
|
|
self.assertEqual(status, 200)
|
|
self.assertTrue(any('hostap_challenge' in s for s in calls))
|
|
|
|
def test_clear_unknown_table(self):
|
|
status, payload = server.h_enterprise_clear(type('C', (), {'body': {'table': 'nope'}})())
|
|
self.assertEqual(status, 400)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|