class MessageBuilderTestCase(BaseTestCase):
"""
Test cases for MessageBuilder class
"""
def setUp(self):
BaseTestCase.setUp(self)
self.mb = MessageBuilder()
def compare_to(self, expected):
self.assertEqual(
base64.b64encode(self.mb.build()),
expected)
def testEmptyOnInitialization(self):
assert self.mb.txt == ''
assert self.mb.html == ''
assert self.mb.formatting == ''
assert self.mb.img is None
assert self.mb.response is None
def testPlainMessage(self):
self.mb.add_text('plain text')
self.compare_to('UwAAAAsAAAAAAAAAAAAAADxzcGFuIHN0eWxlPSJjb2xvcjojM' + \
'DAwMDAwOyBmb250LWZhbWlseTonTVMgU2hlbGwgRGxnIDInOyBmb250LXNpem' + \
'U6OXB0OyAiPjwvc3Bhbj4AcGxhaW4gdGV4dAA=')
def testHtmlMessage(self):
self.mb.add_html('<b>html content</b>')
self.compare_to('ZgAAAAEAAAAAAAAAAAAAADxzcGFuIHN0eWxlPSJjb2xvcjojM' + \
'DAwMDAwOyBmb250LWZhbWlseTonTVMgU2hlbGwgRGxnIDInOyBmb250LXNpem' + \
'U6OXB0OyAiPjxiPmh0bWwgY29udGVudDwvYj48L3NwYW4+AAA=')
def testPlainAndHtmlMessage(self):
self.mb.add_text('plain text').add_html('<b>html content</b>')
self.compare_to('ZgAAAAsAAAAAAAAAAAAAADxzcGFuIHN0eWxlPSJjb2xvcjojMDAwMDAwOyBmb250LWZhbWlseTonTVMgU2hlbGwgRGxnIDInOyBmb250LXNpemU6OXB0OyAiPjxiPmh0bWwgY29udGVudDwvYj48L3NwYW4+AHBsYWluIHRleHQA')
def test_send(self):
message = MessageBuilder()
message.add_text('tests')
message.add_html('test bota')
self.botmaster.send(message,[123],True)
def setUp(self):
BaseTestCase.setUp(self)
self.mb = MessageBuilder()
