release: Mark VIII 1.0

tests/test_pineap_enterprise.py

@@ -0,0 +1,43 @@
+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()