def test_hydrate(self):
mt = MailTree('*****@*****.**')
mt.hydrate(self.msgA)
self.assertEqual(mt.parent.author, 'From test <*****@*****.**>')
self.assertEqual(mt.nodes['*****@*****.**'].message_id, '*****@*****.**')
self.assertEqual(len(mt.nodes), 1)
def test_single_hydration(self):
mt = MailTree('*****@*****.**')
mt.hydrate(self.msgA)
self.assertEqual(mt.parent.author, 'From test <*****@*****.**>')
mt.parent.author = 'ERASED'
mt.hydrate(self.msgA)
self.assertEqual(mt.parent.author, 'ERASED')
def test_hydrate_reply(self):
mt = MailTree('*****@*****.**')
mt.hydrate(self.msgB)
self.assertEqual(mt.parent.author, 'From test <*****@*****.**>')
self.assertEqual(len(mt.nodes), 2)
self.assertEqual(len(mt.nodes['*****@*****.**'].children), 1)
self.assertEqual(mt.nodes['*****@*****.**'].children[0].message_id, '*****@*****.**')
self.assertEqual(mt.message_id, '*****@*****.**')
