from cubicweb.devtools.apptest import EnvBasedTC

from cubicweb import ValidationError

class WorkflowTC(EnvBasedTC):

    def setup_database(self):

        rschema = self.schema['in_state']

        for x, y in rschema.iter_rdefs():

            self.assertEquals(rschema.rproperty(x, y, 'cardinality'), '1*')

        self.member = self.create_user('member')

    def test_workflow_base(self):

        e = self.create_user('toto')

        self.assertEquals(e.state, 'activated')

        e.change_state('deactivated', u'deactivate 1')

        self.commit()

        e.change_state('activated', u'activate 1')

        self.commit()

        e.change_state('deactivated', u'deactivate 2')

        self.commit()

        e.clear_related_cache('wf_info_for', 'object')

        self.assertEquals([tr.comment for tr in e.reverse_wf_info_for],

                          ['deactivate 1', 'activate 1', 'deactivate 2'])

        self.assertEquals(e.latest_trinfo().comment, 'deactivate 2')

    def test_possible_transitions(self):

        user = self.entity('CWUser X')

        trs = list(user.possible_transitions())

        self.assertEquals(len(trs), 1)

        self.assertEquals(trs[0].name, u'deactivate')

        self.assertEquals(trs[0].destination().name, u'deactivated')

        # test a std user get no possible transition

        cnx = self.login('member')

        # fetch the entity using the new session

        trs = list(cnx.user().possible_transitions())

        self.assertEquals(len(trs), 0)

    def _test_manager_deactivate(self, user):

        user.clear_related_cache('in_state', 'subject')

        self.assertEquals(len(user.in_state), 1)

        self.assertEquals(user.state, 'deactivated')

        trinfo = user.latest_trinfo()

        self.assertEquals(trinfo.previous_state.name, 'activated')

        self.assertEquals(trinfo.new_state.name, 'deactivated')

        self.assertEquals(trinfo.comment, 'deactivate user')

        self.assertEquals(trinfo.comment_format, 'text/plain')

        return trinfo

    def test_change_state(self):

        user = self.user()

        user.change_state('deactivated', comment=u'deactivate user')

        trinfo = self._test_manager_deactivate(user)

        self.assertEquals(trinfo.transition, None)

    def test_fire_transition(self):

        user = self.user()

        user.fire_transition('deactivate', comment=u'deactivate user')

        self.assertEquals(user.state, 'deactivated')

        self._test_manager_deactivate(user)

        trinfo = self._test_manager_deactivate(user)

        self.assertEquals(trinfo.transition.name, 'deactivate')

    # XXX test managers can change state without matching transition

    def _test_stduser_deactivate(self):

        ueid = self.member.eid

        self.create_user('tutu')

        cnx = self.login('tutu')

        req = self.request()

        member = req.entity_from_eid(self.member.eid)

        ex = self.assertRaises(ValidationError,

                               member.fire_transition, 'deactivate')

        self.assertEquals(ex.errors, {'by_transition': "transition may not be fired"})

        cnx.close()

        cnx = self.login('member')

        req = self.request()

        member = req.entity_from_eid(self.member.eid)

        member.fire_transition('deactivate')

        cnx.commit()

        ex = self.assertRaises(ValidationError,

                               member.fire_transition, 'activate')

        self.assertEquals(ex.errors, {'by_transition': "transition may not be fired"})

    def test_fire_transition_owned_by(self):

        self.execute('INSERT RQLExpression X: X exprtype "ERQLExpression", '

                     'X expression "X owned_by U", T condition X '

                     'WHERE T name "deactivate"')

        self._test_stduser_deactivate()

    def test_fire_transition_has_update_perm(self):

        self.execute('INSERT RQLExpression X: X exprtype "ERQLExpression", '

                     'X expression "U has_update_permission X", T condition X '

                     'WHERE T name "deactivate"')

        self._test_stduser_deactivate()

    def _init_wf_with_shared_state_or_tr(self):

        req = self.request()

        etypes = dict(self.execute('Any N, ET WHERE ET is CWEType, ET name N'

                                   ', ET name IN ("CWGroup", "Bookmark")'))

        self.grpwf = req.create_entity('Workflow', ('workflow_of', 'ET'),

                                       ET=etypes['CWGroup'], name=u'group workflow')

        self.bmkwf = req.execute('Any X WHERE X is Workflow, X workflow_of ET, ET name "Bookmark"').get_entity(0, 0)

        self.state1 = self.grpwf.add_state(u'state1', initial=True)

        self.execute('SET S state_of WF WHERE S eid %(s)s, WF eid %(wf)s',

                     {'s': self.state1.eid, 'wf': self.bmkwf.eid})

        self.execute('SET WF initial_state S WHERE S eid %(s)s, WF eid %(wf)s',

                     {'s': self.state1.eid, 'wf': self.bmkwf.eid})

        self.state2 = self.grpwf.add_state(u'state2')

        self.group = self.add_entity('CWGroup', name=u't1')

        self.bookmark = self.add_entity('Bookmark', title=u'111', path=u'/view')

        # commit to link to the initial state

        self.commit()

    def test_transitions_selection(self):

        """

        ------------------------  tr1    -----------------

        | state1 (CWGroup, Bookmark) | ------> | state2 (CWGroup) |

        ------------------------         -----------------

                  |  tr2    ------------------

                  `------>  | state3 (Bookmark) |

                            ------------------

        """

        self._init_wf_with_shared_state_or_tr()

        state3 = self.bmkwf.add_state(u'state3')

        tr1 = self.grpwf.add_transition(u'tr1', (self.state1,), self.state2)

        tr2 = self.bmkwf.add_transition(u'tr2', (self.state1,), state3)

        transitions = list(self.group.possible_transitions())

        self.assertEquals(len(transitions), 1)

        self.assertEquals(transitions[0].name, 'tr1')

        transitions = list(self.bookmark.possible_transitions())

        self.assertEquals(len(transitions), 1)

        self.assertEquals(transitions[0].name, 'tr2')

    def test_transitions_selection2(self):

        """

        ------------------------  tr1 (Bookmark)   -----------------------

        | state1 (CWGroup, Bookmark) | -------------> | state2 (CWGroup,Bookmark) |

        ------------------------                -----------------------

                  |  tr2 (CWGroup)                     |

                  `---------------------------------/

        """

        self._init_wf_with_shared_state_or_tr()

        self.execute('SET S state_of WF WHERE S eid %(s)s, WF eid %(wf)s',

                     {'s': self.state2.eid, 'wf': self.bmkwf.eid})

        tr1 = self.bmkwf.add_transition(u'tr1', (self.state1,), self.state2)

        tr2 = self.grpwf.add_transition(u'tr2', (self.state1,), self.state2)

        transitions = list(self.group.possible_transitions())

        self.assertEquals(len(transitions), 1)

        self.assertEquals(transitions[0].name, 'tr2')

        transitions = list(self.bookmark.possible_transitions())

        self.assertEquals(len(transitions), 1)

        self.assertEquals(transitions[0].name, 'tr1')

from cubicweb.devtools.apptest import RepositoryBasedTC

class WorkflowHooksTC(RepositoryBasedTC):

    def setUp(self):

        RepositoryBasedTC.setUp(self)

        self.wf = self.session.user.current_workflow

        self.s_activated = self.wf.state_by_name('activated').eid

        self.s_deactivated = self.wf.state_by_name('deactivated').eid

        self.s_dummy = self.wf.add_state(u'dummy').eid

        self.wf.add_transition(u'dummy', (self.s_deactivated,), self.s_dummy)

        ueid = self.create_user('stduser', commit=False)

        # test initial state is set

        rset = self.execute('Any N WHERE S name N, X in_state S, X eid %(x)s',

                            {'x' : ueid})

        self.failIf(rset, rset.rows)

        self.commit()

        initialstate = self.execute('Any N WHERE S name N, X in_state S, X eid %(x)s',

                                    {'x' : ueid})[0][0]

        self.assertEquals(initialstate, u'activated')

        # give access to users group on the user's wf transitions

        # so we can test wf enforcing on euser (managers don't have anymore this

        # enforcement

        self.execute('SET X require_group G '

                     'WHERE G name "users", X transition_of WF, WF eid %(wf)s',

                     {'wf': self.wf.eid})

        self.commit()

    def tearDown(self):

        self.execute('DELETE X require_group G '

                     'WHERE G name "users", X transition_of WF, WF eid %(wf)s',

                     {'wf': self.wf.eid})

        self.commit()

        RepositoryBasedTC.tearDown(self)

    # XXX currently, we've to rely on hooks to set initial state, or to use unsafe_execute

    # def test_initial_state(self):

    #     cnx = self.login('stduser')

    #     cu = cnx.cursor()

    #     self.assertRaises(ValidationError, cu.execute,

    #                       'INSERT CWUser X: X login "badaboum", X upassword %(pwd)s, '

    #                       'X in_state S WHERE S name "deactivated"', {'pwd': 'oops'})

    #     cnx.close()

    #     # though managers can do whatever he want

    #     self.execute('INSERT CWUser X: X login "badaboum", X upassword %(pwd)s, '

    #                  'X in_state S, X in_group G WHERE S name "deactivated", G name "users"', {'pwd': 'oops'})

    #     self.commit()

    # test that the workflow is correctly enforced

    def test_transition_checking1(self):

        cnx = self.login('stduser')

        user = cnx.user(self.current_session())

        ex = self.assertRaises(ValidationError,

                               user.fire_transition, 'activate')

        self.assertEquals(ex.errors, {'by_transition': u"transition isn't allowed"})

        cnx.close()

    def test_transition_checking2(self):

        cnx = self.login('stduser')

        user = cnx.user(self.current_session())

        assert user.state == 'activated'

        ex = self.assertRaises(ValidationError,

                               user.fire_transition, 'dummy')

        self.assertEquals(ex.errors, {'by_transition': u"transition isn't allowed"})

        cnx.close()

    def test_transition_checking3(self):

        cnx = self.login('stduser')

        session = self.current_session()

        user = cnx.user(session)

        user.fire_transition('deactivate')

        cnx.commit()

        session.set_pool()

        ex = self.assertRaises(ValidationError,

                               user.fire_transition, 'deactivate')

        self.assertEquals(ex.errors, {'by_transition': u"transition isn't allowed"})

        # get back now

        user.fire_transition('activate')

        cnx.commit()

        cnx.close()

if __name__ == '__main__':

    from logilab.common.testlib import unittest_main

    unittest_main()