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
