# copyright 2003-2010 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

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):

        return self.request().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))

if __name__ == '__main__':

    unittest_main()