# copyright 2016 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/>.

"""Unit tests for "cubes" importer."""

from contextlib import contextmanager

import os

from os import path

import shutil

import sys

import tempfile

import unittest

from six import PY2

from cubicweb import _CubesImporter

from cubicweb.cwconfig import CubicWebConfiguration

@contextmanager

def temp_cube():

    tempdir = tempfile.mkdtemp()

    try:

        libdir = path.join(tempdir, 'libpython')

        cubedir = path.join(libdir, 'cubicweb_foo')

        os.makedirs(cubedir)

        with open(path.join(cubedir, '__init__.py'), 'w') as f:

            f.write('"""cubicweb_foo application package"""')

        with open(path.join(cubedir, 'bar.py'), 'w') as f:

            f.write('baz = 1')

        sys.path.append(libdir)

        yield cubedir

    finally:

        shutil.rmtree(tempdir)

        sys.path.remove(libdir)

class CubesImporterTC(unittest.TestCase):

    def setUp(self):

        # During discovery, CubicWebConfiguration.cls_adjust_sys_path may be

        # called (probably because of cubicweb.devtools's __init__.py), so

        # uninstall _CubesImporter.

        for x in sys.meta_path:

            if isinstance(x, _CubesImporter):

                sys.meta_path.remove(x)

        # Keep track of initial sys.path and sys.meta_path.

        self.orig_sys_path = sys.path[:]

        self.orig_sys_meta_path = sys.meta_path[:]

    def tearDown(self):

        # Cleanup any imported "cubes".

        for name in list(sys.modules):

            if name.startswith('cubes') or name.startswith('cubicweb_'):

                del sys.modules[name]

        # Restore sys.{meta_,}path

        sys.path[:] = self.orig_sys_path

        sys.meta_path[:] = self.orig_sys_meta_path

    def test_importer_install(self):

        _CubesImporter.install()

        self.assertIsInstance(sys.meta_path[-1], _CubesImporter)

    def test_config_installs_importer(self):

        CubicWebConfiguration.cls_adjust_sys_path()

        self.assertIsInstance(sys.meta_path[-1], _CubesImporter)

    def test_import_cube_as_package_legacy_name(self):

        """Check for import of an actual package-cube using legacy name"""

        with temp_cube() as cubedir:

            import cubicweb_foo  # noqa

            del sys.modules['cubicweb_foo']

            with self.assertRaises(ImportError):

                import cubes.foo

            CubicWebConfiguration.cls_adjust_sys_path()

            import cubes.foo  # noqa

            self.assertEqual(cubes.foo.__path__, [cubedir])

            self.assertEqual(cubes.foo.__doc__,

                             'cubicweb_foo application package')

            # Import a submodule.

            from cubes.foo import bar

            self.assertEqual(bar.baz, 1)

    def test_import_legacy_cube(self):

        """Check that importing a legacy cube works when sys.path got adjusted.

        """

        CubicWebConfiguration.cls_adjust_sys_path()

        import cubes.card  # noqa

    def test_import_cube_as_package_after_legacy_cube(self):

        """Check import of a "cube as package" after a legacy cube."""

        CubicWebConfiguration.cls_adjust_sys_path()

        with temp_cube() as cubedir:

            import cubes.card

            import cubes.foo

        self.assertEqual(cubes.foo.__path__, [cubedir])

    def test_cube_inexistant(self):

        """Check for import of an inexistant cube"""

        CubicWebConfiguration.cls_adjust_sys_path()

        with self.assertRaises(ImportError) as cm:

            import cubes.doesnotexists  # noqa

        msg = "No module named " + ("doesnotexists" if PY2 else "'cubes.doesnotexists'")

        self.assertEqual(str(cm.exception), msg)

if __name__ == '__main__':

    import unittest

    unittest.main()