對因設計而未反映在函數返回值中的函數更改進行單元測試

Question

import smtplib
from email.mime.text import MIMEText

def test():
    frm = "a@b.com"
    to = "c@d.com"
    # Original line: msg = MIMEText("Hello, Dear")
    msg = MIMEText("Hello, World")   # Changed line

    try:
        s = smtplib.SMTP("localhost")
        s.sendmail(frm, to, msg.as_string())
    except Exception as e:
        return False

    return True


test()

現在，我要對在味精行上所做的更改進行單元測試。 換句話說，我希望能夠斷言在代碼運行時msg值確實是“ Hello，World”，或者sendmail（）調用確實發送了一條消息，說“ Hello，World”。 我該怎么做？

Answer 1

這是使用模擬的一個很好的例子。 在任何情況下，您都不希望單元測試實際發送電子郵件。