import asyncio
import unittest
from microdot import Microdot
from microdot.session import Session, with_session
from microdot.test_client import TestClient

session_ext = Session(secret_key='top-secret!')


class TestSession(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        if hasattr(asyncio, 'set_event_loop'):
            asyncio.set_event_loop(asyncio.new_event_loop())
        cls.loop = asyncio.get_event_loop()

    def _run(self, coro):
        return self.loop.run_until_complete(coro)

    def test_session_async(self):
        app = Microdot()
        session_ext.initialize(app, secret_key='some-other-secret')
        client = TestClient(app)

        @app.get('/')
        async def index(req):
            session = session_ext.get(req)
            session2 = session_ext.get(req)
            session2['foo'] = 'bar'
            self.assertEqual(session['foo'], 'bar')
            return str(session.get('name'))

        @app.get('/with')
        @with_session
        async def session_context_manager(req, session):
            return str(session.get('name'))

        @app.post('/set')
        @with_session
        async def save_session(req, session):
            session['name'] = 'joe'
            session.save()
            return 'OK'

        @app.post('/del')
        @with_session
        async def delete_session(req, session):
            session.delete()
            return 'OK'

        res = self._run(client.get('/'))
        self.assertEqual(res.text, 'None')
        res = self._run(client.get('/with'))
        self.assertEqual(res.text, 'None')

        res = self._run(client.post('/set'))
        self.assertEqual(res.text, 'OK')

        res = self._run(client.get('/'))
        self.assertEqual(res.text, 'joe')
        res = self._run(client.get('/with'))
        self.assertEqual(res.text, 'joe')

        res = self._run(client.post('/del'))
        self.assertEqual(res.text, 'OK')

        res = self._run(client.get('/'))
        self.assertEqual(res.text, 'None')
        res = self._run(client.get('/with'))
        self.assertEqual(res.text, 'None')

    def test_session_no_secret_key(self):
        app = Microdot()
        session_ext = Session(app)
        client = TestClient(app)

        @app.get('/')
        def index(req):
            self.assertRaises(ValueError, session_ext.get, req)
            self.assertRaises(ValueError, session_ext.update, req, {})
            return ''

        res = self._run(client.get('/'))
        self.assertEqual(res.status_code, 200)

    def test_session_default_path(self):
        app = Microdot()
        Session(app, secret_key='some-other-secret')
        client = TestClient(app)

        @app.get('/')
        @with_session
        def index(req, session):
            session['foo'] = 'bar'
            session.save()
            return ''

        @app.get('/child')
        @with_session
        def child(req, session):
            return str(session.get('foo'))

        @app.get('/delete')
        @with_session
        def delete(req, session):
            session.delete()
            return ''

        res = self._run(client.get('/'))
        self.assertEqual(res.status_code, 200)
        res = self._run(client.get('/child'))
        self.assertEqual(res.text, 'bar')
        res = self._run(client.get('/delete'))
        res = self._run(client.get('/child'))
        self.assertEqual(res.text, 'None')

    def test_session_custom_path(self):
        app = Microdot()
        session_ext = Session()
        session_ext.initialize(app, secret_key='some-other-secret',
                               cookie_options={'path': '/child',
                                               'http_only': False})
        client = TestClient(app)

        @app.get('/')
        @with_session
        def index(req, session):
            return str(session.get('foo'))

        @app.get('/child')
        @with_session
        def child(req, session):
            session['foo'] = 'bar'
            session.save()
            return ''

        @app.get('/child/foo')
        @with_session
        def foo(req, session):
            return str(session.get('foo'))

        @app.get('/child/delete')
        @with_session
        def delete(req, session):
            session.delete()
            return ''

        res = self._run(client.get('/child'))
        self.assertEqual(res.status_code, 200)
        res = self._run(client.get('/'))
        self.assertEqual(res.text, 'None')
        res = self._run(client.get('/child/foo'))
        self.assertEqual(res.text, 'bar')
        res = self._run(client.get('/child/delete'))
        res = self._run(client.get('/'))
        self.assertEqual(res.text, 'None')
        res = self._run(client.get('/child/foo'))
        self.assertEqual(res.text, 'None')