remove unused code and add more tests

datashaman · Nov 23, 2017 · 5fa0b5e · 5fa0b5e
1 parent cc46b13
commit 5fa0b5e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 36 deletions.
diff --git a/app/__init__.py b/app/__init__.py
@@ -12,7 +12,6 @@
 
 from app import constants
 
-from app.admin import VoucherAdmin
 from app.models import User, Role, db, users
 from app.resources import GatewayResource, \
         NetworkResource, \

diff --git a/app/admin.py b/app/admin.py
diff --git a/tests/test_wifidog.py b/tests/test_wifidog.py
@@ -13,3 +13,15 @@ def test_login_with_invalid_gw_id(self):
     def test_login_with_valid_gw(self):
         response = self.client.get('/wifidog/login/?gw_id=main-gateway1')
         self.assertEqual(200, response.status_code)
+
+    def test_portal_without_gw_id(self):
+        response = self.client.get('/wifidog/portal/')
+        self.assertEqual(404, response.status_code)
+
+    def test_portal_with_invalid_gw_id(self):
+        response = self.client.get('/wifidog/portal/?gw_id=foobar')
+        self.assertEqual(404, response.status_code)
+
+    def test_portal_with_valid_gw(self):
+        response = self.client.get('/wifidog/portal/?gw_id=main-gateway1')
+        self.assertEqual(200, response.status_code)