test: test_api: add auth test

2 files changed, 16 insertions, 15 deletions

diff --git a/tests/fixtures.py b/tests/fixtures.py
index 53da607..4f0224a 100644
--- a/tests/fixtures.py
+++ b/tests/fixtures.py
@@ -8,10 +8,16 @@
 from squad.core import models as m
 from squad.ci import models as mci
 
+from rest_framework.authtoken.models import Token
+
+user = m.User.objects.create(username='admin_user', is_superuser=True)
+token = Token.objects.create(user=user, key='193cd8bb41ab9217714515954e8724f651ef8601')
+
 group = m.Group.objects.create(slug='my_group')
 group2 = m.Group.objects.create(slug='my_other_group')
 
 project = group.projects.create(slug='my_project')
+project_private = group.projects.create(slug='my_private_project', is_public=False)
 
 build = project.builds.create(version='my_build')
 build2 = project.builds.create(version='my_build2')
diff --git a/tests/test_api.py b/tests/test_api.py
index f2c8763..85e4224 100644
--- a/tests/test_api.py
+++ b/tests/test_api.py
@@ -4,14 +4,6 @@ from . import settings
 from squad_client.core.api import SquadApi, ApiException
 
 
-def is_test_server_running():
-    import socket
-    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-    result = sock.connect_ex(('localhost', settings.DEFAULT_SQUAD_PORT))
-    sock.close()
-    return result == 0
-
-
 class SquadApiTest(TestCase):
 
     def setUp(self):
@@ -25,10 +17,13 @@ class SquadApiTest(TestCase):
         with self.assertRaises(ApiException):
             SquadApi.get('http://some.other.url')
 
-    def test_server_response(self):
-        # will require automatic test server to shutdown, for now, just to it by hand before running this test
-        if not is_test_server_running():
-            with self.assertRaises(ApiException):
-                SquadApi.get('/api/groups')
-        else:
-            self.assertTrue(SquadApi.get('/api/groups') is not None)
+    def test_unauthorized_access(self):
+        response = SquadApi.get('/my_group/my_private_project')
+        self.assertFalse(response.ok)
+
+        SquadApi.configure(url='http://localhost:%s' % settings.DEFAULT_SQUAD_PORT, token='193cd8bb41ab9217714515954e8724f651ef8601')
+        response = SquadApi.get('/my_group/my_private_project')
+        self.assertTrue(response.ok)
+
+        # reset config
+        self.setUp()