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

"""this module contains base classes and utilities for cubicweb tests"""

__docformat__ = "restructuredtext en"

import sys

import re

import urlparse

from os.path import dirname, join, abspath

from urllib import unquote

from math import log

from contextlib import contextmanager

from warnings import warn

from types import NoneType

from itertools import chain

import yams.schema

from logilab.common.testlib import TestCase, InnerTest, Tags

from logilab.common.pytest import nocoverage, pause_tracing, resume_tracing

from logilab.common.debugger import Debugger

from logilab.common.umessage import message_from_string

from logilab.common.decorators import cached, classproperty, clear_cache, iclassmethod

from logilab.common.deprecation import deprecated, class_deprecated

from logilab.common.shellutils import getlogin

from cubicweb import ValidationError, NoSelectableObject, AuthenticationError

from cubicweb import cwconfig, dbapi, devtools, web, server, repoapi

from cubicweb.utils import json

from cubicweb.sobjects import notification

from cubicweb.web import Redirect, application

from cubicweb.server.hook import SendMailOp

from cubicweb.devtools import SYSTEM_ENTITIES, SYSTEM_RELATIONS, VIEW_VALIDATORS

from cubicweb.devtools import fake, htmlparser, DEFAULT_EMPTY_DB_ID

from cubicweb.utils import json

# low-level utilities ##########################################################

class CubicWebDebugger(Debugger):

    """special debugger class providing a 'view' function which saves some

    html into a temporary file and open a web browser to examinate it.

    """

    def do_view(self, arg):

        import webbrowser

        data = self._getval(arg)

        with file('/tmp/toto.html', 'w') as toto:

            toto.write(data)

        webbrowser.open('file:///tmp/toto.html')

def line_context_filter(line_no, center, before=3, after=None):

    """return true if line are in context

    if after is None: after = before

    """

    if after is None:

        after = before

    return center - before <= line_no <= center + after

def unprotected_entities(schema, strict=False):

    """returned a set of each non final entity type, excluding "system" entities

    (eg CWGroup, CWUser...)

    """

    if strict:

        protected_entities = yams.schema.BASE_TYPES

    else:

        protected_entities = yams.schema.BASE_TYPES.union(SYSTEM_ENTITIES)

    return set(schema.entities()) - protected_entities

class JsonValidator(object):

    def parse_string(self, data):

        return json.loads(data)

# email handling, to test emails sent by an application ########################

MAILBOX = []

class Email:

    """you'll get instances of Email into MAILBOX during tests that trigger

    some notification.

    * `msg` is the original message object

    * `recipients` is a list of email address which are the recipients of this

      message

    """

    def __init__(self, recipients, msg):

        self.recipients = recipients

        self.msg = msg

    @property

    def message(self):

        return message_from_string(self.msg)

    @property

    def subject(self):

        return self.message.get('Subject')

    @property

    def content(self):

        return self.message.get_payload(decode=True)

    def __repr__(self):

        return '<Email to %s with subject %s>' % (','.join(self.recipients),

                                                  self.message.get('Subject'))

# the trick to get email into MAILBOX instead of actually sent: monkey patch

# cwconfig.SMTP object

class MockSMTP:

    def __init__(self, server, port):

        pass

    def close(self):

        pass

    def sendmail(self, helo_addr, recipients, msg):

        MAILBOX.append(Email(recipients, msg))

cwconfig.SMTP = MockSMTP

class TestCaseConnectionProxy(object):

    """thin wrapper around `cubicweb.dbapi.Connection` context-manager

    used in CubicWebTC (cf. `cubicweb.devtools.testlib.CubicWebTC.login` method)

    It just proxies to the default connection context manager but

    restores the original connection on exit.

    """

    def __init__(self, testcase, cnx):

        self.testcase = testcase

        self.cnx = cnx

    def __getattr__(self, attrname):

        return getattr(self.cnx, attrname)

    def __enter__(self):

        # already open

        return self.cnx

    def __exit__(self, exctype, exc, tb):

        try:

            return self.cnx.__exit__(exctype, exc, tb)

        finally:

            self.testcase.restore_connection()

# base class for cubicweb tests requiring a full cw environments ###############

class CubicWebTC(TestCase):

    """abstract class for test using an apptest environment

    attributes:

    * `vreg`, the vregistry

    * `schema`, self.vreg.schema

    * `config`, cubicweb configuration

    * `cnx`, dbapi connection to the repository using an admin user

    * `session`, server side session associated to `cnx`

    * `app`, the cubicweb publisher (for web testing)

    * `repo`, the repository object

    * `admlogin`, login of the admin user

    * `admpassword`, password of the admin user

    * `shell`, create and use shell environment

    """

    appid = 'data'

    configcls = devtools.ApptestConfiguration

    tags = TestCase.tags | Tags('cubicweb', 'cw_repo')

    test_db_id = DEFAULT_EMPTY_DB_ID

    _cnxs = set() # establised connection

                  # stay on connection for leak detection purpose

    def __init__(self, *args, **kwargs):

        self._admin_session = None

        self._admin_clt_cnx = None

        self._current_session = None

        self._current_clt_cnx = None

        self.repo = None

        super(CubicWebTC, self).__init__(*args, **kwargs)

    # repository connection handling ###########################################

    def set_cnx(self, cnx):

        # XXX we want to deprecate this

        assert getattr(cnx, '_session', None) is not None

        if cnx is self._admin_clt_cnx:

            self._pop_custom_cnx()

        else:

            self._cnxs.add(cnx) # register the cns to make sure it is removed

            self._current_session = cnx._session

            self._current_clt_cnx = cnx

    @property

    def cnx(self):

        # XXX we want to deprecate this

        clt_cnx = self._current_clt_cnx

        if clt_cnx is None:

            clt_cnx = self._admin_clt_cnx

        return clt_cnx

    def _close_cnx(self):

        """ensure that all cnx used by a test have been closed"""

        for cnx in list(self._cnxs):

            if cnx._open and not cnx._session.closed:

                cnx.rollback()

                cnx.close()

            self._cnxs.remove(cnx)

    @property

    def session(self):

        """return current server side session"""

        # XXX We want to use a srv_connection instead and deprecate this

        # property

        session = self._current_session

        if session is None:

            session = self._admin_session

        session.set_cnxset()

        return session

    @property

    def websession(self):

        return self.session

    @property

    def adminsession(self):

        """return current server side session (using default manager account)"""

        return self._admin_session

    def login(self, login, **kwargs):

        """return a connection for the given login/password"""

        __ = kwargs.pop('autoclose', True) # not used anymore

        if login == self.admlogin:

            # definitly don't want autoclose when used as a context manager

            clt_cnx = repoapi.ClientConnection(self._admin_session)

        else:

            if not kwargs:

                kwargs['password'] = str(login)

            clt_cnx = repoapi.connect(self.repo, login, **kwargs)

        self.set_cnx(clt_cnx)

        clt_cnx.__enter__()

        return TestCaseConnectionProxy(self, clt_cnx)

    def restore_connection(self):

        self._pop_custom_cnx()

    def _pop_custom_cnx(self):

        if self._current_clt_cnx is not None:

            if self._current_clt_cnx._open:

                self._current_clt_cnx.close()

            if not  self._current_session.closed:

                self.repo.close(self._current_session.id)

            self._current_clt_cnx = None

            self._current_session = None

    #XXX this doesn't need to a be classmethod anymore

    def _init_repo(self):

        """init the repository and connection to it.

        """

        # setup configuration for test

        self.init_config(self.config)

        # get or restore and working db.

        db_handler = devtools.get_test_db_handler(self.config)

        db_handler.build_db_cache(self.test_db_id, self.pre_setup_database)

        db_handler.restore_database(self.test_db_id)

        self.repo = db_handler.get_repo(startup=True)

        # get an admin session (without actual login)

        sources = db_handler.config.sources()

        login = unicode(sources['admin']['login'])

        with self.repo.internal_session() as session:

            rset = session.execute('CWUser U WHERE U login %(u)s', {'u': login})

            user = rset.get_entity(0, 0)

            user.groups

            user.properties

            from cubicweb.server.session import Session

            self._admin_session = Session(user, self.repo)

            self.repo._sessions[self._admin_session.id] =  self._admin_session

            self._admin_session.user._cw = self._admin_session

        self._admin_clt_cnx = repoapi.ClientConnection(self._admin_session)

        # no direct assignation to cls.cnx anymore.

        # cnx is now an instance property that use a class protected attributes.

        self._cnxs.add(self._admin_clt_cnx)

        self._admin_clt_cnx.__enter__()

        self.config.repository = lambda x=None: self.repo

    # db api ##################################################################

    @nocoverage

    def cursor(self, req=None):

        if req is not None:

            return req.cnx

        else:

            return self.cnx

    @nocoverage

    def execute(self, rql, args=None, eidkey=None, req=None):

        """executes <rql>, builds a resultset, and returns a couple (rset, req)

        where req is a FakeRequest

        """

        if eidkey is not None:

            warn('[3.8] eidkey is deprecated, you can safely remove this argument',

                 DeprecationWarning, stacklevel=2)

        req = req or self.request(rql=rql)

        return req.execute(unicode(rql), args)

    @nocoverage

    def commit(self):

        try:

            return self.cnx.commit()

        finally:

            self.session.set_cnxset() # ensure cnxset still set after commit

    @nocoverage

    def rollback(self):

        try:

            self.cnx.rollback()

        except dbapi.ProgrammingError:

            pass # connection closed

        finally:

            self.session.set_cnxset() # ensure cnxset still set after commit

    requestcls = fake.FakeRequest

    def request(self, rollbackfirst=False, url=None, headers={}, **kwargs):

        """return a web ui request"""

        if rollbackfirst:

            self.cnx.rollback()

        req = self.requestcls(self.vreg, url=url, headers=headers, form=kwargs)

        req.set_cnx(self.cnx)

        return req

    # server side db api #######################################################

    def sexecute(self, rql, args=None, eid_key=None):

        if eid_key is not None:

            warn('[3.8] eid_key is deprecated, you can safely remove this argument',

                 DeprecationWarning, stacklevel=2)

        self.session.set_cnxset()

        return self.session.execute(rql, args)

    # config management ########################################################

    @classproperty

    def config(cls):

        """return the configuration object

        Configuration is cached on the test class.

        """

        try:

            assert not cls is CubicWebTC, "Don't use CubicWebTC directly to prevent database caching issue"

            return cls.__dict__['_config']

        except KeyError:

            home = abspath(join(dirname(sys.modules[cls.__module__].__file__), cls.appid))

            config = cls._config = cls.configcls(cls.appid, apphome=home)

            config.mode = 'test'

            return config

    @classmethod

    def init_config(cls, config):

        """configuration initialization hooks.

        You may only want to override here the configuraton logic.

        Otherwise, consider to use a different :class:`ApptestConfiguration`

        defined in the `configcls` class attribute"""

        source = config.sources()['system']

        cls.admlogin = unicode(source['db-user'])

        cls.admpassword = source['db-password']

        # uncomment the line below if you want rql queries to be logged

        #config.global_set_option('query-log-file',

        #                         '/tmp/test_rql_log.' + `os.getpid()`)

        config.global_set_option('log-file', None)

        # set default-dest-addrs to a dumb email address to avoid mailbox or

        # mail queue pollution

        config.global_set_option('default-dest-addrs', ['whatever'])