def setUp(self):
self.msg_helper = self.add_helper(MessageHelper())
self.request = DummyRequest([''])
self.formatter = CsvFormatter()
def setUp(self):
self.msg_helper = self.add_helper(MessageHelper())
self.request = DummyRequest([''])
self.formatter = CsvFormatter()
class TestCsvFormatter(VumiTestCase):
def setUp(self):
self.msg_helper = self.add_helper(MessageHelper())
self.request = DummyRequest([''])
self.formatter = CsvFormatter()
def test_implements_IMessageFormatter(self):
self.assertTrue(IMessageFormatter.providedBy(self.formatter))
def test_add_http_headers(self):
self.formatter.add_http_headers(self.request)
self.assertEqual(
self.request.responseHeaders.getRawHeaders('Content-Type'),
['text/csv; charset=utf-8'])
def test_write_row_header(self):
self.formatter.write_row_header(self.request)
self.assertEqual(self.request.written, [
"timestamp,message_id,to_addr,from_addr,in_reply_to,session_event,"
"content,group\r\n"
])
def assert_row_written(self, row, row_template, msg):
self.assertEqual(row, [
row_template % {
'ts': msg['timestamp'].isoformat(),
'id': msg['message_id'],
}
])
def test_write_row(self):
msg = self.msg_helper.make_inbound("foo")
self.formatter.write_row(self.request, msg)
self.assert_row_written(self.request.written,
"%(ts)s,%(id)s,9292,+41791234567,,,foo,\r\n",
msg)
def test_write_row_with_in_reply_to(self):
msg = self.msg_helper.make_inbound("foo", in_reply_to="msg-2")
self.formatter.write_row(self.request, msg)
self.assert_row_written(
self.request.written,
"%(ts)s,%(id)s,9292,+41791234567,msg-2,,foo,\r\n", msg)
def test_write_row_with_session_event(self):
msg = self.msg_helper.make_inbound("foo", session_event="new")
self.formatter.write_row(self.request, msg)
self.assert_row_written(
self.request.written,
"%(ts)s,%(id)s,9292,+41791234567,,new,foo,\r\n", msg)
def test_write_row_with_group(self):
msg = self.msg_helper.make_inbound("foo", group="#channel")
self.formatter.write_row(self.request, msg)
self.assert_row_written(
self.request.written,
"%(ts)s,%(id)s,9292,+41791234567,,,foo,#channel\r\n", msg)
def test_write_row_with_unicode_content(self):
msg = self.msg_helper.make_inbound(u"føø", group="#channel")
self.formatter.write_row(self.request, msg)
self.assert_row_written(
self.request.written,
u"%(ts)s,%(id)s,9292,+41791234567,,,føø,#channel\r\n".encode(
"utf-8"), msg)
class TestCsvFormatter(VumiTestCase):
def setUp(self):
self.msg_helper = self.add_helper(MessageHelper())
self.request = DummyRequest([''])
self.formatter = CsvFormatter()
def test_implements_IMessageFormatter(self):
self.assertTrue(IMessageFormatter.providedBy(self.formatter))
def test_add_http_headers(self):
self.formatter.add_http_headers(self.request)
self.assertEqual(
self.request.responseHeaders.getRawHeaders('Content-Type'),
['text/csv; charset=utf-8'])
def test_write_row_header(self):
self.formatter.write_row_header(self.request)
self.assertEqual(self.request.written, [
"timestamp,message_id,to_addr,from_addr,in_reply_to,session_event,"
"content,group\r\n"
])
def assert_row_written(self, row, row_template, msg):
self.assertEqual(row, [row_template % {
'ts': msg['timestamp'].isoformat(),
'id': msg['message_id'],
}])
def test_write_row(self):
msg = self.msg_helper.make_inbound("foo")
self.formatter.write_row(self.request, msg)
self.assert_row_written(
self.request.written,
"%(ts)s,%(id)s,9292,+41791234567,,,foo,\r\n", msg)
def test_write_row_with_in_reply_to(self):
msg = self.msg_helper.make_inbound("foo", in_reply_to="msg-2")
self.formatter.write_row(self.request, msg)
self.assert_row_written(
self.request.written,
"%(ts)s,%(id)s,9292,+41791234567,msg-2,,foo,\r\n", msg)
def test_write_row_with_session_event(self):
msg = self.msg_helper.make_inbound("foo", session_event="new")
self.formatter.write_row(self.request, msg)
self.assert_row_written(
self.request.written,
"%(ts)s,%(id)s,9292,+41791234567,,new,foo,\r\n", msg)
def test_write_row_with_group(self):
msg = self.msg_helper.make_inbound("foo", group="#channel")
self.formatter.write_row(self.request, msg)
self.assert_row_written(
self.request.written,
"%(ts)s,%(id)s,9292,+41791234567,,,foo,#channel\r\n", msg)
def test_write_row_with_unicode_content(self):
msg = self.msg_helper.make_inbound(u"føø", group="#channel")
self.formatter.write_row(self.request, msg)
self.assert_row_written(
self.request.written,
u"%(ts)s,%(id)s,9292,+41791234567,,,føø,#channel\r\n".encode(
"utf-8"),
msg)
