def test_conver_to_html(mocker):
file = 'MS-DOCX-190319.docx'
mocker.patch.object(demisto, 'args', return_value={'entry_id': 'test', 'format': 'html', 'all_files': 'yes'})
ext = os.path.splitext(file)[1]
mocker.patch.object(demisto, 'getFilePath', return_value={"path": "test_data/" + file, 'name': 'test' + ext})
mocker.patch.object(demisto, 'results')
# validate our mocks are good
assert demisto.args()['entry_id'] == 'test'
main()
assert demisto.results.call_count > 10 # we have also a bunch of images
# call_args_list holds all calls as a tuple (args list, kwargs). we only need the first one
# we check if we have an html file
results = [x[0] for x in demisto.results.call_args_list if x[0][0]['File'].endswith('.html')][0]
assert len(results) == 1
assert results[0]['Type'] == entryTypes['file']
assert results[0]['File'] == 'test.html'
glob_list = glob.glob('./*' + results[0]['FileID'])
logging.getLogger().info('glob list for results: {}. list: {}'.format(results[0], glob_list))
assert glob_list
with open(glob_list[0], "r") as f:
contents = f.read()
assert 'Extensions to the Office Open XML' in contents
# assert the next result is an image
results = [x[0] for x in demisto.results.call_args_list if x[0][0]['File'].endswith('.png')][0]
assert results[0]['Type'] == entryTypes['file']
assert results[0]['File'].startswith('test_html_')
assert glob.glob('./*' + results[0]['FileID'])
def test_convert_pdf_to_html(mocker):
file = 'text-only.pdf'
mocker.patch.object(demisto,
'args',
return_value={
'entry_id': 'test',
'format': 'html',
'all_files': 'yes'
})
ext = os.path.splitext(file)[1]
mocker.patch.object(demisto,
'getFilePath',
return_value={
"path": "test_data/" + file,
'name': 'test' + ext
})
mocker.patch.object(demisto, 'results')
# validate our mocks are good
assert demisto.args()['entry_id'] == 'test'
main()
assert demisto.results.call_count == 1
results = [
x[0] for x in demisto.results.call_args_list
if x[0][0]['File'].endswith('.html')
][0]
assert results[0]['Type'] == entryTypes['file']
assert results[0]['File'] == 'test.html'
glob_list = glob.glob('./*' + results[0]['FileID'])
logging.getLogger().info('glob list for results: {}. list: {}'.format(
results[0], glob_list))
assert glob_list
# check no defunct processed
zombies, output = find_zombie_processes()
assert not zombies
assert 'defunct' not in output
def test_convert_failure(mocker):
# test with BAD format to see that we fail
file = 'MS-DOCX-190319.docx' # disable-secrets-detection
mocker.patch.object(demisto,
'args',
return_value={
'entry_id': 'test',
'format': 'BAD',
'all_files': 'yes'
})
ext = os.path.splitext(file)[1]
mocker.patch.object(demisto,
'getFilePath',
return_value={
"path": "test_data/" + file,
'name': 'test' + ext
})
return_error_mock = mocker.patch(RETURN_ERROR_TARGET)
# validate our mocks are good
assert demisto.args()['entry_id'] == 'test'
main()
assert return_error_mock.call_count == 1
# call_args last call with a tuple of args list and kwargs
err_msg = return_error_mock.call_args[0][0]
assert 'BAD' in err_msg
assert 'Error: no export filter' in err_msg
def test_conver_to_pdf(mocker, file):
mocker.patch.object(demisto, 'args', return_value={'entry_id': 'test'})
ext = os.path.splitext(file)[1]
mocker.patch.object(demisto, 'getFilePath', return_value={"path": "test_data/" + file, 'name': 'test' + ext})
mocker.patch.object(demisto, 'results')
# validate our mocks are good
assert demisto.args()['entry_id'] == 'test'
main()
# call_args is tuple (args list, kwargs). we only need the first one
results = demisto.results.call_args[0]
assert demisto.results.call_count == 1
assert len(results) == 1
assert results[0]['Type'] == entryTypes['file']
assert results[0]['File'] == 'test.pdf'
assert glob.glob('./*' + results[0]['FileID'])