Fix Indicator.negate tests

clenk · clenk · commit b6ef77d413a5 · 2016-08-01T15:38:10.000-04:00
diff --git a/stix/indicator/indicator.py b/stix/indicator/indicator.py
@@ -722,6 +722,22 @@ def add_object(self, object_):
         observable = Observable(object_)
         self.add_observable(observable)
 
+    def _finalize_obj(self, entity_obj):
+        """Omits the `negate` field if it is not equal to True.
+        """
+        if self.negate:
+            entity_obj.negate = True
+        elif hasattr(entity_obj, 'negate'):
+           entity_obj.negate = None
+
+    def _finalize_dict(self, entity_dict):
+        """Omits the `negate` field if it is not equal to True.
+        """
+        if self.negate:
+            entity_dict['negate'] = True
+        elif 'negate' in entity_dict:
+           del entity_dict['negate']
+
 
 def check_operator(composite_indicator_exp, value):
     allowed = CompositeIndicatorExpression.OPERATORS
diff --git a/stix/test/indicator_test.py b/stix/test/indicator_test.py
@@ -37,6 +37,9 @@ def test_negate(self):
         d2 = self.klass.from_dict(d).to_dict()
         self.assertTrue('negate' not in d2)
 
+        o2 = self.klass.from_dict(d).to_obj()
+        self.assertTrue(o2.negate is None)
+        
 
     def test_indicator_types(self):
         d = {
