unittest_link.py

# -*- coding: utf-8 -*-
from cubicweb.devtools.testlib import CubicWebTC
from cubicweb.web.views import uicfg, actions, embedding

from cubes.link import views

afs = uicfg.autoform_section

class LinkTC(CubicWebTC):

    def setup_database(self):
        self.request().create_entity('Link', title=u"vous êtes perdu ?", url=u"http://www.perdu.com")

    def test_possible_actions(self):
        req = self.request()
        rset = self.execute('Any X WHERE X is Link')
        allactions = self.pactionsdict(req, rset)
        self.assertEqual(allactions['mainactions'],
                         [actions.ModifyAction,
                          views.LinkFollowAction])
        self.assertEqual(allactions['moreactions'],
                         [actions.ManagePermissionsAction,
                          actions.AddRelatedActions,
                          actions.DeleteAction,
                          actions.CopyAction,
                          embedding.EmbedAction])

    def test_relations_by_category(self):
        def rbc(iterable):
            return [(rschema.type, x) for rschema, tschemas, x in iterable]
        e = self.vreg["etypes"].etype_class('Link')(self.request())
        # we should only see embed when we are in the managers group
        self.assertEqual(rbc(afs.relations_by_section(e, 'main', 'attributes', 'update')),
                          [('title', 'subject'), ('url', 'subject'),
                           ('embed', 'subject'),
                           ('description', 'subject')])
        self.create_user(self.session, 'toto')
        self.login('toto')
        # create a new instance with the new connection
        e = self.vreg["etypes"].etype_class('Link')(self.request())
        self.assertEqual(rbc(afs.relations_by_section(e, 'main', 'attributes', 'update')),
                          [('title', 'subject'), ('url', 'subject'),
                           ('description', 'subject')])

if __name__ == '__main__':
    from logilab.common.testlib import unittest_main
    unittest_main()