def test_send_message_correct_call_when_message_is_too_long(self): message_content = "" for i in range(256): message_content += 'm' assert len(message_content) == 256 with pytest.raises(ValueError): Network.send_message(self.socket_mock, message_content)
def test_send_message_correct_call(self): message_content = "" for i in range(255): message_content += 'm' assert len(message_content) == 255 self.socket_mock.send.side_effect = [0, 1, 0, 100, 0, 155] Network.send_message(self.socket_mock, message_content) self.socket_mock.send.assert_has_calls([call(struct.pack("B", 255)), call(struct.pack("B", 255)), call(message_content), call(message_content), call(message_content[100:]), call(message_content[100:]), ])