Fix tests for test_add_related_indicator/observable TypeErrors

apsillers · apsillers · commit 12798621c7d7 · 2015-11-13T14:57:35.000-05:00
diff --git a/stix/indicator/indicator.py b/stix/indicator/indicator.py
@@ -177,6 +177,7 @@ class Indicator(stix.BaseCoreComponent):
     _ALL_VERSIONS = ("2.0", "2.0.1", "2.1", "2.1.1", "2.2")
     _ALLOWED_COMPOSITION_OPERATORS = ('AND', 'OR')
     _ID_PREFIX = "indicator"
+    _try_cast = False
 
     producer = fields.TypedField("Producer", InformationSource)
     observable = fields.TypedField("Observable", Observable, postset_hook=lambda inst,value: inst.set_observables([value]))
diff --git a/stix/test/incident_test.py b/stix/test/incident_test.py
@@ -516,7 +516,7 @@ def test_add_related_observable(self):
 
         # Test that this fails
         self.assertRaises(
-            ValueError,
+            TypeError,
             i.add_related_observable,
             "THIS SHOULD FAIL"
         )

Original file line number	Diff line number	Diff line change
`@@ -516,7 +516,7 @@ def test_add_related_observable(self):`
`516`	`516`
`517`	`517`	`# Test that this fails`
`518`	`518`	`self.assertRaises(`
`519`		`- ValueError,`
	`519`	`+ TypeError,`
`520`	`520`	`i.add_related_observable,`
`521`	`521`	`"THIS SHOULD FAIL"`
`522`	`522`	`)`