change class TTPs to no longer be EntityList - fixes #357

Author: emmanvg

stix/core/stix_package.py

@@ -180,7 +180,7 @@ def add_ttp(self, ttp):
         """
         if self.ttps is None:
             self.ttps = TTPs()
-        self.ttps.append(ttp)
+        self.ttps.ttp.append(ttp)
 
     def add_report(self, report):
         """Adds a :class:`.Report` object to the :attr:`reports` collection.

stix/core/ttps.py

@@ -5,12 +5,10 @@
 from functools import partial
 
 # mixbox
-from mixbox import entities
 from mixbox import fields
 
 # stix
 import stix
-from stix import utils
 from stix.ttp import TTP
 from stix.common.kill_chains import KillChains
 from stix.bindings import stix_core as core_binding
@@ -19,24 +17,18 @@
 from stix.utils.deprecated import IdrefDeprecatedList
 
 
-class TTPs(stix.EntityList):
+class TTPs(stix.Entity):
     _binding = core_binding
     _binding_class = _binding.TTPsType
     _namespace = 'http://stix.mitre.org/stix-1'
 
-    ttps = fields.TypedField(
-        name="TTP",
-        type_=TTP,
-        multiple=True,
-        key_name="ttps",
-        listfunc=partial(IdrefDeprecatedList, type=TTP)
-    )
-
+    ttp = fields.TypedField("TTP", TTP, multiple=True, key_name="ttps", listfunc=partial(IdrefDeprecatedList, type=TTP))
     kill_chains = fields.TypedField("Kill_Chains", KillChains)
 
     def __init__(self, ttps=None):
-        super(TTPs, self).__init__(ttps)
+        super(TTPs, self).__init__()
+        self.ttp = ttps
         self.kill_chains = KillChains()
 
     def add_ttp(self, ttp):
-        self.append(ttp)
+        self.ttp.append(ttp)

stix/test/extensions/malware/maec_4_1_malware_test.py

@@ -134,14 +134,14 @@ def test_parse_malware(self):
         """Test parsing a normal MalwareInstance from XML
         """
         stix_pkg = STIXPackage.from_xml(self.XML)
-        mw = stix_pkg.ttps[0].behavior.malware_instances[0].to_dict()
+        mw = stix_pkg.ttps.ttp[0].behavior.malware_instances[0].to_dict()
         self.assertTrue('names' in mw)
 
     def test_parse_malware_maec(self):
         """Test parsing a MaecInstance from XML
         """
         stix_pkg = STIXPackage.from_xml(self.XML_MAEC)
-        mw = stix_pkg.ttps[0].behavior.malware_instances[0].to_dict()
+        mw = stix_pkg.ttps.ttp[0].behavior.malware_instances[0].to_dict()
         self.assertTrue('names' in mw)