"""unit tests for selectors mechanism

:organization: Logilab

:copyright: 2001-2009 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr

"""

from logilab.common.testlib import TestCase, unittest_main

from cubicweb.devtools.testlib import EnvBasedTC

from cubicweb.vregistry import Selector, AndSelector, OrSelector

from cubicweb.selectors import implements, match_user_groups

from cubicweb.interfaces import IDownloadable

from cubicweb.web import action

class _1_(Selector):

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

        return 1

class _0_(Selector):

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

        return 0

def _2_(*args, **kwargs):

    return 2

class SelectorsTC(TestCase):

    def test_basic_and(self):

        selector = _1_() & _1_()

        self.assertEquals(selector(None), 2)

        selector = _1_() & _0_()

        self.assertEquals(selector(None), 0)

        selector = _0_() & _1_()

        self.assertEquals(selector(None), 0)

    def test_basic_or(self):

        selector = _1_() | _1_()

        self.assertEquals(selector(None), 1)

        selector = _1_() | _0_()

        self.assertEquals(selector(None), 1)

        selector = _0_() | _1_()

        self.assertEquals(selector(None), 1)

        selector = _0_() | _0_()

        self.assertEquals(selector(None), 0)

    def test_selector_and_function(self):

        selector = _1_() & _2_

        self.assertEquals(selector(None), 3)

        selector = _2_ & _1_()

        self.assertEquals(selector(None), 3)

    def test_three_and(self):

        selector = _1_() & _1_() & _1_()

        self.assertEquals(selector(None), 3)

        selector = _1_() & _0_() & _1_()

        self.assertEquals(selector(None), 0)

        selector = _0_() & _1_() & _1_()

        self.assertEquals(selector(None), 0)

    def test_three_or(self):

        selector = _1_() | _1_() | _1_()

        self.assertEquals(selector(None), 1)

        selector = _1_() | _0_() | _1_()

        self.assertEquals(selector(None), 1)

        selector = _0_() | _1_() | _1_()

        self.assertEquals(selector(None), 1)

        selector = _0_() | _0_() | _0_()

        self.assertEquals(selector(None), 0)

    def test_composition(self):

        selector = (_1_() & _1_()) & (_1_() & _1_())

        self.failUnless(isinstance(selector, AndSelector))

        self.assertEquals(len(selector.selectors), 4)

        self.assertEquals(selector(None), 4)

        selector = (_1_() & _0_()) | (_1_() & _1_())

        self.failUnless(isinstance(selector, OrSelector))

        self.assertEquals(len(selector.selectors), 2)

        self.assertEquals(selector(None), 2)

    def test_search_selectors(self):

        sel = implements('something')

        self.assertIs(sel.search_selector(implements), sel)

        csel = AndSelector(sel, Selector())

        self.assertIs(csel.search_selector(implements), sel)

        csel = AndSelector(Selector(), sel)

        self.assertIs(csel.search_selector(implements), sel)

class ImplementsSelectorTC(EnvBasedTC):

    def test_etype_priority(self):

        req = self.request()

        cls = self.vreg.etype_class('File')

        anyscore = implements('Any').score_class(cls, req)

        idownscore = implements(IDownloadable).score_class(cls, req)

        self.failUnless(idownscore > anyscore, (idownscore, anyscore))

        filescore = implements('File').score_class(cls, req)

        self.failUnless(filescore > idownscore, (filescore, idownscore))

    def test_etype_inheritance_no_yams_inheritance(self):

        cls = self.vreg.etype_class('Personne')

        self.failIf(implements('Societe').score_class(cls, self.request()))

class MatchUserGroupsTC(EnvBasedTC):

    def test_owners_group(self):

        """tests usage of 'owners' group with match_user_group"""

        class SomeAction(action.Action):

            id = 'yo'

            category = 'foo'

            __select__ = match_user_groups('owners')

        self.vreg._loadedmods[__name__] = {}

        self.vreg.register_vobject_class(SomeAction)

        self.failUnless(SomeAction in self.vreg['actions']['yo'], self.vreg['actions'])

        try:

            # login as a simple user

            self.create_user('john')

            self.login('john')

            # it should not be possible to use SomeAction not owned objects

            rset, req = self.env.get_rset_and_req('Any G WHERE G is CWGroup, G name "managers"')

            self.failIf('yo' in dict(self.pactions(req, rset)))

            # insert a new card, and check that we can use SomeAction on our object

            self.execute('INSERT Card C: C title "zoubidou"')

            self.commit()

            rset, req = self.env.get_rset_and_req('Card C WHERE C title "zoubidou"')

            self.failUnless('yo' in dict(self.pactions(req, rset)), self.pactions(req, rset))

            # make sure even managers can't use the action

            self.restore_connection()

            rset, req = self.env.get_rset_and_req('Card C WHERE C title "zoubidou"')

            self.failIf('yo' in dict(self.pactions(req, rset)))

        finally:

            del self.vreg[SomeAction.__registry__][SomeAction.id]

if __name__ == '__main__':

    unittest_main()