From 20d04e363721d1aec2caa14638acba5d8579ae5b Mon Sep 17 00:00:00 2001 From: Edwin Lee Date: Tue, 23 Jan 2018 11:50:52 -0600 Subject: [PATCH] Add support for reference-class-name --- pyiddidf/idd/processor.py | 2 +- test/idd/test_processor.py | 19 +++++++++++++++++++ 2 files changed, 20 insertions(+), 1 deletion(-) diff --git a/pyiddidf/idd/processor.py b/pyiddidf/idd/processor.py index 9badae1..8e6bb9a 100644 --- a/pyiddidf/idd/processor.py +++ b/pyiddidf/idd/processor.py @@ -52,7 +52,7 @@ def __init__(self): self.field_flags = ["\\field", "\\note", "\\required-field", "\\begin-extensible", "\\unitsBasedOnField", "\\units", "\\ip-units", "\\scheduleunits", "\\minimum", "\\maximum", "\\default", "\\deprecated", "\\autosizable", "\\autocalculatable", "\\type", "\\retaincase", - "\\key", "\\object-list", "\\reference", "\\external-list"] + "\\key", "\\object-list", "\\reference-class-name", "\\reference", "\\external-list"] def process_file_given_file_path(self, file_path): """ diff --git a/test/idd/test_processor.py b/test/idd/test_processor.py index ab04ce3..86bc804 100644 --- a/test/idd/test_processor.py +++ b/test/idd/test_processor.py @@ -67,6 +67,25 @@ def test_repeated_object_meta_idd(self): version_obj = ret_value.get_object_by_type("version") self.assertEquals(1, len(version_obj.fields)) + def test_new_reference_class_name(self): + idd_object = """ +!IDD_Version 0.1.4 +!IDD_BUILD abcded0810 +\\group Simulation Parameters +NewObject, + A1; \\field Name + \\required-field + \\type alpha + \\reference-class-name validBranchEquipmentTypes + \\reference validBranchEquipmentNames + """ + processor = IDDProcessor() + ret_value = processor.process_file_via_stream(StringIO.StringIO(idd_object)) + self.assertEquals(1, len(ret_value.groups)) + self.assertEquals(1, len(ret_value.groups[0].objects)) + version_obj = ret_value.get_object_by_type("NewObject") + self.assertEquals(1, len(version_obj.fields)) + def test_single_line_obj_lookup(self): idd_object = """ !IDD_Version 1.2.0