def test_get_not_worked(self):
test_message = "test-message"
cli_output = CliOutput()
return_expected = f"{cli_output._CliOutput__not_worked}{Colors.LIGHT_GRAY}{test_message}{Colors.RESET}"
returned_msg = cli_output._CliOutput__get_not_worked(test_message)
self.assertIsInstance(returned_msg, str)
self.assertEqual(returned_msg, return_expected)
def test_get_abort(self):
test_message = "test-message"
cli_output = CliOutput()
return_expected = f"{cli_output._CliOutput__abort}{test_message}"
returned_msg = cli_output._CliOutput__get_abort(test_message)
self.assertIsInstance(returned_msg, str)
self.assertEqual(returned_msg, return_expected)
def test_get_formatted_result_with_result_custom(self,
mock_format_items: Mock):
test_payload = "test_payload"
test_status_code = "200"
test_rtt = "300 ms"
test_length = "50 KB"
test_words = "50"
test_lines = "10"
mock_format_items.return_value = (test_payload, test_status_code,
test_rtt, test_length, test_words,
test_lines)
test_result = Result(response=ResponseMock())
test_result.custom['test_0'] = None
test_result.custom['test_1'] = "test_custom"
return_expected = (
f"{test_payload} {Colors.GRAY}["
f"{Colors.LIGHT_GRAY}Code{Colors.RESET} {test_status_code} | "
f"{Colors.LIGHT_GRAY}RTT{Colors.RESET} {test_rtt} | "
f"{Colors.LIGHT_GRAY}Size{Colors.RESET} {test_length} | "
f"{Colors.LIGHT_GRAY}Words{Colors.RESET} {test_words} | "
f"{Colors.LIGHT_GRAY}Lines{Colors.RESET} {test_lines}{Colors.GRAY}]{Colors.RESET}"
f"\n{Colors.LIGHT_YELLOW}|_ test_1: "
f"{ResultUtils.format_custom_field(test_result.custom['test_1'])}{Colors.RESET}"
)
returned_data = CliOutput()._CliOutput__get_formatted_result(
test_result)
mock_format_items.assert_called_once_with(test_result)
self.assertIsInstance(returned_data, str)
self.assertEqual(returned_data, return_expected)
def test_get_formatted_status_with_status_500(self):
test_status = 500
expected_color = f"{Colors.BOLD}{Colors.RED}"
return_expected = f"{expected_color}{test_status}{Colors.RESET}"
returned_status = CliOutput()._CliOutput__get_formatted_status(
test_status)
self.assertIsInstance(returned_status, str)
self.assertEqual(returned_status, return_expected)
def test_get_formatted_payload_with_path_fuzz(self):
test_result = Result(
response=ResponseMock(),
fuzz_type=PATH_FUZZING,
)
returned_payload = CliOutput()._CliOutput__get_formatted_payload(
test_result)
self.assertIsInstance(returned_payload, str)
self.assertEqual(returned_payload, get_path(test_result.url))
def test_get_formatted_payload(self):
test_result = Result(
response=ResponseMock(),
payload=Payload("test-payload"),
)
returned_payload = CliOutput()._CliOutput__get_formatted_payload(
test_result)
self.assertIsInstance(returned_payload, str)
self.assertEqual(returned_payload, test_result.payload)
def test_get_formatted_result_items(self, mock_format_result: Mock,
mock_format_status: Mock):
test_result = Result(response=ResponseMock(),
payload=Payload("test-payload"))
expected_status = "test_status"
mock_format_status.return_value = expected_status
return_expected = ('', expected_status, '', '', '', '')
mock_format_result.return_value = ('', '', '', '', '')
cli_output = CliOutput()
returned_items = cli_output._CliOutput__get_formatted_result_items(
test_result)
mock_format_status.assert_called_once_with(test_result.status)
mock_format_result.assert_called_once_with(
cli_output._CliOutput__get_formatted_payload(test_result),
test_result.rtt, test_result.body_size, test_result.words,
test_result.lines)
self.assertIsInstance(returned_items, tuple)
self.assertEqual(returned_items, return_expected)
def test_get_time(self, mock_datetime: Mock):
test_datetime_now = datetime(2021, 1, 1, 0, 0)
mock_datetime.now.return_value = test_datetime_now
expected_time = test_datetime_now.strftime("%H:%M:%S")
return_expected = (f'{Colors.GRAY}[{Colors.LIGHT_GREEN}{expected_time}'
f'{Colors.GRAY}]{Colors.RESET} ')
returned_time = CliOutput()._CliOutput__get_time()
self.assertIsInstance(returned_time, str)
self.assertEqual(returned_time, return_expected)
def test_get_formatted_payload_with_path_fuzz_and_raise_exception(
self, mock_get_path: Mock):
test_result = Result(
response=ResponseMock(),
fuzz_type=PATH_FUZZING,
)
mock_get_path.side_effect = ValueError
returned_payload = CliOutput()._CliOutput__get_formatted_payload(
test_result)
self.assertIsInstance(returned_payload, str)
self.assertEqual(returned_payload, test_result.url)