def test_write_sv_file_list_of_lists_number_format():
mock_output = MagicMock()
f = File()
f.options.set_value('number_format', 'hexadecimal')
f.write_sv_file([[10, 20, 3], [4, 5], [6]], mock_output, ' ')
mock_output.write.assert_called_once_with(
os.linesep.join(['0xa 0x14 0x3', '0x4 0x5', '0x6']))
def test_output_defaults():
with patch('builtins.open', mock_open()) as mock_open_f:
f = File()
f.output(TEST_OUTPUT_LINES)
mock_open_f.assert_called_once_with('acsploit_output.dat', 'w')
mock_open_f().write.assert_has_calls(
[call(os.linesep.join(TEST_OUTPUT_LINES_STR)),
call(os.linesep)])
def test_output_no_trailing_newline():
with patch('builtins.open', mock_open()) as mock_open_f:
f = File()
f.options.set_value('final_newline', 'No')
f.output(TEST_OUTPUT_LINES)
mock_open_f.assert_called_once_with('acsploit_output.dat', 'w')
mock_open_f().write.assert_called_once_with(
os.linesep.join(TEST_OUTPUT_LINES_STR))
def test_output_custom_filename():
with patch('builtins.open', mock_open()) as mock_open_f:
f = File()
f.options.set_value('filename', 'very_suspicious_file.exe')
f.output(TEST_OUTPUT_LINES)
mock_open_f.assert_called_once_with('very_suspicious_file.exe', 'w')
mock_open_f().write.assert_has_calls(
[call(os.linesep.join(TEST_OUTPUT_LINES_STR)),
call(os.linesep)])
def test_write_template_file_replace_first_only():
mock_output = MagicMock()
f = File()
f.options['replace_first_only'] = True
template_file = os.path.join(os.path.dirname(__file__), 'template.txt')
f.write_template_file(TEST_OUTPUT_LINES, mock_output, ' ', template_file,
'!!!')
mock_output.write.assert_called_once_with('hello %s !!!' %
' '.join(TEST_OUTPUT_LINES_STR))
def test_write_template_file():
mock_output = MagicMock()
f = File()
template_file = os.path.join(os.path.dirname(__file__), 'template.txt')
f.write_template_file(TEST_OUTPUT_LINES, mock_output, ' ', template_file,
'!!!')
mock_output.write.assert_called_once_with(
'hello %s %s' %
(' '.join(TEST_OUTPUT_LINES_STR), ' '.join(TEST_OUTPUT_LINES_STR)))
def test_output_number_format():
with patch('builtins.open', mock_open()) as mock_open_f:
f = File()
f.options.set_value('number_format', 'octal')
f.output(TEST_OUTPUT_LINES)
mock_open_f.assert_called_once_with('acsploit_output.dat', 'w')
mock_open_f().write.assert_has_calls([
call(os.linesep.join(['0o21', 'hello', 'c', '[1, 20, 3]'])),
call(os.linesep)
])
def test_convert_item_ints_octal():
f = File()
f.options.set_value('number_format', 'octal')
for item in [7, 8, 15, 20]:
assert f.convert_item(item) == oct(item)
def test_convert_item_ints_hexadecimal():
f = File()
f.options.set_value('number_format', 'hexadecimal')
for item in [7, 8, 15, 20]:
assert f.convert_item(item) == hex(item)
def test_write_sv_file_list_of_lists():
mock_output = MagicMock()
f = File()
f.write_sv_file([[1, 2, 3], [4, 5], [6]], mock_output, ' ')
mock_output.write.assert_called_once_with(
os.linesep.join(['1 2 3', '4 5', '6']))
def test_write_sv_file():
mock_output = MagicMock()
f = File()
f.write_sv_file(TEST_OUTPUT_LINES, mock_output, ' ')
mock_output.write.assert_called_once_with(' '.join(TEST_OUTPUT_LINES_STR))
def test_convert_item_non_int():
f = File()
for item in ['a', [1, 2, 3]]:
assert f.convert_item(item) == str(item)
def test_write_binary_file():
mock_output = MagicMock()
f = File()
f.write_binary_file(TEST_OUTPUT_LINES, mock_output)
mock_output.write.assert_has_calls(
[call(line) for line in TEST_OUTPUT_LINES])