def test_write(self):
mock_socket = get_mock_socket()
socket_wrapper = SMTPSocket(mock_socket)
test_data = b"abc123\r\n"
socket_wrapper.write(test_data)
mock_socket.send.assert_called_with(test_data)
def _send_response(self, smtp_socket: SMTPSocket, response: BaseResponse):
if self._shared_state.esmtp_capable:
self.__logger.info("%s Sending extended response to client with SMTP code %s",
self._shared_state.transaction_id, response.get_code())
smtp_socket.write(response.get_extended_smtp_response().replace("<domain>", self.__server_name).encode())
else:
self.__logger.info("%s Sending response to client with SMTP code %s",
self._shared_state.transaction_id,
response.get_code())
smtp_socket.write(response.get_smtp_response().replace("<domain>", self.__server_name).encode())