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()