64 lines
2.1 KiB
Python
64 lines
2.1 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)
|
|
|
|
_orig_hak5 = server.hak5
|
|
_orig_assoc_clients = server.assoc_clients
|
|
|
|
|
|
class NormalizeTest(unittest.TestCase):
|
|
def test_normalize(self):
|
|
self.assertEqual(server.normalize_mac(' 00:11:22:33:44:55 '), '00:11:22:33:44:55')
|
|
self.assertEqual(server.normalize_mac('aa:bb:cc:dd:ee:ff'), 'AA:BB:CC:DD:EE:FF')
|
|
|
|
def test_invalid_returns_none(self):
|
|
self.assertIsNone(server.normalize_mac('nope'))
|
|
|
|
|
|
class ClientsTest(unittest.TestCase):
|
|
def tearDown(self):
|
|
server.hak5 = _orig_hak5
|
|
server.assoc_clients = _orig_assoc_clients
|
|
|
|
def test_clients_handler(self):
|
|
server.assoc_clients = lambda: [{'mac': 'AA:BB:CC:DD:EE:FF', 'iface': 'wlan0open', 'rssi': -55}]
|
|
class Ctx:
|
|
args = ()
|
|
status, payload = server.h_clients(Ctx())
|
|
self.assertEqual(status, 200)
|
|
self.assertEqual(payload['count'], 1)
|
|
|
|
def test_kick_validates_and_deny_adds(self):
|
|
calls = []
|
|
def fake(*args):
|
|
calls.append(args)
|
|
return 'ok'
|
|
server.hak5 = fake
|
|
server.h_client_kick(type('C', (), {'args': (), 'body': {'mac': '00:11:22:33:44:55'}})())
|
|
self.assertIn(('PINEAPPLE_DEVICE_FILTER_ADD', 'deny', '00:11:22:33:44:55'), calls)
|
|
self.assertTrue(any(c[0] == 'PINEAPPLE_DEAUTH_CLIENT' for c in calls))
|
|
|
|
def test_kick_bad_mac_400(self):
|
|
status, payload = server.h_client_kick(type('C', (), {'args': (), 'body': {'mac': 'x'}})())
|
|
self.assertEqual(status, 400)
|
|
|
|
def test_deauth_client(self):
|
|
calls = []
|
|
def fake(*args):
|
|
calls.append(args)
|
|
return 'ok'
|
|
server.hak5 = fake
|
|
server.h_deauth_client(type('C', (), {'args': (), 'body': {'mac': '00:11:22:33:44:55'}})())
|
|
self.assertEqual(calls[0][0], 'PINEAPPLE_DEAUTH_CLIENT')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|