def test_add_message_id_header_existing(self):
     env = Envelope()
     env.parse(b'Message-Id: testing\r\n')
     amih = AddMessageIdHeader()
     self.assertEqual('testing', env.headers['Message-Id'])
     amih.apply(env)
     self.assertEqual('testing', env.headers['Message-Id'])
 def test_add_message_id_header_existing(self):
     env = Envelope()
     env.parse('Message-Id: testing\r\n')
     amih = AddMessageIdHeader()
     self.assertEqual('testing', env.headers['Message-Id'])
     amih.apply(env)
     self.assertEqual('testing', env.headers['Message-Id'])
 def test_add_message_id_header(self):
     env = Envelope()
     env.parse(b'')
     env.timestamp = 1234567890
     amih = AddMessageIdHeader('example.com')
     self.assertEqual(None, env.headers['Message-Id'])
     amih.apply(env)
     pattern = r'^<[0-9a-fA-F]{32}\[email protected]>$'
     self.assertRegexpMatches(env.headers['Message-Id'], pattern)
 def test_add_message_id_header(self):
     env = Envelope()
     env.parse('')
     env.timestamp = 1234567890
     amih = AddMessageIdHeader('example.com')
     self.assertEqual(None, env.headers['Message-Id'])
     amih.apply(env)
     pattern = r'^<[0-9a-fA-F]{32}\[email protected]>$'
     self.assertRegexpMatches(env.headers['Message-Id'], pattern)