from logilab.common.testlib import TestCase, unittest_main, mock_object

from cubicweb.devtools import init_test_database, TestServerConfiguration

from cubicweb.devtools.apptest import RepositoryBasedTC

from cubicweb.devtools.repotest import RQLGeneratorTC

from cubicweb.server.sources.ldapuser import *

def nopwd_authenticate(self, session, login, upassword):

    """used to monkey patch the source to get successful authentication without

    upassword checking

    """

    assert login, 'no login!'

    searchfilter = [filter_format('(%s=%s)', (self.user_login_attr, login))]

    searchfilter.extend([filter_format('(%s=%s)', ('objectClass', o))

                         for o in self.user_classes])

    searchstr = '(&%s)' % ''.join(searchfilter)

    # first search the user

    try:

        user = self._search(session, self.user_base_dn, self.user_base_scope,

                            searchstr)[0]

    except IndexError:

        # no such user

        raise AuthenticationError()

    # don't check upassword !

    return self.extid2eid(user['dn'], 'CWUser', session)

config = TestServerConfiguration('data')

config.sources_file = lambda : 'data/sourcesldap'

repo, cnx = init_test_database('sqlite', config=config)

class LDAPUserSourceTC(RepositoryBasedTC):

    repo, cnx = repo, cnx

    def patch_authenticate(self):

        self._orig_authenticate = LDAPUserSource.authenticate

        LDAPUserSource.authenticate = nopwd_authenticate

    def setUp(self):

        self._prepare()

        # XXX: need this first query else we get 'database is locked' from

        # sqlite since it doesn't support multiple connections on the same

        # database

        # so doing, ldap inserted users don't get removed between each test

        rset = self.execute('CWUser X')

        self.commit()

        # check we get some users from ldap

        self.assert_(len(rset) > 1)

        self.maxeid = self.execute('Any MAX(X)')[0][0]

    def tearDown(self):

        if hasattr(self, '_orig_authenticate'):

            LDAPUserSource.authenticate = self._orig_authenticate

        RepositoryBasedTC.tearDown(self)

    def test_authenticate(self):

        source = self.repo.sources_by_uri['ldapuser']

        self.assertRaises(AuthenticationError,

                          source.authenticate, self.session, 'toto', 'toto')

    def test_synchronize(self):

        source = self.repo.sources_by_uri['ldapuser']

        source.synchronize()

    def test_base(self):

        # check a known one

        e = self.execute('CWUser X WHERE X login "syt"').get_entity(0, 0)

        self.assertEquals(e.login, 'syt')

        e.complete()

        self.assertEquals(e.creation_date, None)

        self.assertEquals(e.modification_date, None)

        self.assertEquals(e.firstname, None)

        self.assertEquals(e.surname, None)

        self.assertEquals(e.in_group[0].name, 'users')

        self.assertEquals(e.owned_by[0].login, 'syt')

        self.assertEquals(e.created_by, [])

        self.assertEquals(e.primary_email[0].address, 'Sylvain Thenault')

        # email content should be indexed on the user

        rset = self.execute('CWUser X WHERE X has_text "thenault"')

        self.assertEquals(rset.rows, [[e.eid]])

    def test_not(self):

        eid = self.execute('CWUser X WHERE X login "syt"')[0][0]

        rset = self.execute('CWUser X WHERE NOT X eid %s' % eid)

        self.assert_(rset)

        self.assert_(not eid in (r[0] for r in rset))

    def test_multiple(self):

        seid = self.execute('CWUser X WHERE X login "syt"')[0][0]

        aeid = self.execute('CWUser X WHERE X login "adim"')[0][0]

        rset = self.execute('CWUser X, Y WHERE X login "syt", Y login "adim"')

        self.assertEquals(rset.rows, [[seid, aeid]])

        rset = self.execute('Any X,Y,L WHERE X login L, X login "syt", Y login "adim"')

        self.assertEquals(rset.rows, [[seid, aeid, 'syt']])

    def test_in(self):

        seid = self.execute('CWUser X WHERE X login "syt"')[0][0]

        aeid = self.execute('CWUser X WHERE X login "adim"')[0][0]

        rset = self.execute('Any X,L ORDERBY L WHERE X login IN("syt", "adim"), X login L')

        self.assertEquals(rset.rows, [[aeid, 'adim'], [seid, 'syt']])

    def test_relations(self):

        eid = self.execute('CWUser X WHERE X login "syt"')[0][0]

        rset = self.execute('Any X,E WHERE X is CWUser, X login L, X primary_email E')

        self.assert_(eid in (r[0] for r in rset))

        rset = self.execute('Any X,L,E WHERE X is CWUser, X login L, X primary_email E')

        self.assert_('syt' in (r[1] for r in rset))

    def test_count(self):

        nbusers = self.execute('Any COUNT(X) WHERE X is CWUser')[0][0]

        # just check this is a possible number

        self.assert_(nbusers > 1, nbusers)

        self.assert_(nbusers < 30, nbusers)

    def test_upper(self):

        eid = self.execute('CWUser X WHERE X login "syt"')[0][0]

        rset = self.execute('Any UPPER(L) WHERE X eid %s, X login L' % eid)

        self.assertEquals(rset[0][0], 'SYT')

    def test_unknown_attr(self):

        eid = self.execute('CWUser X WHERE X login "syt"')[0][0]

        rset = self.execute('Any L,C,M WHERE X eid %s, X login L, '

                            'X creation_date C, X modification_date M' % eid)

        self.assertEquals(rset[0][0], 'syt')

        self.assertEquals(rset[0][1], None)

        self.assertEquals(rset[0][2], None)

    def test_sort(self):

        logins = [l for l, in self.execute('Any L ORDERBY L WHERE X login L')]

        self.assertEquals(logins, sorted(logins))

    def test_lower_sort(self):

        logins = [l for l, in self.execute('Any L ORDERBY lower(L) WHERE X login L')]

        self.assertEquals(logins, sorted(logins))

    def test_or(self):

        rset = self.execute('DISTINCT Any X WHERE X login "syt" OR (X in_group G, G name "managers")')

        self.assertEquals(len(rset), 2, rset.rows) # syt + admin

    def test_nonregr_set_owned_by(self):

        # test that when a user coming from ldap is triggering a transition

        # the related TrInfo has correct owner information

        self.execute('SET X in_group G WHERE X login "syt", G name "managers"')

        self.commit()

        syt = self.execute('CWUser X WHERE X login "syt"').get_entity(0, 0)

        self.assertEquals([g.name for g in syt.in_group], ['managers', 'users'])

        self.patch_authenticate()

        cnx = self.login('syt', 'dummypassword')

        cu = cnx.cursor()

        cu.execute('SET X in_state S WHERE X login "alf", S name "deactivated"')

        try:

            cnx.commit()

            alf = self.execute('CWUser X WHERE X login "alf"').get_entity(0, 0)

            self.assertEquals(alf.in_state[0].name, 'deactivated')

            trinfo = alf.latest_trinfo()

            self.assertEquals(trinfo.owned_by[0].login, 'syt')

            # select from_state to skip the user's creation TrInfo

            rset = self.execute('Any U ORDERBY D DESC WHERE WF wf_info_for X,'

                                'WF creation_date D, WF from_state FS,'

                                'WF owned_by U?, X eid %(x)s',

                                {'x': alf.eid}, 'x')

            self.assertEquals(rset.rows, [[syt.eid]])

        finally:

            # restore db state

            self.restore_connection()

            self.execute('SET X in_state S WHERE X login "alf", S name "activated"')

            self.execute('DELETE X in_group G WHERE X login "syt", G name "managers"')

    def test_same_column_names(self):

        self.execute('Any X, Y WHERE X copain Y, X login "comme", Y login "cochon"')

    def test_multiple_entities_from_different_sources(self):

        self.create_user('cochon')

        self.failUnless(self.execute('Any X,Y WHERE X login "syt", Y login "cochon"'))

    def test_exists1(self):

        self.add_entity('CWGroup', name=u'bougloup1')

        self.add_entity('CWGroup', name=u'bougloup2')

        self.execute('SET U in_group G WHERE G name ~= "bougloup%", U login "admin"')

        self.execute('SET U in_group G WHERE G name = "bougloup1", U login "syt"')

        rset = self.execute('Any L,SN ORDERBY L WHERE X in_state S, S name SN, X login L, EXISTS(X in_group G, G name ~= "bougloup%")')

        self.assertEquals(rset.rows, [['admin', 'activated'], ['syt', 'activated']])

    def test_exists2(self):

        self.create_user('comme')

        self.create_user('cochon')

        self.execute('SET X copain Y WHERE X login "comme", Y login "cochon"')

        rset = self.execute('Any GN ORDERBY GN WHERE X in_group G, G name GN, (G name "managers" OR EXISTS(X copain T, T login in ("comme", "cochon")))')

        self.assertEquals(rset.rows, [['managers'], ['users']])

    def test_exists3(self):

        self.create_user('comme')

        self.create_user('cochon')

        self.execute('SET X copain Y WHERE X login "comme", Y login "cochon"')

        self.failUnless(self.execute('Any X, Y WHERE X copain Y, X login "comme", Y login "cochon"'))

        self.execute('SET X copain Y WHERE X login "syt", Y login "cochon"')

        self.failUnless(self.execute('Any X, Y WHERE X copain Y, X login "syt", Y login "cochon"'))

        rset = self.execute('Any GN,L WHERE X in_group G, X login L, G name GN, G name "managers" OR EXISTS(X copain T, T login in ("comme", "cochon"))')

        self.assertEquals(sorted(rset.rows), [['managers', 'admin'], ['users', 'comme'], ['users', 'syt']])

    def test_exists4(self):

        self.create_user('comme')

        self.create_user('cochon', groups=('users', 'guests'))

        self.create_user('billy')

        self.execute('SET X copain Y WHERE X login "comme", Y login "cochon"')

        self.execute('SET X copain Y WHERE X login "cochon", Y login "cochon"')

        self.execute('SET X copain Y WHERE X login "comme", Y login "billy"')

        self.execute('SET X copain Y WHERE X login "syt", Y login "billy"')

        # search for group name, login where

        #   CWUser copain with "comme" or "cochon" AND same login as the copain

        # OR

        #   CWUser in_state activated AND not copain with billy

        #

        # SO we expect everybody but "comme" and "syt"

        rset= self.execute('Any GN,L WHERE X in_group G, X login L, G name GN, '

                           'EXISTS(X copain T, T login L, T login in ("comme", "cochon")) OR '

                           'EXISTS(X in_state S, S name "activated", NOT X copain T2, T2 login "billy")')

        all = self.execute('Any GN, L WHERE X in_group G, X login L, G name GN')

        all.rows.remove(['users', 'comme'])

        all.rows.remove(['users', 'syt'])

        self.assertEquals(sorted(rset.rows), sorted(all.rows))

    def test_exists5(self):

        self.create_user('comme')

        self.create_user('cochon', groups=('users', 'guests'))

        self.create_user('billy')

        self.execute('SET X copain Y WHERE X login "comme", Y login "cochon"')

        self.execute('SET X copain Y WHERE X login "cochon", Y login "cochon"')

        self.execute('SET X copain Y WHERE X login "comme", Y login "billy"')

        self.execute('SET X copain Y WHERE X login "syt", Y login "cochon"')

        rset= self.execute('Any L WHERE X login L, '

                           'EXISTS(X copain T, T login in ("comme", "cochon")) AND '

                           'NOT EXISTS(X copain T2, T2 login "billy")')

        self.assertEquals(sorted(rset.rows), [['cochon'], ['syt']])

        rset= self.execute('Any GN,L WHERE X in_group G, X login L, G name GN, '

                           'EXISTS(X copain T, T login in ("comme", "cochon")) AND '

                           'NOT EXISTS(X copain T2, T2 login "billy")')

        self.assertEquals(sorted(rset.rows), [['guests', 'cochon'],

                                              ['users', 'cochon'],

                                              ['users', 'syt']])

    def test_cd_restriction(self):

        rset = self.execute('CWUser X WHERE X creation_date > "2009-02-01"')

        self.assertEquals(len(rset), 2) # admin/anon but no ldap user since it doesn't support creation_date

    def test_union(self):

        afeids = self.execute('State X')

        ueids = self.execute('CWUser X')

        rset = self.execute('(Any X WHERE X is State) UNION (Any X WHERE X is CWUser)')

        self.assertEquals(sorted(r[0] for r in rset.rows),

                          sorted(r[0] for r in afeids + ueids))

    def _init_security_test(self):

        self.create_user('iaminguestsgrouponly', groups=('guests',))

        cnx = self.login('iaminguestsgrouponly')

        return cnx.cursor()

    def test_security1(self):

        cu = self._init_security_test()

        rset = cu.execute('Any X WHERE X login "syt"')

        self.assertEquals(rset.rows, [])

        rset = cu.execute('Any X WHERE X login "iaminguestsgrouponly"')

        self.assertEquals(len(rset.rows), 1)

    def test_security2(self):

        cu = self._init_security_test()

        rset = cu.execute('Any X WHERE X has_text "syt"')

        self.assertEquals(rset.rows, [])

        rset = cu.execute('Any X WHERE X has_text "iaminguestsgrouponly"')

        self.assertEquals(len(rset.rows), 1)

    def test_security3(self):

        cu = self._init_security_test()

        rset = cu.execute('Any F WHERE X has_text "syt", X firstname F')

        self.assertEquals(rset.rows, [])

        rset = cu.execute('Any F WHERE X has_text "iaminguestsgrouponly", X firstname F')

        self.assertEquals(rset.rows, [[None]])

    def test_nonregr1(self):

        self.execute('Any X,AA ORDERBY AA DESC WHERE E eid %(x)s, E owned_by X, '

                     'X modification_date AA',

                     {'x': cnx.user(self.session).eid})

    def test_nonregr2(self):

        self.execute('Any X,L,AA WHERE E eid %(x)s, E owned_by X, '

                     'X login L, X modification_date AA',

                     {'x': cnx.user(self.session).eid})

    def test_nonregr3(self):

        self.execute('Any X,AA ORDERBY AA DESC WHERE E eid %(x)s, '

                     'X modification_date AA',

                     {'x': cnx.user(self.session).eid})

    def test_nonregr4(self):

        emaileid = self.execute('INSERT EmailAddress X: X address "toto@logilab.org"')[0][0]

        self.execute('Any X,AA WHERE X use_email Y, Y eid %(x)s, X modification_date AA',

                     {'x': emaileid})

    def test_nonregr5(self):

        # original jpl query:

        # Any X, NOW - CD, P WHERE P is Project, U interested_in P, U is CWUser, U login "sthenault", X concerns P, X creation_date CD ORDERBY CD DESC LIMIT 5

        rql = 'Any X, NOW - CD, P ORDERBY CD DESC LIMIT 5 WHERE P bookmarked_by U, U login "%s", P is X, X creation_date CD' % self.session.user.login

        self.execute(rql, )#{'x': })

    def test_nonregr6(self):

        self.execute('Any B,U,UL GROUPBY B,U,UL WHERE B created_by U?, B is File '

                     'WITH U,UL BEING (Any U,UL WHERE ME eid %(x)s, (EXISTS(U identity ME) '

                     'OR (EXISTS(U in_group G, G name IN("managers", "staff")))) '

                     'OR (EXISTS(U in_group H, ME in_group H, NOT H name "users")), U login UL, U is CWUser)',

                     {'x': self.session.user.eid})

class GlobTrFuncTC(TestCase):

    def test_count(self):

        trfunc = GlobTrFunc('count', 0)

        res = trfunc.apply([[1], [2], [3], [4]])

        self.assertEquals(res, [[4]])

        trfunc = GlobTrFunc('count', 1)

        res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]])

        self.assertEquals(res, [[1, 2], [2, 1], [3, 1]])

    def test_sum(self):

        trfunc = GlobTrFunc('sum', 0)

        res = trfunc.apply([[1], [2], [3], [4]])

        self.assertEquals(res, [[10]])

        trfunc = GlobTrFunc('sum', 1)

        res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]])

        self.assertEquals(res, [[1, 7], [2, 4], [3, 6]])

    def test_min(self):

        trfunc = GlobTrFunc('min', 0)

        res = trfunc.apply([[1], [2], [3], [4]])

        self.assertEquals(res, [[1]])

        trfunc = GlobTrFunc('min', 1)

        res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]])

        self.assertEquals(res, [[1, 2], [2, 4], [3, 6]])

    def test_max(self):

        trfunc = GlobTrFunc('max', 0)

        res = trfunc.apply([[1], [2], [3], [4]])

        self.assertEquals(res, [[4]])

        trfunc = GlobTrFunc('max', 1)

        res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]])

        self.assertEquals(res, [[1, 5], [2, 4], [3, 6]])

class RQL2LDAPFilterTC(RQLGeneratorTC):

    schema = repo.schema