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

solution to get 404 mocking problem

stix2.0
= 2018-05-18 11:19:05 -04:00
parent 8fc421e7d4
commit 6004ec597c
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
stix2/test/test_datastore_taxii.py
View File

@ -45,17 +45,17 @@ class MockTAXIICollectionEndpoint(Collection):
resp.status_code = 404 resp.status_code = 404
resp.raise_for_status() resp.raise_for_status()
def get_object(self, id_, version=None): def get_object(self, id, version=None):
self._verify_can_read() self._verify_can_read()
kwargs = {"id": id_} query_params = None
if version: if version:
kwargs.update({"version": version}) query_params = _filter_kwargs_to_query_params({"version": version})
if query_params:
query_params = _filter_kwargs_to_query_params(kwargs) query_params = json.loads(query_params)
full_filter = BasicFilter(query_params or {}) full_filter = BasicFilter(query_params or {})
objs = full_filter.process_filter( objs = full_filter.process_filter(
self.objects, self.objects,
("version", "id"), ("version",),
[] []
) )
if objs: if objs:
@ -339,7 +339,7 @@ def test_can_write_error(collection_no_rw_access):
assert "Collection object provided does not have write access" in str(excinfo.value) assert "Collection object provided does not have write access" in str(excinfo.value)
def test_get_404(collection): def test_get_404():
"""a TAXIICollectionSource.get() call that receives an HTTP 404 response """a TAXIICollectionSource.get() call that receives an HTTP 404 response
code from the taxii2client should be be returned as None. code from the taxii2client should be be returned as None.
@ -347,7 +347,16 @@ def test_get_404(collection):
nonexistent resources or lack of access. Decided that None is acceptable nonexistent resources or lack of access. Decided that None is acceptable
reponse to imply that state of the TAXII endpoint. reponse to imply that state of the TAXII endpoint.
""" """
ds = TAXIICollectionStore(collection)
class TAXIICollection404():
can_read = True
def get_object(self, id, version=None):
resp = Response()
resp.status_code = 404
resp.raise_for_status()
ds = TAXIICollectionSource(TAXIICollection404())
# this will raise 404 from mock TAXII Client but TAXIICollectionStore # this will raise 404 from mock TAXII Client but TAXIICollectionStore
# should handle gracefully and return None # should handle gracefully and return None