def test_protocol_attribute(self): h = SmtpSession(None, None, None) self.assertEqual('SMTP', h.protocol) h.extended_smtp = True self.assertEqual('ESMTP', h.protocol) h.security = 'TLS' self.assertEqual('ESMTPS', h.protocol) h.auth = 'test' self.assertEqual('ESMTPSA', h.protocol)