MISP
/
cti-python-stix2
mirror of https://github.com/MISP/cti-python-stix2
2
0
Fork 0
Code Issues Releases Wiki Activity

Removed some more hard-codings of v20 in the workbench test 

suite.
master
Michael Chisholm 2019-07-24 17:39:00 -04:00
parent b0eb518997
commit 5649559c6d
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
stix2/test/test_workbench.py
View File

@ -3,13 +3,14 @@ import os
import stix2 import stix2
from stix2.workbench import ( from stix2.workbench import (
_STIX_VID, AttackPattern, Campaign, CourseOfAction, ExternalReference, _STIX_VID, AttackPattern, Bundle, Campaign, CourseOfAction,
FileSystemSource, Filter, Identity, Indicator, IntrusionSet, Malware, ExternalReference, File, FileSystemSource, Filter, Identity, Indicator,
MarkingDefinition, ObservedData, Relationship, Report, StatementMarking, IntrusionSet, Malware, MarkingDefinition, NTFSExt, ObservedData,
ThreatActor, Tool, Vulnerability, add_data_source, all_versions, Relationship, Report, StatementMarking, ThreatActor, Tool, Vulnerability,
attack_patterns, campaigns, courses_of_action, create, get, identities, add_data_source, all_versions, attack_patterns, campaigns,
indicators, intrusion_sets, malware, observed_data, query, reports, save, courses_of_action, create, get, identities, indicators, intrusion_sets,
set_default_created, set_default_creator, set_default_external_refs, malware, observed_data, query, reports, save, set_default_created,
set_default_creator, set_default_external_refs,
set_default_object_marking_refs, threat_actors, tools, vulnerabilities, set_default_object_marking_refs, threat_actors, tools, vulnerabilities,
) )
@ -164,7 +165,7 @@ def test_workbench_get_all_vulnerabilities():
def test_workbench_add_to_bundle(): def test_workbench_add_to_bundle():
vuln = Vulnerability(**constants.VULNERABILITY_KWARGS) vuln = Vulnerability(**constants.VULNERABILITY_KWARGS)
bundle = stix2.v20.Bundle(vuln) bundle = Bundle(vuln)
assert bundle.objects[0].name == 'Heartbleed' assert bundle.objects[0].name == 'Heartbleed'
@ -305,12 +306,12 @@ def test_default_object_marking_refs():
def test_workbench_custom_property_object_in_observable_extension(): def test_workbench_custom_property_object_in_observable_extension():
ntfs = stix2.v20.NTFSExt( ntfs = NTFSExt(
allow_custom=True, allow_custom=True,
sid=1, sid=1,
x_foo='bar', x_foo='bar',
) )
artifact = stix2.v20.File( artifact = File(
name='test', name='test',
extensions={'ntfs-ext': ntfs}, extensions={'ntfs-ext': ntfs},
) )
@ -327,7 +328,7 @@ def test_workbench_custom_property_object_in_observable_extension():
def test_workbench_custom_property_dict_in_observable_extension(): def test_workbench_custom_property_dict_in_observable_extension():
artifact = stix2.v20.File( artifact = File(
allow_custom=True, allow_custom=True,
name='test', name='test',
extensions={ extensions={