class UppercaseSMTP(smtplib.SMTP):
def putcmd(self, cmd, args=""):
return super().putcmd(cmd.upper(), args)
Test
import unittest
from unittest.mock import patch
import smtplib
class UppercaseSMTP(smtplib.SMTP):
def putcmd(self, cmd, args=""):
return super().putcmd(cmd.upper(), args)
class TestUppercaseSMTP(unittest.TestCase):
@patch('smtplib.SMTP.putcmd')
def test_putcmd_uppercase(self, mock_putcmd):
smtp = UppercaseSMTP()
smtp.putcmd("ehlo", "example.com")
mock_putcmd.assert_called_once_with("EHLO", "example.com")
if __name__ == '__main__':
unittest.main()