def test_add_received_header(self):
     env = Envelope('*****@*****.**', ['*****@*****.**'])
     env.parse('')
     env.timestamp = 1234567890
     env.client['name'] = 'mail.example.com'
     env.client['ip'] = '1.2.3.4'
     env.client['protocol'] = 'ESMTPS'
     env.receiver = 'test.com'
     arh = AddReceivedHeader()
     arh.apply(env)
     self.assertRegexpMatches(env.headers['Received'],
             r'from mail\.example\.com \(unknown \[1.2.3.4\]\) by test.com '
             r'\(slimta [^\)]+\) with ESMTPS for <*****@*****.**>; ')
 def test_add_received_header(self):
     env = Envelope('*****@*****.**', ['*****@*****.**'])
     env.parse(b'From: [email protected]\r\n')
     env.timestamp = 1234567890
     env.client['name'] = 'mail.example.com'
     env.client['ip'] = '1.2.3.4'
     env.client['protocol'] = 'ESMTPS'
     env.receiver = 'test.com'
     arh = AddReceivedHeader()
     arh.apply(env)
     self.assertRegexpMatches(
         env.headers['Received'],
         r'from mail\.example\.com \(unknown \[1.2.3.4\]\) by test.com '
         r'\(slimta [^\)]+\) with ESMTPS for <*****@*****.**>; ')