# copyright 2003-2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr

#

# This file is part of CubicWeb.

#

# CubicWeb is free software: you can redistribute it and/or modify it under the

# terms of the GNU Lesser General Public License as published by the Free

# Software Foundation, either version 2.1 of the License, or (at your option)

# any later version.

#

# CubicWeb is distributed in the hope that it will be useful, but WITHOUT

# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more

# details.

#

# You should have received a copy of the GNU Lesser General Public License along

# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.

from json import loads

from os.path import join

import tempfile

try:

    import requests

    assert [int(n) for n in requests.__version__.split('.', 2)][:2] >= [1, 2]

except (ImportError, AssertionError):

    requests = None

from logilab.common.testlib import TestCase, unittest_main

from cubicweb.devtools.httptest import CubicWebServerTC

from cubicweb.devtools.fake import FakeRequest

class AjaxReplaceUrlTC(TestCase):

    def test_ajax_replace_url_1(self):

        self._test_arurl("fname=view&rql=Person%20P&vid=list",

                         rql='Person P', vid='list')

    def test_ajax_replace_url_2(self):

        self._test_arurl("age=12&fname=view&name=bar&rql=Person%20P&vid=oneline",

                         rql='Person P', vid='oneline', name='bar', age=12)

    def _test_arurl(self, qs, **kwargs):

        req = FakeRequest()

        arurl = req.ajax_replace_url

        # NOTE: for the simplest use cases, we could use doctest

        url = arurl('foo', **kwargs)

        self.assertTrue(url.startswith('javascript:'))

        self.assertTrue(url.endswith('()'))

        cbname = url.split()[1][:-2]

        self.assertMultiLineEqual(

            'function %s() { $("#foo").loadxhtml("http://testing.fr/cubicweb/ajax?%s",'

            '{pageid: "%s"},"get","replace"); }' %

            (cbname, qs, req.pageid),

            req.html_headers.post_inlined_scripts[0])

class FileUploadTC(CubicWebServerTC):

    def setUp(self):

        "Skip whole test class if a suitable requests module is not available"

        if requests is None:

            self.skipTest('Python ``requests`` module is not available')

        super(FileUploadTC, self).setUp()

    @property

    def _post_url(self):

        with self.admin_access.web_request() as req:

            return req.build_url('ajax', fname='fileupload')

    def _fobject(self, fname):

        return open(join(self.datadir, fname), 'rb')

    def _fcontent(self, fname):

        return self._fobject(fname).read()

    def test_single_file_upload(self):

        files = {'file': ('schema.py', self._fobject('schema.py'))}

        webreq = requests.post(self._post_url, files=files)

        # check backward compat : a single uploaded file leads to a single

        # 2-uple in the request form

        expect = {'fname': u'fileupload',

                  'file': ['schema.py', self._fcontent('schema.py')]}

        self.assertEqual(webreq.status_code, 200)

        self.assertDictEqual(expect, loads(webreq.content))

    def test_multiple_file_upload(self):

        files = [('files', ('schema.py', self._fobject('schema.py'))),

                 ('files', ('views.py',  self._fobject('views.py')))]

        webreq = requests.post(self._post_url, files=files,)

        expect = {'fname': u'fileupload',

                  'files': [['schema.py', self._fcontent('schema.py')],

                            ['views.py', self._fcontent('views.py')]],}

        self.assertEqual(webreq.status_code, 200)

        self.assertDictEqual(expect, loads(webreq.content))

class LanguageTC(CubicWebServerTC):

    def test_language_neg(self):

        headers = {'Accept-Language': 'fr'}

        webreq = self.web_request(headers=headers)

        self.assertIn('lang="fr"', webreq.read())

        vary = [h.lower().strip() for h in webreq.getheader('Vary').split(',')]

        self.assertIn('accept-language', vary)

        headers = {'Accept-Language': 'en'}

        webreq = self.web_request(headers=headers)

        self.assertIn('lang="en"', webreq.read())

        vary = [h.lower().strip() for h in webreq.getheader('Vary').split(',')]

        self.assertIn('accept-language', vary)

    def test_response_codes(self):

        with self.admin_access.client_cnx() as cnx:

            admin_eid = cnx.user.eid

        # guest can't see admin

        webreq = self.web_request('/%d' % admin_eid)

        self.assertEqual(webreq.status, 403)

        # but admin can

        self.web_login()

        webreq = self.web_request('/%d' % admin_eid)

        self.assertEqual(webreq.status, 200)

    def test_session_cookie_httponly(self):

        webreq = self.web_request()

        self.assertIn('HttpOnly', webreq.getheader('set-cookie'))

class LogQueriesTC(CubicWebServerTC):

    @classmethod

    def init_config(cls, config):

        super(LogQueriesTC, cls).init_config(config)

        cls.logfile = tempfile.NamedTemporaryFile()

        config.global_set_option('query-log-file', cls.logfile.name)

    def test_log_queries(self):

        self.web_request()

        self.assertTrue(self.logfile.read())

    @classmethod

    def tearDownClass(cls):

        cls.logfile.close()

if __name__ == '__main__':

    unittest_main()