def test_write(self): mock_out = MagicMock() writer = CalendarWriter(mock_out) writer.write("I like writing stuff") mock_out.write.assert_called_once_with(b"I like writing stuff")
def test_write_wrap(self): out = six.BytesIO() writer = CalendarWriter(out) message = ( "This is a very long piece of text which will have to be wrapped " "otherwise it would be toooooooo long. It might even have to be " "wrapped twice because it's just that long." ) writer.write(message) value = out.getvalue() # Don't strip the space after newline, otherwise line length would # be off by 1. lines = value.split(b"\r\n") self.assertEqual(max(len(l) for l in lines), 75) self.assertFalse(value.endswith(b"\r\n "))