add functional test for session-related API endpoint

tracim/tests/__init__.py

@@ -3,16 +3,21 @@ import unittest
 import transaction
 from depot.manager import DepotManager
 from pyramid import testing
+from sqlalchemy.exc import IntegrityError
 
-
+from tracim.command.initializedb import InitializeDBCommand
 from tracim.lib.core.content import ContentApi
 from tracim.lib.core.workspace import WorkspaceApi
+from tracim.models import get_engine, DeclarativeBase, get_session_factory, \
+    get_tm_session
 from tracim.models.data import Workspace, ContentType
 from tracim.models.data import Content
 from tracim.lib.utils.logger import logger
 from tracim.fixtures import FixturesLoader
 from tracim.fixtures.users_and_groups import Base as BaseFixture
 from tracim.config import CFG
+from tracim import main
+from webtest import TestApp
 
 
 def eq_(a, b, msg=None):
@@ -20,6 +25,53 @@ def eq_(a, b, msg=None):
     assert a == b, msg or "%r != %r" % (a, b)
 
 
+class FunctionalTest(unittest.TestCase):
+    def setUp(self):
+        DepotManager._clear()
+        settings = {
+            'sqlalchemy.url': 'sqlite:///tracim_test.sqlite',
+            'user.auth_token.validity': '604800',
+            'depot_storage_dir': '/tmp/test/depot',
+            'depot_storage_name': 'test',
+            'preview_cache_dir': '/tmp/test/preview_cache',
+
+        }
+        from tracim.extensions import hapic
+        hapic._context = None
+        app = main({}, **settings)
+        self.init_database(settings)
+        self.testapp = TestApp(app)
+
+    def init_database(self, settings):
+        self.engine = get_engine(settings)
+        DeclarativeBase.metadata.create_all(self.engine)
+        session_factory = get_session_factory(self.engine)
+        app_config = CFG(settings)
+        with transaction.manager:
+            dbsession = get_tm_session(session_factory, transaction.manager)
+            try:
+                fixtures_loader = FixturesLoader(dbsession, app_config)
+                fixtures_loader.loads([BaseFixture])
+                transaction.commit()
+                print("Database initialized.")
+            except IntegrityError:
+                print('Warning, there was a problem when adding default data'
+                      ', it may have already been added:')
+                import traceback
+                print(traceback.format_exc())
+                transaction.abort()
+                print('Database initialization failed')
+
+    def tearDown(self):
+        logger.debug(self, 'TearDown Test...')
+        from tracim.models.meta import DeclarativeBase
+
+        testing.tearDown()
+        transaction.abort()
+        DeclarativeBase.metadata.drop_all(self.engine)
+        DepotManager._clear()
+
+
 class BaseTest(unittest.TestCase):
     """
     Pyramid default test.

tracim/tests/functional/test_session.py

@@ -0,0 +1,79 @@
+# coding=utf-8
+from tracim.tests import FunctionalTest
+
+
+class TestLogoutEndpoint(FunctionalTest):
+
+    def test_logout(self):
+        res = self.testapp.get('/api/v2/sessions/logout', status=200)
+        assert res.json_body == {'message': 'ok'}
+
+
+class TestLoginEndpoint(FunctionalTest):
+
+    def test_login_ok(self):
+        params = {
+            'email': 'admin@admin.admin',
+            'password': 'admin@admin.admin',
+        }
+        res = self.testapp.get(
+            '/api/v2/sessions/login',
+            status=200,
+            params=params
+        )
+        assert res.json_body == {'message': 'ok'}
+
+    def test_bad_password(self):
+        params = {
+            'email': 'admin@admin.admin',
+            'password': 'bad_password',
+        }
+        res = self.testapp.get(
+            '/api/v2/sessions/login',
+            status=400,
+            params=params,
+        )
+
+    def test_bad_user(self):
+        params = {
+            'email': 'unknown_user@unknown.unknown',
+            'password': 'bad_password',
+        }
+        res = self.testapp.get(
+            '/api/v2/sessions/login',
+            status=400,
+            params=params,
+        )
+
+    def test_uncomplete(self):
+        res = self.testapp.get('/api/v2/sessions/login', status=400)
+
+
+class TestWhoamiEndpoint(FunctionalTest):
+
+    def test_login_ok(self):
+        self.testapp.authorization = (
+            'Basic',
+            (
+                'admin@admin.admin',
+                'admin@admin.admin'
+            )
+        )
+        res = self.testapp.get('/api/v2/sessions/whoami', status=200)
+        assert res.json_body['display_name'] == 'Global manager'
+        assert res.json_body['email'] == 'admin@admin.admin'
+        assert res.json_body['created']
+        assert res.json_body['is_active']
+        assert res.json_body['profile']
+        assert isinstance(res.json_body['profile']['id'], int)
+        assert res.json_body['profile']['name'] == 'administrators'
+
+    def test_unauthenticated(self):
+        self.testapp.authorization = (
+            'Basic',
+            (
+                'john@doe.doe',
+                'lapin'
+            )
+        )
+        res = self.testapp.get('/api/v2/sessions/whoami', status=401)