+
−# -*- coding: utf-8 -*-
+
−# 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 contextlib import contextmanager
+
−
+
−from logilab.common import tempattr
+
−from logilab.common.testlib import Tags
+
−from cubicweb.devtools.testlib import CubicWebTC
+
−
+
−import os
+
−import os.path as osp
+
−import glob
+
−
+
−from cubicweb.utils import HTMLHead
+
−from cubicweb.web.views.staticcontrollers import ConcatFilesHandler
+
−
+
−class staticfilespublishermixin(object):
+
−
+
−    @contextmanager
+
−    def _publish_static_files(self, url, header={}):
+
−        with self.admin_access.web_request(headers=header) as req:
+
−            req._url = url
+
−            self.app_handle_request(req, url)
+
−            yield req
+
−
+
−class StaticControllerCacheTC(staticfilespublishermixin, CubicWebTC):
+
−    tags = CubicWebTC.tags | Tags('static_controller', 'cache', 'http')
+
−
+
−    def test_static_file_are_cached(self):
+
−        with self._publish_static_files('data/cubicweb.css') as req:
+
−            self.assertEqual(200, req.status_out)
+
−            self.assertIn('last-modified', req.headers_out)
+
−        next_headers = {
+
−            'if-modified-since': req.get_response_header('last-modified', raw=True),
+
−        }
+
−        with self._publish_static_files('data/cubicweb.css', next_headers) as req:
+
−            self.assertEqual(304, req.status_out)
+
−
+
−class StaticDirectoryControllerTC(staticfilespublishermixin, CubicWebTC):
+
−
+
−    def test_check_static_dir_access(self):
+
−        """write a file in the static directory and test the access"""
+
−        staticdir = osp.join(self.session.vreg.config.static_directory)
+
−        if not os.path.exists(staticdir):
+
−            os.makedirs(staticdir)
+
−        filename = osp.join(staticdir, 'test')
+
−        with open(filename, 'a') as f:
+
−            with self._publish_static_files('static/test') as req:
+
−                self.assertEqual(200, req.status_out)
+
−
+
−class DataControllerTC(staticfilespublishermixin, CubicWebTC):
+
−    tags = CubicWebTC.tags | Tags('static_controller', 'data', 'http')
+
−
+
−    def _check_datafile_ok(self, fname):
+
−        with self._publish_static_files(fname) as req:
+
−            self.assertEqual(200, req.status_out)
+
−            self.assertIn('last-modified', req.headers_out)
+
−            self.assertIn('expires', req.headers_out)
+
−            self.assertEqual(req.get_response_header('cache-control'),
+
−                             {'max-age': 604800})
+
−        next_headers = {
+
−            'if-modified-since': req.get_response_header('last-modified', raw=True),
+
−        }
+
−        with self._publish_static_files(fname, next_headers) as req:
+
−            self.assertEqual(304, req.status_out)
+
−
+
−    def _check_datafile_redirect(self, fname, expected):
+
−        with self._publish_static_files(fname) as req:
+
−            self.assertEqual(302, req.status_out)
+
−            self.assertEqual(req.get_response_header('location'),
+
−                             req.base_url() + expected)
+
−
+
−    def _check_no_datafile(self, fname):
+
−        with self._publish_static_files(fname) as req:
+
−            self.assertEqual(404, req.status_out)
+
−
+
−    def test_static_data_mode(self):
+
−        hash = self.vreg.config.instance_md5_version()
+
−        self.assertEqual(32, len(hash))
+
−
+
−        with tempattr(self.vreg.config, 'mode', 'test'):
+
−            self._check_datafile_ok('data/cubicweb.css')
+
−            self._check_no_datafile('data/does/not/exist')
+
−            self._check_no_datafile('data/%s/cubicweb.css' % ('0'*len(hash)))
+
−
+
−        with tempattr(self.vreg.config, 'mode', 'notest'):
+
−            self.config._init_base_url()  # reset config.datadir_url
+
−            self._check_datafile_redirect('data/cubicweb.css', 'data/%s/cubicweb.css' % hash)
+
−            self._check_datafile_ok('data/%s/cubicweb.css' % hash)
+
−            self._check_no_datafile('data/%s/does/not/exist' % hash)
+
−            self._check_datafile_redirect('data/%s/does/not/exist' % ('0'*len(hash)),
+
−                                          'data/%s/%s/does/not/exist' % (hash, '0'*len(hash)))
+
−
+
−
+
−class ConcatFilesTC(CubicWebTC):
+
−
+
−    tags = CubicWebTC.tags | Tags('static_controller', 'concat')
+
−
+
−    def tearDown(self):
+
−        super(ConcatFilesTC, self).tearDown()
+
−        self._cleanup_concat_cache()
+
−
+
−    def _cleanup_concat_cache(self):
+
−        uicachedir = osp.join(self.config.apphome, 'uicache')
+
−        for fname in glob.glob(osp.join(uicachedir, 'cache_concat_*')):
+
−            os.unlink(osp.join(uicachedir, fname))
+
−
+
−    @contextmanager
+
−    def _publish_js_files(self, js_files):
+
−        with self.admin_access.web_request() as req:
+
−            head = HTMLHead(req)
+
−            url = head.concat_urls([req.data_url(js_file)
+
−                                    for js_file in js_files])[len(req.base_url()):]
+
−            req._url = url
+
−            res = self.app_handle_request(req, url)
+
−            yield res, req
+
−
+
−    def expected_content(self, js_files):
+
−        content = b''
+
−        for js_file in js_files:
+
−            dirpath, rid = self.config.locate_resource(js_file)
+
−            if dirpath is not None: # ignore resources not found
+
−                with open(osp.join(dirpath, rid), 'rb') as f:
+
−                    content += f.read() + b'\n'
+
−        return content
+
−
+
−    def test_cache(self):
+
−        js_files = ('cubicweb.ajax.js', 'jquery.js')
+
−        with self._publish_js_files(js_files) as (result, req):
+
−            self.assertNotEqual(404, req.status_out)
+
−            # check result content
+
−            self.assertEqual(result, self.expected_content(js_files))
+
−            # make sure we kept a cached version on filesystem
+
−            concat_hander = ConcatFilesHandler(self.config)
+
−            filepath = concat_hander.build_filepath(js_files)
+
−            self.assertTrue(osp.isfile(filepath))
+
−
+
−
+
−    def test_invalid_file_in_debug_mode(self):
+
−        js_files = ('cubicweb.ajax.js', 'dummy.js')
+
−        # in debug mode, an error is raised
+
−        self.config.debugmode = True
+
−        try:
+
−            with self._publish_js_files(js_files) as (result, req):
+
−                #print result
+
−                self.assertEqual(404, req.status_out)
+
−        finally:
+
−            self.config.debugmode = False
+
−
+
−    def test_invalid_file_in_production_mode(self):
+
−        js_files = ('cubicweb.ajax.js', 'dummy.js')
+
−        with self._publish_js_files(js_files) as (result, req):
+
−            self.assertNotEqual(404, req.status_out)
+
−            # check result content
+
−            self.assertEqual(result, self.expected_content(js_files))
+
−
+
−
+
−if __name__ == '__main__':
+
−    from logilab.common.testlib import unittest_main
+
−    unittest_main()
+
−