# -*- coding: UTF-8 -*-
# Copyright 2017 Red Hat, Inc.
# Part of clufter project
# Licensed under GPLv2+ (a copy included | http://gnu.org/licenses/gpl-2.0.txt)
"""Common base for testing"""
__author__ = "Jan Pokorný <jpokorny @at@ Red Hat .dot. com>"


from os.path import join, dirname; from sys import modules as m  # 2/3 compat
b = m.get('builtins', m.get('__builtin__')); e, E, h = 'exec', 'execfile', hash
c = lambda x: compile(x.read(), x.name, 'exec')
with open(join(dirname(dirname(__file__)), '_go')) as F:
    getattr(b, e, getattr(b, E, h)(F.name).__repr__.__name__.__ne__)(c(F))


from os.path import dirname, join, split
from sys import modules
from unittest import TestCase

from .filter_manager import FilterManager
from .format_manager import FormatManager
from .utils import filterdict_keep
from .utils_prog import getenv_namespaced, setenv_namespaced


def rewrite_root(flt, new_root, **kwargs):
    # /foo/bar -> (/foo, bar)
    # /foo/bar/ -> (/foo/bar, )
    new_root_dir, new_xml_root = split(new_root)
    old_root_dir = dirname(modules[flt.__class__.__module__].__file__)
    new_root_dir = join(old_root_dir, new_root_dir)

    def _wrap(flt_cls):
        fnc = flt_cls._fnc
        def _fnc_wrapper(flt_ctxt, *args, **kws):
            flt_ctxt.update(
                filterdict_keep(kwargs, 'walk_transform',
                                root_dir=new_root_dir, xml_root=new_xml_root)
            )
            return fnc(flt_ctxt, *args, **kws)
        return _fnc_wrapper
    flt._fnc = _wrap(flt)
    return flt


class CommonFilterTestCase(TestCase):
    def setUp(self):
        self.fmt_mgr = FormatManager.init_lookup(ext_plugins=False)
        self.flt_mgr = FilterManager.init_lookup(ext_plugins=False,
                                                 fmt_mgr=self.fmt_mgr)

    def tearDown(self):
        self.flt_mgr = None
        self.fmt_mgr = None


class TeardownFilterTestCase(CommonFilterTestCase):
    def setUp(self):
        # start from scratch
        self.fmt_mgr = FormatManager.init_lookup(ext_plugins=False,
                                                 from_scratch=True)
        self.flt_mgr = FilterManager.init_lookup(ext_plugins=False,
                                                 from_scratch=True,
                                                 fmt_mgr=self.fmt_mgr)


class DeterministicFilterTestCase(TeardownFilterTestCase):
    def setUp(self):
        self.nosalt_prev = getenv_namespaced('NOSALT', None)
        setenv_namespaced('NOSALT', '1')
        # start from scratch, incl. proper recycling of the cached "walk"
        # symbols (needed, because there's some dynamics relied upon in tests)
        super(DeterministicFilterTestCase, self).setUp()
        for m in tuple(m for m in modules
                        if m.startswith(self.fmt_mgr.registry.namespace + '.')
                        and '.walk_' in m):
                modules.pop(m)

    def tearDown(self):
        setenv_namespaced('NOSALT', self.nosalt_prev)
        super(DeterministicFilterTestCase, self).tearDown()
