def test_apiInput(self):
""" load api_test.py and check the results """
Config.load_plugin('inputs/api_input.py')
module = Config.load_plugin('test/api_test.py')
doc = Doc.from_url("api_test")
result = doc.query( { 'page' : 1 } )
self.assertEqual( search( ('reqres_ids', 'name'), result), "George Bluth")
def do_query(path, dic):
d = Doc.from_url(path)
rv = d.query(dic)
global queryCount
queryCount += 1
print("Query{}: {}".format(queryCount, path))
pprint(rv)
print('============')
def test_csvInput(self):
""" load csv_test.py and check the results """
module = Config.load_plugin('test/csv_test.py')
doc = Doc.from_url("csv_test")
result = doc.query( { 'hash' : '41e25e514d90e9c8bc570484dbaff62b' } )
self.assertEqual( result, {'name':'cmd.exe',
'hash':'41e25e514d90e9c8bc570484dbaff62b',
'from_csv_input': True,
'date.created': datetime(2018,2,20,11,23,0),
'nonce' : 'ca79d9cbb8c73cbe610dfa05030a2183'} )
def query(path):
# simple and it works
src = None
try:
src = Doc.from_url(path)
except ValueError as e:
return "Error: <{}>".format(e)
except BadRequest as e:
return json.dumps({"Error": "Bad Request <{}>".format(e)})
if not src:
return json.dumps(
["Error: Unknown API endpoint <{}>".format(path),
Doc.urls()])
#logger.info(pformat(request.values))
rv = src.query(request.values)
#logger.info(pformat(rv))
return json.dumps(rv,
indent=Config.get("JSON_PRETTY"),
default=custom_json_encoder)
