# -*- coding: iso-8859-1 -*-

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

"""unit tests for module cubicweb.server.repository"""

import threading

import time

import logging

from yams.constraints import UniqueConstraint

from yams import register_base_type, unregister_base_type

from logilab.database import get_db_helper

from cubicweb import (BadConnectionId, ValidationError,

                      UnknownEid, AuthenticationError, Unauthorized, QueryError)

from cubicweb.predicates import is_instance

from cubicweb.schema import RQLConstraint

from cubicweb.devtools.testlib import CubicWebTC

from cubicweb.devtools.repotest import tuplify

from cubicweb.server import hook

from cubicweb.server.sqlutils import SQL_PREFIX

from cubicweb.server.hook import Hook

from cubicweb.server.sources import native

from cubicweb.server.session import SessionClosedError

class RepositoryTC(CubicWebTC):

    """ singleton providing access to a persistent storage for entities

    and relation

    """

    def test_unique_together_constraint(self):

        with self.admin_access.repo_cnx() as cnx:

            cnx.execute('INSERT Societe S: S nom "Logilab", S type "SSLL", S cp "75013"')

            with self.assertRaises(ValidationError) as wraperr:

                cnx.execute('INSERT Societe S: S nom "Logilab", S type "SSLL", S cp "75013"')

            self.assertEqual(

                {'cp': u'%(KEY-rtype)s is part of violated unicity constraint',

                 'nom': u'%(KEY-rtype)s is part of violated unicity constraint',

                 'type': u'%(KEY-rtype)s is part of violated unicity constraint',

                 '': u'some relations violate a unicity constraint'},

                wraperr.exception.args[1])

    def test_unique_together_schema(self):

        person = self.repo.schema.eschema('Personne')

        self.assertEqual(len(person._unique_together), 1)

        self.assertItemsEqual(person._unique_together[0],

                                           ('nom', 'prenom', 'inline2'))

    def test_all_entities_have_owner(self):

        with self.admin_access.repo_cnx() as cnx:

            self.assertFalse(cnx.execute('Any X WHERE NOT X owned_by U'))

    def test_all_entities_have_is(self):

        with self.admin_access.repo_cnx() as cnx:

            self.assertFalse(cnx.execute('Any X WHERE NOT X is ET'))

    def test_all_entities_have_cw_source(self):

        with self.admin_access.repo_cnx() as cnx:

            self.assertFalse(cnx.execute('Any X WHERE NOT X cw_source S'))

    def test_connect(self):

        cnxid = self.repo.connect(self.admlogin, password=self.admpassword)

        self.assert_(cnxid)

        self.repo.close(cnxid)

        self.assertRaises(AuthenticationError,

                          self.repo.connect, self.admlogin, password='nimportnawak')

        self.assertRaises(AuthenticationError,

                          self.repo.connect, self.admlogin, password='')

        self.assertRaises(AuthenticationError,

                          self.repo.connect, self.admlogin, password=None)

        self.assertRaises(AuthenticationError,

                          self.repo.connect, None, password=None)

        self.assertRaises(AuthenticationError,

                          self.repo.connect, self.admlogin)

        self.assertRaises(AuthenticationError,

                          self.repo.connect, None)

        repo = self.repo

        cnxid = repo.connect(u"barnab�", password=u"h�h�h�".encode('UTF8'))

        self.assert_(cnxid)

        repo.close(cnxid)

    def test_rollback_on_execute_validation_error(self):

        class ValidationErrorAfterHook(Hook):

            __regid__ = 'valerror-after-hook'

            __select__ = Hook.__select__ & is_instance('CWGroup')

            events = ('after_update_entity',)

            def __call__(self):

                raise ValidationError(self.entity.eid, {})

        with self.admin_access.repo_cnx() as cnx:

            with self.temporary_appobjects(ValidationErrorAfterHook):

                self.assertRaises(ValidationError,

                                  cnx.execute, 'SET X name "toto" WHERE X is CWGroup, X name "guests"')

                self.assertTrue(cnx.execute('Any X WHERE X is CWGroup, X name "toto"'))

                with self.assertRaises(QueryError) as cm:

                    cnx.commit()

                self.assertEqual(str(cm.exception), 'transaction must be rolled back')

                cnx.rollback()

                self.assertFalse(cnx.execute('Any X WHERE X is CWGroup, X name "toto"'))

    def test_rollback_on_execute_unauthorized(self):

        class UnauthorizedAfterHook(Hook):

            __regid__ = 'unauthorized-after-hook'

            __select__ = Hook.__select__ & is_instance('CWGroup')

            events = ('after_update_entity',)

            def __call__(self):

                raise Unauthorized()

        with self.admin_access.repo_cnx() as cnx:

            with self.temporary_appobjects(UnauthorizedAfterHook):

                self.assertRaises(Unauthorized,

                                  cnx.execute, 'SET X name "toto" WHERE X is CWGroup, X name "guests"')

                self.assertTrue(cnx.execute('Any X WHERE X is CWGroup, X name "toto"'))

                with self.assertRaises(QueryError) as cm:

                    cnx.commit()

                self.assertEqual(str(cm.exception), 'transaction must be rolled back')

                cnx.rollback()

                self.assertFalse(cnx.execute('Any X WHERE X is CWGroup, X name "toto"'))

    def test_close(self):

        repo = self.repo

        cnxid = repo.connect(self.admlogin, password=self.admpassword)

        self.assert_(cnxid)

        repo.close(cnxid)

    def test_shared_data(self):

        repo = self.repo

        cnxid = repo.connect(self.admlogin, password=self.admpassword)

        repo.set_shared_data(cnxid, 'data', 4)

        cnxid2 = repo.connect(self.admlogin, password=self.admpassword)

        self.assertEqual(repo.get_shared_data(cnxid, 'data'), 4)

        self.assertEqual(repo.get_shared_data(cnxid2, 'data'), None)

        repo.set_shared_data(cnxid2, 'data', 5)

        self.assertEqual(repo.get_shared_data(cnxid, 'data'), 4)

        self.assertEqual(repo.get_shared_data(cnxid2, 'data'), 5)

        repo.get_shared_data(cnxid2, 'data', pop=True)

        self.assertEqual(repo.get_shared_data(cnxid, 'data'), 4)

        self.assertEqual(repo.get_shared_data(cnxid2, 'data'), None)

        repo.close(cnxid)

        repo.close(cnxid2)

        self.assertRaises(BadConnectionId, repo.get_shared_data, cnxid, 'data')

        self.assertRaises(BadConnectionId, repo.get_shared_data, cnxid2, 'data')

        self.assertRaises(BadConnectionId, repo.set_shared_data, cnxid, 'data', 1)

        self.assertRaises(BadConnectionId, repo.set_shared_data, cnxid2, 'data', 1)

    def test_check_session(self):

        repo = self.repo

        cnxid = repo.connect(self.admlogin, password=self.admpassword)

        self.assertIsInstance(repo.check_session(cnxid), float)

        repo.close(cnxid)

        self.assertRaises(BadConnectionId, repo.check_session, cnxid)

    def test_initial_schema(self):

        schema = self.repo.schema

        # check order of attributes is respected

        notin = set(('eid', 'is', 'is_instance_of', 'identity',

                     'creation_date', 'modification_date', 'cwuri',

                     'owned_by', 'created_by', 'cw_source',

                     'update_permission', 'read_permission',

                     'add_permission', 'in_basket'))

        self.assertListEqual(['relation_type',

                              'from_entity', 'to_entity',

                              'constrained_by',

                              'cardinality', 'ordernum', 'formula',

                              'indexed', 'fulltextindexed', 'internationalizable',

                              'defaultval', 'extra_props',

                              'description', 'description_format'],

                             [r.type

                              for r in schema.eschema('CWAttribute').ordered_relations()

                              if r.type not in notin])

        self.assertEqual(schema.eschema('CWEType').main_attribute(), 'name')

        self.assertEqual(schema.eschema('State').main_attribute(), 'name')

        constraints = schema.rschema('name').rdef('CWEType', 'String').constraints

        self.assertEqual(len(constraints), 2)

        for cstr in constraints[:]:

            if isinstance(cstr, UniqueConstraint):

                constraints.remove(cstr)

                break

        else:

            self.fail('unique constraint not found')

        sizeconstraint = constraints[0]

        self.assertEqual(sizeconstraint.min, None)

        self.assertEqual(sizeconstraint.max, 64)

        constraints = schema.rschema('relation_type').rdef('CWAttribute', 'CWRType').constraints

        self.assertEqual(len(constraints), 1)

        cstr = constraints[0]

        self.assert_(isinstance(cstr, RQLConstraint))

        self.assertEqual(cstr.expression, 'O final TRUE')

        ownedby = schema.rschema('owned_by')

        self.assertEqual(ownedby.objects('CWEType'), ('CWUser',))

    def test_internal_api(self):

        repo = self.repo

        cnxid = repo.connect(self.admlogin, password=self.admpassword)

        session = repo._get_session(cnxid, setcnxset=True)

        self.assertEqual(repo.type_and_source_from_eid(2, session),

                         ('CWGroup', None, 'system'))

        self.assertEqual(repo.type_from_eid(2, session), 'CWGroup')

        repo.close(cnxid)

    def test_public_api(self):

        self.assertEqual(self.repo.get_schema(), self.repo.schema)

        self.assertEqual(self.repo.source_defs(), {'system': {'type': 'native',

                                                              'uri': 'system',

                                                              'use-cwuri-as-url': False}

                                                  })

        # .properties() return a result set

        self.assertEqual(self.repo.properties().rql, 'Any K,V WHERE P is CWProperty,P pkey K, P value V, NOT P for_user U')

    def test_session_api(self):

        repo = self.repo

        cnxid = repo.connect(self.admlogin, password=self.admpassword)

        self.assertEqual(repo.user_info(cnxid), (6, 'admin', set([u'managers']), {}))

        self.assertEqual({'type': u'CWGroup', 'extid': None, 'source': 'system'},

                         repo.entity_metas(cnxid, 2))

        self.assertEqual(repo.describe(cnxid, 2), (u'CWGroup', 'system', None, 'system'))

        repo.close(cnxid)

        self.assertRaises(BadConnectionId, repo.user_info, cnxid)

        self.assertRaises(BadConnectionId, repo.describe, cnxid, 1)

    def test_shared_data_api(self):

        repo = self.repo

        cnxid = repo.connect(self.admlogin, password=self.admpassword)

        self.assertEqual(repo.get_shared_data(cnxid, 'data'), None)

        repo.set_shared_data(cnxid, 'data', 4)

        self.assertEqual(repo.get_shared_data(cnxid, 'data'), 4)

        repo.get_shared_data(cnxid, 'data', pop=True)

        repo.get_shared_data(cnxid, 'whatever', pop=True)

        self.assertEqual(repo.get_shared_data(cnxid, 'data'), None)

        repo.close(cnxid)

        self.assertRaises(BadConnectionId, repo.set_shared_data, cnxid, 'data', 0)

        self.assertRaises(BadConnectionId, repo.get_shared_data, cnxid, 'data')

    def test_schema_is_relation(self):

        with self.admin_access.repo_cnx() as cnx:

            no_is_rset = cnx.execute('Any X WHERE NOT X is ET')

            self.assertFalse(no_is_rset, no_is_rset.description)

#     def test_perfo(self):

#         self.set_debug(True)

#         from time import time, clock

#         t, c = time(), clock()

#         try:

#             self.create_user('toto')

#         finally:

#             self.set_debug(False)

#         print 'test time: %.3f (time) %.3f (cpu)' % ((time() - t), clock() - c)

    def test_delete_if_singlecard1(self):

        with self.admin_access.repo_cnx() as cnx:

            note = cnx.create_entity('Affaire')

            p1 = cnx.create_entity('Personne', nom=u'toto')

            cnx.execute('SET A todo_by P WHERE A eid %(x)s, P eid %(p)s',

                        {'x': note.eid, 'p': p1.eid})

            rset = cnx.execute('Any P WHERE A todo_by P, A eid %(x)s',

                               {'x': note.eid})

            self.assertEqual(len(rset), 1)

            p2 = cnx.create_entity('Personne', nom=u'tutu')

            cnx.execute('SET A todo_by P WHERE A eid %(x)s, P eid %(p)s',

                        {'x': note.eid, 'p': p2.eid})

            rset = cnx.execute('Any P WHERE A todo_by P, A eid %(x)s',

                                {'x': note.eid})

            self.assertEqual(len(rset), 1)

            self.assertEqual(rset.rows[0][0], p2.eid)

    def test_delete_if_object_inlined_singlecard(self):

        with self.admin_access.repo_cnx() as cnx:

            c = cnx.create_entity('Card', title=u'Carte')

            cnx.create_entity('Personne', nom=u'Vincent', fiche=c)

            cnx.create_entity('Personne', nom=u'Florent', fiche=c)

            cnx.commit()

            self.assertEqual(len(c.reverse_fiche), 1)

    def test_delete_computed_relation_nonregr(self):

        with self.admin_access.repo_cnx() as cnx:

            c = cnx.create_entity('Personne', nom=u'Adam', login_user=cnx.user.eid)

            cnx.commit()

            c.cw_delete()

            cnx.commit()

    def test_cw_set_in_before_update(self):

        # local hook

        class DummyBeforeHook(Hook):

            __regid__ = 'dummy-before-hook'

            __select__ = Hook.__select__ & is_instance('EmailAddress')

            events = ('before_update_entity',)

            def __call__(self):

                # safety belt: avoid potential infinite recursion if the test

                #              fails (i.e. RuntimeError not raised)

                pendings = self._cw.transaction_data.setdefault('pending', set())

                if self.entity.eid not in pendings:

                    pendings.add(self.entity.eid)

                    self.entity.cw_set(alias=u'foo')

        with self.admin_access.repo_cnx() as cnx:

            with self.temporary_appobjects(DummyBeforeHook):

                addr = cnx.create_entity('EmailAddress', address=u'a@b.fr')

                addr.cw_set(address=u'a@b.com')

                rset = cnx.execute('Any A,AA WHERE X eid %(x)s, X address A, X alias AA',

                                   {'x': addr.eid})

                self.assertEqual(rset.rows, [[u'a@b.com', u'foo']])

    def test_cw_set_in_before_add(self):

        # local hook

        class DummyBeforeHook(Hook):

            __regid__ = 'dummy-before-hook'

            __select__ = Hook.__select__ & is_instance('EmailAddress')

            events = ('before_add_entity',)

            def __call__(self):

                # cw_set is forbidden within before_add_entity()

                self.entity.cw_set(alias=u'foo')

        with self.admin_access.repo_cnx() as cnx:

            with self.temporary_appobjects(DummyBeforeHook):

                # XXX will fail with python -O

                self.assertRaises(AssertionError, cnx.create_entity,

                                  'EmailAddress', address=u'a@b.fr')