def test_write_end_contenetline(self):
out = six.BytesIO()
writer = CalendarWriter(out)
writer.end_contentline()
self.assertEqual(out.getvalue(), b"\r\n")
def test_write_start_contenetline(self):
out = six.BytesIO()
writer = CalendarWriter(out)
writer.start_contentline("foo")
self.assertEqual(out.getvalue(), b"foo:")
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 "))
def test_begin(self):
writer = CalendarWriter(six.BytesIO())
writer.start_contentline = MagicMock()
writer.value = MagicMock()
writer.end_contentline = MagicMock()
writer.begin(sentinel.section)
writer.start_contentline.assert_called_once_with("BEGIN")
writer.value.assert_called_once_with(sentinel.section)
writer.end_contentline.assert_called_once()
def test_contentline(self):
writer = CalendarWriter(six.BytesIO())
writer.start_contentline = MagicMock()
writer.value = MagicMock()
writer.end_contentline = MagicMock()
writer.contentline(sentinel.name, sentinel.value)
writer.start_contentline.assert_called_once_with(sentinel.name)
writer.value.assert_called_once_with(sentinel.value)
writer.end_contentline.assert_called_once()
