def __create_resource_tree(self, res_obj, user): res_name = res_obj._meta.resource_name res_model = res_obj._meta.object_class schema = self._registry[res_name].build_schema() res_tree = {"allowed_http_methods": [], "fields": {}} pub_states = {} states = get_init_states(res_model, user) if states: for state in states: pub_states.update({state.codename: unicode(state.title)}) res_tree.update({"states": pub_states}) workflow = get_workflow(res_model) if workflow: workflow.get_initial_state() for fn, attrs in schema["fields"].items(): field_attrs = {"readonly": False, "nullable": False, "disabled": False} res_tree["fields"].update({fn: field_attrs}) res_tree["allowed_http_methods"] = schema["allowed_detail_http_methods"] return res_tree
def test_set_workflow_existing_model_relation(self): self.workflow.set_to_model(Article) self.new_workflow = Workflow.objects.create(title="New Workflow", description="Test desc.") self.new_workflow.set_to_model(Article) tools.assert_equals(get_workflow(Article), self.new_workflow)