def test_pure_policies(self): xproto = \ """ policy my_policy < exists x:a=b > """ proto = \ """ option my_policy = "policy:< exists x:a=b >"; """ target = XProtoTestHelpers.write_tmp_target( """ {{ policies }} """) args_xproto = FakeArgs() args_xproto.inputs = xproto args_xproto.target = target xproto_gen = XOSGenerator.generate(args_xproto) args_proto = FakeArgs() args_proto.inputs = proto args_proto.target = target args_proto.rev = True proto_gen = XOSGenerator.generate(args_proto) self.assertEqual(proto_gen, xproto_gen)
def test_pure_proto(self): xproto = \ """ message VRouterPort (XOSBase){ optional string name = 1 [help_text = "port friendly name", max_length = 20, null = True, db_index = False, blank = True]; required string openflow_id = 2 [help_text = "port identifier in ONOS", max_length = 21, null = False, db_index = False, blank = False]; required manytoone vrouter_device->VRouterDevice:ports = 3 [db_index = True, null = False, blank = False]; required manytoone vrouter_service->VRouterService:device_ports = 4 [db_index = True, null = False, blank = False]; } """ proto = \ """ message VRouterPort { option bases = "XOSBase"; optional string name = 1 [ null = "True", max_length = "20", blank = "True", help_text = "port friendly name", modifier = "optional", db_index = "False" ]; required string openflow_id = 2 [ null = "False", max_length = "21", blank = "False", help_text = "port identifier in ONOS", modifier = "required", db_index = "False" ]; required int32 vrouter_device = 3 [ null = "False", blank = "False", model = "VRouterDevice", modifier = "required", type = "link", port = "ports", db_index = "True", link = "manytoone"]; required int32 vrouter_service = 4 [ null = "False", blank = "False", model = "VRouterService", modifier = "required", type = "link", port = "device_ports", db_index = "True", link = "manytoone"]; } """ target = XProtoTestHelpers.write_tmp_target( """ from header import * {% for m in proto.messages %} {% if file_exists(xproto_base_name(m.name)|lower+'_header.py') -%}from {{xproto_base_name(m.name)|lower }}_header import *{% endif %} {% if file_exists(xproto_base_name(m.name)|lower+'_top.py') -%}{{ include_file(xproto_base_name(m.name)|lower+'_top.py') }} {% endif %} {%- for l in m.links %} {% if l.peer.name != m.name %} from core.models.{{ l.peer.name | lower }} import {{ l.peer.name }} {% endif %} {%- endfor %} {% for b in m.bases %} {% if b!='XOSBase' and 'Mixin' not in b%} from core.models.{{b.name | lower}} import {{ b.name }} {% endif %} {% endfor %} class {{ m.name }}{{ xproto_base_def(m, m.bases) }}: # Primitive Fields (Not Relations) {% for f in m.fields %} {%- if not f.link -%} {{ f.name }} = {{ xproto_django_type(f.type, f.options) }}( {{ xproto_django_options_str(f) }} ) {% endif %} {%- endfor %} # Relations {% for l in m.links %} {{ l.src_port }} = {{ xproto_django_link_type(l) }}( {%- if l.peer.name==m.name -%}'self'{%- else -%}{{ l.peer.name }} {%- endif -%}, {{ xproto_django_link_options_str(l, l.dst_port ) }} ) {%- endfor %} {% if file_exists(m.name|lower + '_model.py') -%}{{ include_file(m.name|lower + '_model.py') | indent(width=2)}}{%- endif %} pass {% if file_exists(xproto_base_name(m.name)|lower+'_bottom.py') -%}{{ include_file(xproto_base_name(m.name)|lower+'_bottom.py') }}{% endif %} {% endfor %} """) args_xproto = FakeArgs() args_xproto.inputs = xproto args_xproto.target = target xproto_gen = XOSGenerator.generate(args_xproto) count1 = len(xproto_gen.split('\n')) args_proto = FakeArgs() args_proto.inputs = proto args_proto.target = target args_proto.rev = True proto_gen = XOSGenerator.generate(args_proto) count2 = len(proto_gen.split('\n')) self.assertEqual(count1, count2)