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

Style/lint fixes

stix2.1
Greg Back 2017-05-31 08:58:14 -05:00
parent 6c3a689f02
commit 53ddf32e8c
1 changed files with 69 additions and 82 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

151
stix2/test/test_data_sources.py
View File

@ -1,63 +1,58 @@
import pytest
import requests
from stix2.sources import taxii
#Data Source (common API)
#TAXII
def test_ds_taxii():
ds = taxii.TAXIIDataSource()
assert ds.name == 'TAXII'
def test_ds_taxii_name():
ds = taxii.TAXIIDataSource(name='My Data Source Name')
assert ds.name == "My Data Source Name"
def test_ds_params():
url = "http://taxii_url.com:5000"
creds = {"username":"Wade", "password":"Wilson"}
ds = taxii.TAXIIDataSource(api_root=url,
auth= creds)
creds = {"username": "Wade", "password": "Wilson"}
ds = taxii.TAXIIDataSource(api_root=url, auth=creds)
assert ds.taxii_info['api_root']['url'] == url
assert ds.taxii_info['auth'] == creds
def test_parse_taxii_filters():
query = [
{
"field":"added_after",
"op":"=",
"value":"2016-02-01T00:00:01.000Z"
"field": "added_after",
"op": "=",
"value": "2016-02-01T00:00:01.000Z"
},
{
"field":"id",
"op":"=",
"value":"taxii stix object ID"
"field": "id",
"op": "=",
"value": "taxii stix object ID"
},
{
"field":"type",
"op":"=",
"value":"taxii stix object ID"
"field": "type",
"op": "=",
"value": "taxii stix object ID"
},
{
"field":"version",
"op":"=",
"value":"first"
"field": "version",
"op": "=",
"value": "first"
},
{
"field":"created_by_ref",
"op":"=",
"value":"Bane"
"field": "created_by_ref",
"op": "=",
"value": "Bane"
}
]
expected_params = {
"added_after":"2016-02-01T00:00:01.000Z",
"match[id]":"taxii stix object ID",
"match[type]":"taxii stix object ID",
"match[version]":"first"
"added_after": "2016-02-01T00:00:01.000Z",
"match[id]": "taxii stix object ID",
"match[type]": "taxii stix object ID",
"match[version]": "first"
}
ds = taxii.TAXIIDataSource()
@ -73,45 +68,45 @@ def test_add_get_remove_filter():
obj_1 = dummy()
#First 3 filters are valid, remaining fields are erroneous in some way
# First 3 filters are valid, remaining fields are erroneous in some way
filters = [
{
"field": "type",
"op": '=',
"value":"malware"
"value": "malware"
},
{
"field":"id",
"op":"!=",
"value":"stix object id"
"field": "id",
"op": "!=",
"value": "stix object id"
},
{
"field":"labels",
"op":"in",
"value":["heartbleed","malicious-activity"]
"field": "labels",
"op": "in",
"value": ["heartbleed", "malicious-activity"]
},
{
"field":"revoked",
"value":"filter missing \'op\' field"
"field": "revoked",
"value": "filter missing \'op\' field"
},
{
"field":"granular_markings",
"op":"=",
"value":"not supported field - just place holder"
"field": "granular_markings",
"op": "=",
"value": "not supported field - just place holder"
},
{
"field":"modified",
"op":"*",
"value":"not supported operator - just place holder"
"field": "modified",
"op": "*",
"value": "not supported operator - just place holder"
},
{
"field":"created",
"op":"=",
"value":obj_1
"field": "created",
"op": "=",
"value": obj_1
}
]
expected_errors =[
expected_errors = [
"Filter was missing a required field(key). Each filter requires 'field', 'op', 'value' keys.",
"Filter 'field' is not a STIX 2.0 common property. Currently only STIX object common properties supported",
"Filter operation(from 'op' field) not supported",
@ -119,49 +114,41 @@ def test_add_get_remove_filter():
]
ds = taxii.TAXIIDataSource()
#add
# add
ids, statuses = ds.add_filter(filters)
#7 filters should have been successfully added
# 7 filters should have been successfully added
assert len(ids) == 7
#all filters added to data source
# all filters added to data source
for idx, status in enumerate(statuses):
assert status['filter'] == filters[idx]
#proper status warnings were triggered
# proper status warnings were triggered
assert statuses[3]['errors'][0] == expected_errors[0]
assert statuses[4]['errors'][0] == expected_errors[1]
assert statuses[5]['errors'][0] == expected_errors[2]
assert statuses[6]['errors'][0] == expected_errors[3]
#File
'''
def test_data_source_file():
ds = file.FileDataSource()
assert ds.name == "DataSource"
def test_data_source_name():
ds = file.FileDataSource(name="My File Data Source")
assert ds.name == "My File Data Source"
def test_data_source_get():
ds = file.FileDataSource(name="My File Data Source")
with pytest.raises(NotImplementedError):
ds.get("foo")
#filter testing
def test_add_filter():
ds = file.FileDataSource()
'''
# def test_data_source_file():
# ds = file.FileDataSource()
#
# assert ds.name == "DataSource"
#
#
# def test_data_source_name():
# ds = file.FileDataSource(name="My File Data Source")
#
# assert ds.name == "My File Data Source"
#
#
# def test_data_source_get():
# ds = file.FileDataSource(name="My File Data Source")
#
# with pytest.raises(NotImplementedError):
# ds.get("foo")
#
# #filter testing
# def test_add_filter():
# ds = file.FileDataSource()