release: Mark VIII 1.0
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
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)
|
||||
|
||||
|
||||
def ctx(body=None):
|
||||
return type('C', (), {'body': body, 'args': (), 'query': {}})()
|
||||
|
||||
|
||||
class GetConfigProxyTest(unittest.TestCase):
|
||||
def test_get_config_proxies_daemon(self):
|
||||
server.daemon_sock_call = lambda method, path, body=None, timeout=10: (
|
||||
200, {'loghandshake': True, 'logrecon': False, 'autossidpool': True})
|
||||
status, payload = server.h_pineap_get_config(ctx())
|
||||
self.assertEqual(status, 200)
|
||||
self.assertTrue(payload['loghandshake'])
|
||||
self.assertFalse(payload['logrecon'])
|
||||
|
||||
def test_set_config_forwards_flags(self):
|
||||
calls = []
|
||||
|
||||
def fake(method, path, body=None, timeout=10):
|
||||
if method == 'GET':
|
||||
return 200, {'reconpath': '/root/recon/', 'loghandshake': False,
|
||||
'logpcap': True, 'logrecon': True}
|
||||
calls.append((path, body))
|
||||
return (200, {'success': True})
|
||||
|
||||
server.daemon_sock_call = fake
|
||||
server.h_pineap_set_config(ctx({'loghandshake': True, 'logpcap': False}))
|
||||
path, body = calls[0]
|
||||
self.assertEqual(path, '/api/pineap/set_config')
|
||||
self.assertEqual(body['loghandshake'], True)
|
||||
self.assertEqual(body['logpcap'], False)
|
||||
self.assertEqual(body['logrecon'], True)
|
||||
self.assertEqual(body['reconpath'], '/root/recon/')
|
||||
self.assertTrue('autossidpool' in body)
|
||||
|
||||
def test_hostapd_get(self):
|
||||
server.daemon_sock_call = lambda method, path, body=None, timeout=10: (
|
||||
200, {'pineap_disabled': False, 'pineape_disabled': False})
|
||||
status, payload = server.h_pineap_hostapd_get(ctx())
|
||||
self.assertEqual(status, 200)
|
||||
self.assertFalse(payload['pineap_disabled'])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user