# -*- coding: utf-8 -*-

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

"""XXX rename, split, reorganize this"""

from logilab.common.testlib import unittest_main

from cubicweb.devtools.testlib import CubicWebTC

from cubicweb import CW_SOFTWARE_ROOT as BASE, Binary, UnknownProperty

from cubicweb.selectors import (match_user_groups, is_instance,

                                specified_etype_implements, rql_condition,

                                traced_selection)

from cubicweb.web import NoSelectableObject

from cubicweb.web.action import Action

from cubicweb.web.views import (primary, baseviews, tableview, editforms,

                                calendar, management, embedding, actions,

                                startup, cwuser, schema, xbel, vcard, owl,

                                treeview, idownloadable, wdoc, debug,

                                cwproperties, workflow, xmlrss, csvexport)

from cubes.folder import views as folderviews

USERACTIONS = [actions.UserPreferencesAction,

               actions.UserInfoAction,

               actions.LogoutAction]

SITEACTIONS = [actions.SiteConfigurationAction,

               actions.ManageAction,

               schema.ViewSchemaAction,

               debug.SiteInfoAction]

FOOTERACTIONS = [wdoc.HelpAction,

                 wdoc.ChangeLogAction,

                 wdoc.AboutAction,

                 actions.PoweredByAction]

class ViewSelectorTC(CubicWebTC):

    def setup_database(self):

        req = self.request()

        req.create_entity('BlogEntry', title=u"une news !", content=u"cubicweb c'est beau")

        req.create_entity('Bookmark', title=u"un signet !", path=u"view?vid=index")

        req.create_entity('EmailAddress', address=u"devel@logilab.fr", alias=u'devel')

        req.create_entity('Tag', name=u'x')

class VRegistryTC(ViewSelectorTC):

    """test the view selector"""

    def _test_registered(self, registry, content):

        try:

            expected = getattr(self, 'all_%s' % registry)

        except AttributeError:

            return

        if registry == 'hooks':

            self.assertEquals(len(content), expected, content)

            return

        try:

            self.assertSetEqual(content.keys(), expected)

        except:

            print registry, sorted(expected), sorted(content.keys())

            print 'no more', [v for v in expected if not v in content.keys()]

            print 'missing', [v for v in content.keys() if not v in expected]

            raise

    def setUp(self):

        super(VRegistryTC, self).setUp()

        assert self.vreg['views']['propertiesform']

    def test_possible_views_none_rset(self):

        req = self.request()

        self.assertListEqual(self.pviews(req, None),

                             [('changelog', wdoc.ChangeLogView),

                              ('gc', debug.GCView),

                              ('index', startup.IndexView),

                              ('info', debug.ProcessInformationView),

                              ('manage', startup.ManageView),

                              ('owl', owl.OWLView),

                              ('propertiesform', cwproperties.CWPropertiesForm),

                              ('registry', debug.RegistryView),

                              ('schema', schema.SchemaView),

                              ('systempropertiesform', cwproperties.SystemCWPropertiesForm),

                              ('tree', folderviews.FolderTreeView),

                              ])

    def test_possible_views_noresult(self):

        req = self.request()

        rset = req.execute('Any X WHERE X eid 999999')

        self.assertListEqual(self.pviews(req, rset),

                             [])

    def test_possible_views_one_egroup(self):

        req = self.request()

        rset = req.execute('CWGroup X WHERE X name "managers"')

        self.assertListEqual(self.pviews(req, rset),

                             [('csvexport', csvexport.CSVRsetView),

                              ('ecsvexport', csvexport.CSVEntityView),

                              ('editable-table', tableview.EditableTableView),

                              ('filetree', treeview.FileTreeView),

                              ('list', baseviews.ListView),

                              ('oneline', baseviews.OneLineView),

                              ('owlabox', owl.OWLABOXView),

                              ('primary', cwuser.CWGroupPrimaryView),

                              ('rsetxml', xmlrss.XMLRsetView),

                              ('rss', xmlrss.RSSView),

                              ('sameetypelist', baseviews.SameETypeListView),

                              ('secondary', baseviews.SecondaryView),

                              ('security', management.SecurityManagementView),

                              ('table', tableview.TableView),

                              ('text', baseviews.TextView),

                              ('treeview', treeview.TreeView),

                              ('xbel', xbel.XbelView),

                              ('xml', xmlrss.XMLView),

                              ])

    def test_possible_views_multiple_egroups(self):

        req = self.request()

        rset = req.execute('CWGroup X')

        self.assertListEqual(self.pviews(req, rset),

                             [('csvexport', csvexport.CSVRsetView),

                              ('ecsvexport', csvexport.CSVEntityView),

                              ('editable-table', tableview.EditableTableView),

                              ('filetree', treeview.FileTreeView),

                              ('list', baseviews.ListView),

                              ('oneline', baseviews.OneLineView),

                              ('owlabox', owl.OWLABOXView),

                              ('primary', cwuser.CWGroupPrimaryView),

                              ('rsetxml', xmlrss.XMLRsetView),

                              ('rss', xmlrss.RSSView),

                              ('sameetypelist', baseviews.SameETypeListView),

                              ('secondary', baseviews.SecondaryView),

                              ('security', management.SecurityManagementView),

                              ('table', tableview.TableView),

                              ('text', baseviews.TextView),

                              ('treeview', treeview.TreeView),

                              ('xbel', xbel.XbelView),

                              ('xml', xmlrss.XMLView),

                              ])

    def test_propertiesform_admin(self):

        assert self.vreg['views']['propertiesform']

        req1 = self.request()

        req2 = self.request()

        rset1 = req1.execute('CWUser X WHERE X login "admin"')

        rset2 = req2.execute('CWUser X WHERE X login "anon"')

        self.failUnless(self.vreg['views'].select('propertiesform', req1, rset=None))

        self.failUnless(self.vreg['views'].select('propertiesform', req1, rset=rset1))

        self.failUnless(self.vreg['views'].select('propertiesform', req2, rset=rset2))

    def test_propertiesform_anon(self):

        self.login('anon')

        req1 = self.request()

        req2 = self.request()

        rset1 = req1.execute('CWUser X WHERE X login "admin"')

        rset2 = req2.execute('CWUser X WHERE X login "anon"')

        self.assertRaises(NoSelectableObject, self.vreg['views'].select, 'propertiesform', req1, rset=None)

        self.assertRaises(NoSelectableObject, self.vreg['views'].select, 'propertiesform', req1, rset=rset1)

        self.assertRaises(NoSelectableObject, self.vreg['views'].select, 'propertiesform', req1, rset=rset2)

    def test_propertiesform_jdoe(self):

        self.create_user('jdoe')

        self.login('jdoe')

        req1 = self.request()

        req2 = self.request()

        rset1 = req1.execute('CWUser X WHERE X login "admin"')

        rset2 = req2.execute('CWUser X WHERE X login "jdoe"')

        self.failUnless(self.vreg['views'].select('propertiesform', req1, rset=None))

        self.assertRaises(NoSelectableObject, self.vreg['views'].select, 'propertiesform', req1, rset=rset1)

        self.failUnless(self.vreg['views'].select('propertiesform', req2, rset=rset2))

    def test_possible_views_multiple_different_types(self):

        req = self.request()

        rset = req.execute('Any X')

        self.assertListEqual(self.pviews(req, rset),

                             [('csvexport', csvexport.CSVRsetView),

                              ('ecsvexport', csvexport.CSVEntityView),

                              ('editable-table', tableview.EditableTableView),

                              ('filetree', treeview.FileTreeView),

                              ('list', baseviews.ListView),

                              ('oneline', baseviews.OneLineView),

                              ('owlabox', owl.OWLABOXView),

                              ('primary', primary.PrimaryView),

                              ('rsetxml', xmlrss.XMLRsetView),

                              ('rss', xmlrss.RSSView),

                              ('secondary', baseviews.SecondaryView),

                              ('security', management.SecurityManagementView),

                              ('table', tableview.TableView),

                              ('text', baseviews.TextView),

                              ('treeview', treeview.TreeView),

                              ('xbel', xbel.XbelView),

                              ('xml', xmlrss.XMLView),

                              ])

    def test_possible_views_any_rset(self):

        req = self.request()

        rset = req.execute('Any N, X WHERE X in_group Y, Y name N')

        self.assertListEqual(self.pviews(req, rset),

                             [('csvexport', csvexport.CSVRsetView),

                              ('editable-table', tableview.EditableTableView),

                              ('rsetxml', xmlrss.XMLRsetView),

                              ('table', tableview.TableView),

                              ])

    def test_possible_views_multiple_eusers(self):

        req = self.request()

        rset = req.execute('CWUser X')

        self.assertListEqual(self.pviews(req, rset),

                             [('csvexport', csvexport.CSVRsetView),

                              ('ecsvexport', csvexport.CSVEntityView),

                              ('editable-table', tableview.EditableTableView),

                              ('filetree', treeview.FileTreeView),

                              ('foaf', cwuser.FoafView),

                              ('list', baseviews.ListView),

                              ('oneline', baseviews.OneLineView),

                              ('owlabox', owl.OWLABOXView),

                              ('primary', primary.PrimaryView),

                              ('rsetxml', xmlrss.XMLRsetView),

                              ('rss', xmlrss.RSSView),

                              ('sameetypelist', baseviews.SameETypeListView),

                              ('secondary', baseviews.SecondaryView),

                              ('security', management.SecurityManagementView),

                              ('table', tableview.TableView),

                              ('text', baseviews.TextView),

                              ('treeview', treeview.TreeView),

                              ('vcard', vcard.VCardCWUserView),

                              ('xbel', xbel.XbelView),

                              ('xml', xmlrss.XMLView),

                              ])

    def test_possible_actions_none_rset(self):

        req = self.request()

        self.assertDictEqual(self.pactionsdict(req, None, skipcategories=()),

                             {'useractions': USERACTIONS,

                              'siteactions': SITEACTIONS,

                              'footer': FOOTERACTIONS,

                              })

    def test_possible_actions_no_entity(self):

        req = self.request()

        rset = req.execute('Any X WHERE X eid 999999')

        self.assertDictEqual(self.pactionsdict(req, rset, skipcategories=()),

                             {'useractions': USERACTIONS,

                              'siteactions': SITEACTIONS,

                              'footer': FOOTERACTIONS,

                              })

    def test_possible_actions_same_type_entities(self):

        req = self.request()

        rset = req.execute('CWGroup X')

        self.assertDictEqual(self.pactionsdict(req, rset, skipcategories=()),

                             {'useractions': USERACTIONS,

                              'siteactions': SITEACTIONS,

                              'footer': FOOTERACTIONS,

                              'mainactions': [actions.MultipleEditAction],

                              'moreactions': [actions.DeleteAction,

                                              actions.AddNewAction],

                              })

    def test_possible_actions_different_types_entities(self):

        req = self.request()

        rset = req.execute('Any X')

        self.assertDictEqual(self.pactionsdict(req, rset, skipcategories=()),

                             {'useractions': USERACTIONS,

                              'siteactions': SITEACTIONS,

                              'footer': FOOTERACTIONS,

                              'moreactions': [actions.DeleteAction],

                              })

    def test_possible_actions_final_entities(self):

        req = self.request()

        rset = req.execute('Any N, X WHERE X in_group Y, Y name N')

        self.assertDictEqual(self.pactionsdict(req, rset, skipcategories=()),

                             {'useractions': USERACTIONS,

                              'siteactions': SITEACTIONS,

                              'footer': FOOTERACTIONS,

                              })

    def test_possible_actions_eetype_cwuser_entity(self):

        req = self.request()

        rset = req.execute('CWEType X WHERE X name "CWUser"')

        self.assertDictEqual(self.pactionsdict(req, rset, skipcategories=()),

                             {'useractions': USERACTIONS,

                              'siteactions': SITEACTIONS,

                              'footer': FOOTERACTIONS,

                              'mainactions': [actions.ModifyAction,

                                              actions.ViewSameCWEType],

                              'moreactions': [actions.ManagePermissionsAction,

                                              actions.AddRelatedActions,

                                              actions.DeleteAction,

                                              actions.CopyAction,

                                              ],

                              })

    def test_select_creation_form(self):

        rset = None

        req = self.request()

        # creation form

        req.form['etype'] = 'CWGroup'

        self.assertIsInstance(self.vreg['views'].select('creation', req, rset=rset),

                              editforms.CreationFormView)

        del req.form['etype']

        # custom creation form

        class CWUserCreationForm(editforms.CreationFormView):

            __select__ = specified_etype_implements('CWUser')

        self.vreg._loadedmods[__name__] = {}

        self.vreg.register(CWUserCreationForm)

        req.form['etype'] = 'CWUser'

        self.assertIsInstance(self.vreg['views'].select('creation', req, rset=rset),

                              CWUserCreationForm)

    def test_select_view(self):

        # no entity

        rset = None

        req = self.request()

        self.assertIsInstance(self.vreg['views'].select('index', req, rset=rset),

                             startup.IndexView)

        self.failUnlessRaises(NoSelectableObject,

                             self.vreg['views'].select, 'primary', req, rset=rset)

        self.failUnlessRaises(NoSelectableObject,

                             self.vreg['views'].select, 'table', req, rset=rset)

        # no entity

        req = self.request()

        rset = req.execute('Any X WHERE X eid 999999')

        self.failUnlessRaises(NoSelectableObject,

                              self.vreg['views'].select, 'index', req, rset=rset)

        self.failUnlessRaises(NoSelectableObject,

                              self.vreg['views'].select, 'creation', req, rset=rset)

        self.failUnlessRaises(NoSelectableObject,

                              self.vreg['views'].select, 'primary', req, rset=rset)

        self.failUnlessRaises(NoSelectableObject,

                             self.vreg['views'].select, 'table', req, rset=rset)

        # one entity

        req = self.request()

        rset = req.execute('CWGroup X WHERE X name "managers"')

        self.assertIsInstance(self.vreg['views'].select('primary', req, rset=rset),

                             primary.PrimaryView)