Merge pull request #632 from chisholm/fix-test-environment-no-datastore

Fix test_environment_no_datastore() unit tests

Author: rpiazza

stix2/test/v20/test_environment.py

@@ -221,9 +221,8 @@ def test_environment_datastore_and_sink():
 def test_environment_no_datastore():
     env = stix2.Environment(factory=stix2.ObjectFactory())
 
-    with pytest.raises(AttributeError) as excinfo:
+    with pytest.raises(AttributeError):
         env.add(stix2.v20.Indicator(**INDICATOR_KWARGS))
-    assert 'Environment has no data sink to put objects in' in str(excinfo.value)
 
     with pytest.raises(AttributeError) as excinfo:
         env.get(INDICATOR_ID)

stix2/test/v21/test_environment.py

@@ -230,9 +230,8 @@ def test_environment_datastore_and_sink():
 def test_environment_no_datastore():
     env = stix2.Environment(factory=stix2.ObjectFactory())
 
-    with pytest.raises(AttributeError) as excinfo:
+    with pytest.raises(AttributeError):
         env.add(stix2.v21.Indicator(**INDICATOR_KWARGS))
-    assert 'Environment has no data sink to put objects in' in str(excinfo.value)
 
     with pytest.raises(AttributeError) as excinfo:
         env.get(INDICATOR_ID)