Remove certain human message assertions from test suites

master
Desai, Kartikey H 2019-07-09 13:34:19 -04:00
parent ae35d2ab01
commit b464a9cc0a
4 changed files with 31 additions and 62 deletions

View File

@ -125,15 +125,13 @@ def rel_fs_store():
def test_filesystem_source_nonexistent_folder(): def test_filesystem_source_nonexistent_folder():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.FileSystemSource('nonexistent-folder') stix2.FileSystemSource('nonexistent-folder')
assert "for STIX data does not exist" in str(excinfo)
def test_filesystem_sink_nonexistent_folder(): def test_filesystem_sink_nonexistent_folder():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.FileSystemSink('nonexistent-folder') stix2.FileSystemSink('nonexistent-folder')
assert "for STIX data does not exist" in str(excinfo)
def test_filesystem_source_bad_json_file(fs_source, bad_json_files): def test_filesystem_source_bad_json_file(fs_source, bad_json_files):
@ -441,9 +439,8 @@ def test_filesystem_attempt_stix_file_overwrite(fs_store):
) )
# Now attempt to overwrite the existing file # Now attempt to overwrite the existing file
with pytest.raises(DataSourceError) as excinfo: with pytest.raises(DataSourceError):
fs_store.add(camp8) fs_store.add(camp8)
assert "Attempted to overwrite file" in str(excinfo)
os.remove(filepath) os.remove(filepath)

View File

@ -257,7 +257,7 @@ def test_and_observable_expression():
def test_invalid_and_observable_expression(): def test_invalid_and_observable_expression():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.AndBooleanExpression([ stix2.AndBooleanExpression([
stix2.EqualityComparisonExpression( stix2.EqualityComparisonExpression(
"user-account:display_name", "user-account:display_name",
@ -268,7 +268,6 @@ def test_invalid_and_observable_expression():
stix2.StringConstant("admin"), stix2.StringConstant("admin"),
), ),
]) ])
assert "All operands to an 'AND' expression must have the same object type" in str(excinfo)
def test_hex(): def test_hex():
@ -352,30 +351,26 @@ def test_list2():
def test_invalid_constant_type(): def test_invalid_constant_type():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.EqualityComparisonExpression( stix2.EqualityComparisonExpression(
"artifact:payload_bin", "artifact:payload_bin",
{'foo': 'bar'}, {'foo': 'bar'},
) )
assert 'Unable to create a constant' in str(excinfo)
def test_invalid_integer_constant(): def test_invalid_integer_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.IntegerConstant('foo') stix2.IntegerConstant('foo')
assert 'must be an integer' in str(excinfo)
def test_invalid_timestamp_constant(): def test_invalid_timestamp_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.TimestampConstant('foo') stix2.TimestampConstant('foo')
assert 'Must be a datetime object or timestamp string' in str(excinfo)
def test_invalid_float_constant(): def test_invalid_float_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.FloatConstant('foo') stix2.FloatConstant('foo')
assert 'must be a float' in str(excinfo)
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -400,9 +395,8 @@ def test_boolean_constant(data, result):
def test_invalid_boolean_constant(): def test_invalid_boolean_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.BooleanConstant('foo') stix2.BooleanConstant('foo')
assert 'must be a boolean' in str(excinfo)
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -412,21 +406,18 @@ def test_invalid_boolean_constant():
], ],
) )
def test_invalid_hash_constant(hashtype, data): def test_invalid_hash_constant(hashtype, data):
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.HashConstant(data, hashtype) stix2.HashConstant(data, hashtype)
assert 'is not a valid {} hash'.format(hashtype) in str(excinfo)
def test_invalid_hex_constant(): def test_invalid_hex_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.HexConstant('mm') stix2.HexConstant('mm')
assert "must contain an even number of hexadecimal characters" in str(excinfo)
def test_invalid_binary_constant(): def test_invalid_binary_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.BinaryConstant('foo') stix2.BinaryConstant('foo')
assert 'must contain a base64' in str(excinfo)
def test_escape_quotes_and_backslashes(): def test_escape_quotes_and_backslashes():
@ -459,15 +450,13 @@ def test_repeat_qualifier():
def test_invalid_repeat_qualifier(): def test_invalid_repeat_qualifier():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.RepeatQualifier('foo') stix2.RepeatQualifier('foo')
assert 'is not a valid argument for a Repeat Qualifier' in str(excinfo)
def test_invalid_within_qualifier(): def test_invalid_within_qualifier():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.WithinQualifier('foo') stix2.WithinQualifier('foo')
assert 'is not a valid argument for a Within Qualifier' in str(excinfo)
def test_startstop_qualifier(): def test_startstop_qualifier():
@ -485,19 +474,17 @@ def test_startstop_qualifier():
def test_invalid_startstop_qualifier(): def test_invalid_startstop_qualifier():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.StartStopQualifier( stix2.StartStopQualifier(
'foo', 'foo',
stix2.TimestampConstant('2016-06-01T00:00:00Z'), stix2.TimestampConstant('2016-06-01T00:00:00Z'),
) )
assert 'is not a valid argument for a Start/Stop Qualifier' in str(excinfo)
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.StartStopQualifier( stix2.StartStopQualifier(
datetime.date(2016, 6, 1), datetime.date(2016, 6, 1),
'foo', 'foo',
) )
assert 'is not a valid argument for a Start/Stop Qualifier' in str(excinfo)
def test_make_constant_already_a_constant(): def test_make_constant_already_a_constant():

View File

@ -124,15 +124,13 @@ def rel_fs_store():
def test_filesystem_source_nonexistent_folder(): def test_filesystem_source_nonexistent_folder():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.FileSystemSource('nonexistent-folder') stix2.FileSystemSource('nonexistent-folder')
assert "for STIX data does not exist" in str(excinfo)
def test_filesystem_sink_nonexistent_folder(): def test_filesystem_sink_nonexistent_folder():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.FileSystemSink('nonexistent-folder') stix2.FileSystemSink('nonexistent-folder')
assert "for STIX data does not exist" in str(excinfo)
def test_filesystem_source_bad_json_file(fs_source, bad_json_files): def test_filesystem_source_bad_json_file(fs_source, bad_json_files):

View File

@ -257,7 +257,7 @@ def test_and_observable_expression():
def test_invalid_and_observable_expression(): def test_invalid_and_observable_expression():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.AndBooleanExpression([ stix2.AndBooleanExpression([
stix2.EqualityComparisonExpression( stix2.EqualityComparisonExpression(
"user-account:display_name", "user-account:display_name",
@ -268,7 +268,6 @@ def test_invalid_and_observable_expression():
stix2.StringConstant("admin"), stix2.StringConstant("admin"),
), ),
]) ])
assert "All operands to an 'AND' expression must have the same object type" in str(excinfo)
def test_hex(): def test_hex():
@ -352,30 +351,26 @@ def test_list2():
def test_invalid_constant_type(): def test_invalid_constant_type():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.EqualityComparisonExpression( stix2.EqualityComparisonExpression(
"artifact:payload_bin", "artifact:payload_bin",
{'foo': 'bar'}, {'foo': 'bar'},
) )
assert 'Unable to create a constant' in str(excinfo)
def test_invalid_integer_constant(): def test_invalid_integer_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.IntegerConstant('foo') stix2.IntegerConstant('foo')
assert 'must be an integer' in str(excinfo)
def test_invalid_timestamp_constant(): def test_invalid_timestamp_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.TimestampConstant('foo') stix2.TimestampConstant('foo')
assert 'Must be a datetime object or timestamp string' in str(excinfo)
def test_invalid_float_constant(): def test_invalid_float_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.FloatConstant('foo') stix2.FloatConstant('foo')
assert 'must be a float' in str(excinfo)
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -400,9 +395,8 @@ def test_boolean_constant(data, result):
def test_invalid_boolean_constant(): def test_invalid_boolean_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.BooleanConstant('foo') stix2.BooleanConstant('foo')
assert 'must be a boolean' in str(excinfo)
@pytest.mark.parametrize( @pytest.mark.parametrize(
@ -412,21 +406,18 @@ def test_invalid_boolean_constant():
], ],
) )
def test_invalid_hash_constant(hashtype, data): def test_invalid_hash_constant(hashtype, data):
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.HashConstant(data, hashtype) stix2.HashConstant(data, hashtype)
assert 'is not a valid {} hash'.format(hashtype) in str(excinfo)
def test_invalid_hex_constant(): def test_invalid_hex_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.HexConstant('mm') stix2.HexConstant('mm')
assert "must contain an even number of hexadecimal characters" in str(excinfo)
def test_invalid_binary_constant(): def test_invalid_binary_constant():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.BinaryConstant('foo') stix2.BinaryConstant('foo')
assert 'must contain a base64' in str(excinfo)
def test_escape_quotes_and_backslashes(): def test_escape_quotes_and_backslashes():
@ -459,15 +450,13 @@ def test_repeat_qualifier():
def test_invalid_repeat_qualifier(): def test_invalid_repeat_qualifier():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.RepeatQualifier('foo') stix2.RepeatQualifier('foo')
assert 'is not a valid argument for a Repeat Qualifier' in str(excinfo)
def test_invalid_within_qualifier(): def test_invalid_within_qualifier():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.WithinQualifier('foo') stix2.WithinQualifier('foo')
assert 'is not a valid argument for a Within Qualifier' in str(excinfo)
def test_startstop_qualifier(): def test_startstop_qualifier():
@ -485,19 +474,17 @@ def test_startstop_qualifier():
def test_invalid_startstop_qualifier(): def test_invalid_startstop_qualifier():
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.StartStopQualifier( stix2.StartStopQualifier(
'foo', 'foo',
stix2.TimestampConstant('2016-06-01T00:00:00Z'), stix2.TimestampConstant('2016-06-01T00:00:00Z'),
) )
assert 'is not a valid argument for a Start/Stop Qualifier' in str(excinfo)
with pytest.raises(ValueError) as excinfo: with pytest.raises(ValueError):
stix2.StartStopQualifier( stix2.StartStopQualifier(
datetime.date(2016, 6, 1), datetime.date(2016, 6, 1),
'foo', 'foo',
) )
assert 'is not a valid argument for a Start/Stop Qualifier' in str(excinfo)
def test_make_constant_already_a_constant(): def test_make_constant_already_a_constant():