#!/usr/bin/python
'''Unit tests for proto.py'''

from proto import Message

class TestMessage:
    
    data = [('test1', 'testing'), ('test2', 'still\nstesting')]
    def test_sanity(self):
        for nick, message in self.data:
            m = Message(nick, message)
            assert m == Message(data=m.data)
    
    def test_repr(self):
        for nick, message in self.data:
            m = Message(nick, message)
            assert m == eval(repr(m))