Fix 'test_memory_store_object_with_custom_property_in_bundle'

Since Bundles objects are now added instead of the Bundle itself, it now works as intended.
stix2.1
Emmanuelle Vargas-Gonzalez 2018-07-10 16:19:40 -04:00
parent 834ef2c847
commit 6bd797e258
2 changed files with 2 additions and 4 deletions

View File

@ -248,8 +248,7 @@ def test_memory_store_object_with_custom_property_in_bundle(mem_store):
bundle = Bundle(camp, allow_custom=True)
mem_store.add(bundle)
bundle_r = mem_store.get(bundle.id)
camp_r = bundle_r['objects'][0]
camp_r = mem_store.get(camp.id)
assert camp_r.id == camp.id
assert camp_r.x_empire == camp.x_empire

View File

@ -256,8 +256,7 @@ def test_memory_store_object_with_custom_property_in_bundle(mem_store):
bundle = Bundle(camp, allow_custom=True)
mem_store.add(bundle)
bundle_r = mem_store.get(bundle.id)
camp_r = bundle_r['objects'][0]
camp_r = mem_store.get(camp.id)
assert camp_r.id == camp.id
assert camp_r.x_empire == camp.x_empire