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()