diff --git a/stix2/datastore/relational_db/input_creation.py b/stix2/datastore/relational_db/input_creation.py index 23cd56e6..1b859751 100644 --- a/stix2/datastore/relational_db/input_creation.py +++ b/stix2/datastore/relational_db/input_creation.py @@ -347,7 +347,7 @@ def generate_insert_for_core(data_sink, stix_object, core_properties, schema_nam insert_statements.append(core_insert_statement) if "object_marking_refs" in stix_object: - if schema_name == "sdo": + if schema_name != "sco": object_markings_ref_table = data_sink.tables_dictionary["common.object_marking_refs_sdo"] else: object_markings_ref_table = data_sink.tables_dictionary["common.object_marking_refs_sco"] diff --git a/stix2/datastore/relational_db/relational_db_testing.py b/stix2/datastore/relational_db/relational_db_testing.py index 540b5eae..b051dc86 100644 --- a/stix2/datastore/relational_db/relational_db_testing.py +++ b/stix2/datastore/relational_db/relational_db_testing.py @@ -103,6 +103,10 @@ def extension_definition_insert(): schema="a schema", version="1.2.3", extension_types=["property-extension", "new-sdo", "new-sro"], + object_marking_refs=[ + "marking-definition--caa0d913-5db8-4424-aae0-43e770287d30", + "marking-definition--122a27a0-b96f-46bc-8fcd-f7a159757e77" + ], granular_markings=[ { "lang": "en_US", @@ -168,7 +172,7 @@ def kill_chain_test(): @stix2.CustomObject('x-custom-type', properties=[ ("phases", stix2.properties.ListProperty(stix2.KillChainPhase)), - ("test", stix2.properties.IntegerProperty()) + ("something_else", stix2.properties.IntegerProperty()) ] ) class CustomClass: @@ -183,7 +187,7 @@ def custom_obj(): "phase_name": "the phase name" } ], - test=5 + something_else=5 ) return obj