Produits Entr'ouvert » w.c.s.

def test_workflow_to_dict(pub):

    workflow = Workflow(name='wf roles')

    st1 = workflow.add_status('Status1', 'st1')

    st2 = workflow.add_status('Status2', 'st2')

    st2.forced_endpoint = True

    workflow.roles['_other'] = 'Other Function'

    root = workflow.to_dict()

    assert set(root.keys()) == set(['statuses', 'name', 'roles'])

    assert root['name'] == 'wf roles'

    assert set(root['statuses'].keys()) == set(['st1', 'st2'])

    assert all(set(status.keys()) == set(['name', 'forced_endpoint']) for status in

               root['statuses'].values())

    assert root['statuses']['st1']['name'] == 'Status1'

    assert root['statuses']['st1']['forced_endpoint'] is False

    assert root['statuses']['st2']['name'] == 'Status2'

    assert root['statuses']['st2']['forced_endpoint'] is True

    assert json.loads(workflow.export_to_json(indent=4)) == root

    def export_to_json(self, include_id=False, indent=None):

        root = self.to_dict(include_id=include_id)

        return json.dumps(root, indent=indent)

    def to_dict(self, include_id=False):

        charset = get_publisher().site_charset

        root = {}

        root['name'] = unicode(self.name, charset)

        if include_id and self.id:

            root['id'] = str(self.id)

        if self.last_modification_time:

            root['last_modification_time'] = time.strftime('%Y-%m-%dT%H:%M:%S',

                                                           self.last_modification_time)

        roles = root['roles'] = {}

        for role, label in self.roles.iteritems():

            roles[role] = unicode(label, charset)

        statuses = root['statuses'] = {}

        for status in self.possible_status:

            statuses[status.id] = {

                'name': unicode(status.name, charset),

                'forced_endpoint': status.forced_endpoint,

            }

        return root