Skip to content

Commit f37e711

Browse files
committed
GH-325: Add tests to verify desired behavior.
1 parent e0c487e commit f37e711

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

stix/test/indicator_test.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
from cybox.core import Observable, ObservableComposition
88
from cybox.objects.file_object import File
9+
import mixbox.typedlist
910
from mixbox.vendor.six import text_type
1011

1112
from stix.core import STIXPackage
@@ -493,6 +494,36 @@ def test_observables_property_standard(self):
493494
self.assertEqual([obs.to_dict()],
494495
[x.to_dict() for x in ind2.observables])
495496

497+
def test_set_indicator_observables_to_single_observable(self):
498+
# https://github.com/STIXProject/python-stix/issues/325
499+
i = Indicator()
500+
o1 = Observable()
501+
o2 = Observable()
502+
503+
i.observables = o1
504+
self.assertEqual(type([]), type(i.observables))
505+
self.assertEqual(1, len(i.observables))
506+
507+
def test_set_indicator_observables_to_list_of_two_observables(self):
508+
# https://github.com/STIXProject/python-stix/issues/325
509+
i = Indicator()
510+
o1 = Observable()
511+
o2 = Observable()
512+
513+
i.observables = [o1, o2]
514+
self.assertEqual(mixbox.typedlist.TypedList, type(i.observables))
515+
self.assertEqual(2, len(i.observables))
516+
517+
def test_set_indicator_observables_to_list_of_one_observable(self):
518+
# https://github.com/STIXProject/python-stix/issues/325
519+
i = Indicator()
520+
o1 = Observable()
521+
o2 = Observable()
522+
523+
i.observables = [o1]
524+
self.assertEqual(type([]), type(i.observables))
525+
self.assertEqual(1, len(i.observables))
526+
496527

497528
class RelatedCampaignReferencesTests(unittest.TestCase, EntityTestCase):
498529
klass = RelatedCampaignRefs

0 commit comments

Comments
 (0)