fix problem with lookup_extension for TestMechanism. closes #361

Author: emmanvg

stix/bindings/indicator.py

@@ -225,12 +225,12 @@ class TestMechanismType(GeneratedsSuper):
     reference to the ID of a Test Mechanism specified elsewhere."""
     subclass = None
     superclass = None
-    def __init__(self, idref=None, id=None, Efficacy=None, Producer=None, xsi_type=None):
+    def __init__(self, idref=None, id=None, Efficacy=None, Producer=None):
         self.idref = _cast(None, idref)
         self.id = _cast(None, id)
         self.Efficacy = Efficacy
         self.Producer = Producer
-        self.xsi_type = xsi_type
+        # self.xsi_type = xsi_type
     def factory(*args_, **kwargs_):
         if TestMechanismType.subclass:
             return TestMechanismType.subclass(*args_, **kwargs_)
@@ -278,9 +278,9 @@ def exportAttributes(self, lwrite, level, already_processed, namespace_='indicat
         if self.id is not None and 'id' not in already_processed:
             already_processed.add('id')
             lwrite(' id=%s' % (quote_attrib(self.id), ))
-        if self.xsi_type is not None and 'xsi:type' not in already_processed:
-            already_processed.add('xsi:type')
-            lwrite('  xsi:type="%s"' % self.xsi_type)
+        # if self.xsi_type is not None and 'xsi:type' not in already_processed:
+        #     already_processed.add('xsi:type')
+        #     lwrite('  xsi:type="%s"' % self.xsi_type)
 
     def exportChildren(self, lwrite, level, nsmap, namespace_=XML_NS, name_='TestMechanismType', fromsubclass_=False, pretty_print=True):
         if pretty_print:

stix/indicator/test_mechanism.py

@@ -15,8 +15,8 @@
 class _BaseTestMechanism(stix.Entity):
     _namespace = "http://stix.mitre.org/Indicator-2"
     _binding = indicator_binding
-    _binding_class = indicator_binding.TestMechanismType()
-    
+    _binding_class = indicator_binding.TestMechanismType
+
     id_ = fields.IdField("id")
     idref = fields.IdField("idref")
     efficacy = fields.TypedField("Efficacy", Statement)
@@ -29,11 +29,10 @@ def __init__(self, id_=None, idref=None):
         self.idref = idref
         self.efficacy = None
         self.producer = None
-    
+
     def to_obj(self, ns_info=None):
         obj = super(_BaseTestMechanism, self).to_obj(ns_info=ns_info)
         obj.xsi_type = self._XSI_TYPE
-
         return obj
 
     def to_dict(self):