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)
