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)