# copyright 2003-2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

def add_wf(shell, etype, name=None, default=False):

    return shell.add_workflow(name, etype, default=default,

                              ensure_workflowable=False)

        with self.admin_access.shell() as shell:

            wf = add_wf(shell, 'Company')

            foo = wf.add_state(u'foo', initial=True)

            bar = wf.add_state(u'bar')

            self.assertEqual(wf.state_by_name('bar').eid, bar.eid)

            self.assertEqual(wf.state_by_name('barrr'), None)

            baz = wf.add_transition(u'baz', (foo,), bar, ('managers',))

            self.assertEqual(wf.transition_by_name('baz').eid, baz.eid)

            self.assertEqual(len(baz.require_group), 1)

            self.assertEqual(baz.require_group[0].name, 'managers')

        with self.admin_access.shell() as shell:

            wf = add_wf(shell, 'Company')

            wf.add_state(u'foo', initial=True)

            shell.commit()

            wf.add_state(u'foo')

            with self.assertRaises(ValidationError) as cm:

                shell.commit()

            self.assertEqual({'name-subject': 'workflow already has a state of that name'},

                             cm.exception.errors)

            # no pb if not in the same workflow

            wf2 = add_wf(shell, 'Company')

            foo = wf2.add_state(u'foo', initial=True)

            shell.commit()

            # gnark gnark

            bar = wf.add_state(u'bar')

            shell.commit()

            bar.cw_set(name=u'foo')

            with self.assertRaises(ValidationError) as cm:

                shell.commit()

            self.assertEqual({'name-subject': 'workflow already has a state of that name'},

                             cm.exception.errors)

        with self.admin_access.shell() as shell:

            wf = add_wf(shell, 'Company')

            foo = wf.add_state(u'foo', initial=True)

            bar = wf.add_state(u'bar')

            wf.add_transition(u'baz', (foo,), bar, ('managers',))

            wf.add_transition(u'baz', (bar,), foo)

            with self.assertRaises(ValidationError) as cm:

                shell.commit()

            self.assertEqual(cm.exception.errors, {'name-subject': 'workflow already has a transition of that name'})

            # no pb if not in the same workflow

            wf2 = add_wf(shell, 'Company')

            foo = wf.add_state(u'foo', initial=True)

            bar = wf.add_state(u'bar')

            wf.add_transition(u'baz', (foo,), bar, ('managers',))

            shell.commit()

            # gnark gnark

            biz = wf.add_transition(u'biz', (bar,), foo)

            shell.commit()

            biz.cw_set(name=u'baz')

            with self.assertRaises(ValidationError) as cm:

                shell.commit()

            self.assertEqual(cm.exception.errors, {'name-subject': 'workflow already has a transition of that name'})

        with self.admin_access.client_cnx() as cnx:

            self.member_eid = self.create_user(cnx, 'member').eid

            cnx.commit()

        with self.admin_access.web_request() as req:

            e = self.create_user(req, 'toto')

            iworkflowable = e.cw_adapt_to('IWorkflowable')

            self.assertEqual(iworkflowable.state, 'activated')

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

            req.cnx.commit()

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

            req.cnx.commit()

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

            req.cnx.commit()

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

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

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

            self.assertEqual(iworkflowable.latest_trinfo().comment, 'deactivate 2')

        with self.admin_access.web_request() as req:

            user = req.execute('CWUser X').get_entity(0, 0)

            iworkflowable = user.cw_adapt_to('IWorkflowable')

            trs = list(iworkflowable.possible_transitions())

            self.assertEqual(len(trs), 1)

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

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

        with self.new_access('member').web_request() as req:

            # fetch the entity using the new session

            trs = list(req.user.cw_adapt_to('IWorkflowable').possible_transitions())

            self.assertEqual(len(trs), 0)

        with self.admin_access.client_cnx() as cnx:

            user = cnx.user

            iworkflowable = user.cw_adapt_to('IWorkflowable')

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

            trinfo = self._test_manager_deactivate(user)

            self.assertEqual(trinfo.transition, None)

        with self.admin_access.shell() as shell:

            wf = add_wf(shell, 'CWUser')

            s = wf.add_state(u'foo', initial=True)

            shell.commit()

        with self.admin_access.repo_cnx() as cnx:

            with cnx.security_enabled(write=False):

                with self.assertRaises(ValidationError) as cm:

                    cnx.execute('SET X in_state S WHERE X eid %(x)s, S eid %(s)s',

                                {'x': cnx.user.eid, 's': s.eid})

                self.assertEqual(cm.exception.errors, {'in_state-subject': "state doesn't belong to entity's workflow. "

                                          "You may want to set a custom workflow for this entity first."})

        with self.admin_access.client_cnx() as cnx:

            user = cnx.user

            iworkflowable = user.cw_adapt_to('IWorkflowable')

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

            user.cw_clear_all_caches()

            self.assertEqual(iworkflowable.state, 'deactivated')

            self._test_manager_deactivate(user)

            trinfo = self._test_manager_deactivate(user)

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

        with self.admin_access.web_request() as req:

            wf = req.user.cw_adapt_to('IWorkflowable').current_workflow

            asleep = wf.add_state('asleep')

            wf.add_transition('rest', (wf.state_by_name('activated'),

                                       wf.state_by_name('deactivated')),

                              asleep)

            wf.add_transition('wake up', asleep)

            user = self.create_user(req, 'stduser')

            iworkflowable = user.cw_adapt_to('IWorkflowable')

            iworkflowable.fire_transition('rest')

            req.cnx.commit()

            iworkflowable.fire_transition('wake up')

            req.cnx.commit()

            self.assertEqual(iworkflowable.state, 'activated')

            iworkflowable.fire_transition('deactivate')

            req.cnx.commit()

            iworkflowable.fire_transition('rest')

            req.cnx.commit()

            iworkflowable.fire_transition('wake up')

            req.cnx.commit()

            user.cw_clear_all_caches()

            self.assertEqual(iworkflowable.state, 'deactivated')

        with self.admin_access.repo_cnx() as cnx:

            self.create_user(cnx, 'tutu')

        with self.new_access('tutu').web_request() as req:

            iworkflowable = req.entity_from_eid(self.member_eid).cw_adapt_to('IWorkflowable')

            with self.assertRaises(ValidationError) as cm:

                iworkflowable.fire_transition('deactivate')

            self.assertEqual(cm.exception.errors, {'by_transition-subject': "transition may not be fired"})

        with self.new_access('member').web_request() as req:

            iworkflowable = req.entity_from_eid(self.member_eid).cw_adapt_to('IWorkflowable')

            req.cnx.commit()

            with self.assertRaises(ValidationError) as cm:

                iworkflowable.fire_transition('activate')

            self.assertEqual(cm.exception.errors, {'by_transition-subject': "transition may not be fired"})

        with self.admin_access.repo_cnx() as cnx:

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

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

                        'WHERE T name "deactivate"')

            cnx.commit()

        with self.admin_access.repo_cnx() as cnx:

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

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

                        'WHERE T name "deactivate"')

            cnx.commit()

        with self.admin_access.shell() as shell:

            swf = add_wf(shell, 'CWGroup', name='subworkflow')

            swfstate1 = swf.add_state(u'swfstate1', initial=True)

            swfstate2 = swf.add_state(u'swfstate2')

            swfstate3 = swf.add_state(u'swfstate3')

            tr1 = swf.add_transition(u'tr1', (swfstate1,), swfstate2)

            tr2 = swf.add_transition(u'tr2', (swfstate1,), swfstate3)

            # main workflow

            mwf = add_wf(shell, 'CWGroup', name='main workflow', default=True)

            state1 = mwf.add_state(u'state1', initial=True)

            state2 = mwf.add_state(u'state2')

            state3 = mwf.add_state(u'state3')

            swftr1 = mwf.add_wftransition(u'swftr1', swf, state1,

                                          [(swfstate2, state2), (swfstate3, state3)])

            swf.cw_clear_all_caches()

            self.assertEqual(swftr1.destination(None).eid, swfstate1.eid)

        with self.admin_access.web_request() as req:

            group = req.create_entity('CWGroup', name=u'grp1')

            req.cnx.commit()

            iworkflowable = group.cw_adapt_to('IWorkflowable')

            self.assertEqual(iworkflowable.current_state.eid, state1.eid)

            self.assertEqual(iworkflowable.current_workflow.eid, mwf.eid)

            self.assertEqual(iworkflowable.main_workflow.eid, mwf.eid)

            self.assertEqual(iworkflowable.subworkflow_input_transition(), None)

            iworkflowable.fire_transition('swftr1', u'go')

            req.cnx.commit()

            group.cw_clear_all_caches()

            self.assertEqual(iworkflowable.current_state.eid, swfstate1.eid)

            self.assertEqual(iworkflowable.current_workflow.eid, swf.eid)

            self.assertEqual(iworkflowable.main_workflow.eid, mwf.eid)

            self.assertEqual(iworkflowable.subworkflow_input_transition().eid, swftr1.eid)

            iworkflowable.fire_transition('tr1', u'go')

            req.cnx.commit()

            group.cw_clear_all_caches()

            self.assertEqual(iworkflowable.current_state.eid, state2.eid)

            self.assertEqual(iworkflowable.current_workflow.eid, mwf.eid)

            self.assertEqual(iworkflowable.main_workflow.eid, mwf.eid)

            self.assertEqual(iworkflowable.subworkflow_input_transition(), None)

            # force back to swfstate1 is impossible since we can't any more find

            # subworkflow input transition

            with self.assertRaises(ValidationError) as cm:

                iworkflowable.change_state(swfstate1, u'gadget')

            self.assertEqual(cm.exception.errors, {'to_state-subject': "state doesn't belong to entity's workflow"})

            req.cnx.rollback()

            # force back to state1

            iworkflowable.change_state('state1', u'gadget')

            iworkflowable.fire_transition('swftr1', u'au')

            group.cw_clear_all_caches()

            iworkflowable.fire_transition('tr2', u'chapeau')

            req.cnx.commit()

            group.cw_clear_all_caches()

            self.assertEqual(iworkflowable.current_state.eid, state3.eid)

            self.assertEqual(iworkflowable.current_workflow.eid, mwf.eid)

            self.assertEqual(iworkflowable.main_workflow.eid, mwf.eid)

            self.assertListEqual(parse_hist(iworkflowable.workflow_history),

                                  [('state1', 'swfstate1', 'swftr1', 'go'),

                                   ('swfstate1', 'swfstate2', 'tr1', 'go'),

                                   ('swfstate2', 'state2', 'swftr1', 'exiting from subworkflow subworkflow'),

                                   ('state2', 'state1', None, 'gadget'),

                                   ('state1', 'swfstate1', 'swftr1', 'au'),

                                   ('swfstate1', 'swfstate3', 'tr2', 'chapeau'),

                                   ('swfstate3', 'state3', 'swftr1', 'exiting from subworkflow subworkflow'),

                                   ])

        with self.admin_access.shell() as shell:

            # sub-workflow

            swf = add_wf(shell, 'CWGroup', name='subworkflow')

            swfstate1 = swf.add_state(u'swfstate1', initial=True)

            swfstate2 = swf.add_state(u'swfstate2')

            tr1 = swf.add_transition(u'tr1', (swfstate1,), swfstate2)

            # main workflow

            mwf = add_wf(shell, 'CWGroup', name='main workflow', default=True)

            state1 = mwf.add_state(u'state1', initial=True)

            state2 = mwf.add_state(u'state2')

            state3 = mwf.add_state(u'state3')

            mwf.add_wftransition(u'swftr1', swf, state1,

                                 [(swfstate2, state2), (swfstate2, state3)])

            with self.assertRaises(ValidationError) as cm:

                shell.commit()

            self.assertEqual(cm.exception.errors, {'subworkflow_exit-subject': u"can't have multiple exits on the same state"})