def test_create_values_for_returned_dict_2(self): """Test XSOAR JSON out""" from ExportIndicators import create_values_for_returned_dict, FORMAT_XSOAR_JSON, CTX_VALUES_KEY, RequestArguments with open('ExportIndicators_test/TestHelperFunctions/demisto_iocs.json', 'r') as iocs_json_f: iocs_json = json.load(iocs_json_f) request_args = RequestArguments(query='', out_format=FORMAT_XSOAR_JSON) returned_dict, _ = create_values_for_returned_dict(iocs_json, request_args) json_out = json.loads(returned_dict.get(CTX_VALUES_KEY)) assert json_out == iocs_json
def test_create_values_for_returned_dict_3(self): """Test XSOAR JSON_SEQ out""" from ExportIndicators import create_values_for_returned_dict, FORMAT_XSOAR_JSON_SEQ, CTX_VALUES_KEY, RequestArguments with open('ExportIndicators_test/TestHelperFunctions/demisto_iocs.json', 'r') as iocs_json_f: iocs_json = json.loads(iocs_json_f.read()) request_args = RequestArguments(query='', out_format=FORMAT_XSOAR_JSON_SEQ) returned_dict, _ = create_values_for_returned_dict(iocs_json, request_args) json_seq_out = returned_dict.get(CTX_VALUES_KEY) for seq_line in json_seq_out.split('\n'): assert json.loads(seq_line) in iocs_json
def test_create_values_for_returned_dict_4(self): """Test TEXT out""" from ExportIndicators import create_values_for_returned_dict, FORMAT_TEXT, CTX_VALUES_KEY, RequestArguments with open('ExportIndicators_test/TestHelperFunctions/demisto_iocs.json', 'r') as iocs_json_f: iocs_json = json.loads(iocs_json_f.read()) request_args = RequestArguments(query='', out_format=FORMAT_TEXT) returned_dict, _ = create_values_for_returned_dict(iocs_json, request_args) text_out = returned_dict.get(CTX_VALUES_KEY) with open('ExportIndicators_test/TestHelperFunctions/iocs_cache_values_text.json', 'r') as iocs_txt_f: iocs_txt_json = json.load(iocs_txt_f) for line in text_out.split('\n'): assert line in iocs_txt_json
def test_create_values_for_returned_dict_7(self): """Test CSV out""" from ExportIndicators import create_values_for_returned_dict, FORMAT_CSV, RequestArguments, CTX_VALUES_KEY with open('ExportIndicators_test/TestHelperFunctions/demisto_iocs.json', 'r') as iocs_json_f: iocs_json = json.loads(iocs_json_f.read()) request_args = RequestArguments(query='', out_format=FORMAT_CSV) returned_dict, _ = create_values_for_returned_dict(iocs_json, request_args) csv_out = returned_dict.get(CTX_VALUES_KEY) # assert len(csv_out) == IOC_RES_LEN + 1 with open('ExportIndicators_test/TestHelperFunctions/iocs_out_csv_old.txt', 'r') as iocs_out_f: expected_csv_out = iocs_out_f.read() for csv_lint in csv_out.split('\n'): assert csv_lint in expected_csv_out