模擬方法屬性蟒蛇拋出AttributionError

Question

我有以下方法：

def get_countries(cities): 
    prefix = os.getenv("PREFIX") 
    ... 
    ... 

我想嘲笑這個前綴屬性（因爲我沒有找到一個方法來嘲笑在os.getenv("PREFIX")和@mock.patch.dict(os.environ, {"PREFIX": "P"})只需一個鍵 - 會嘲笑整個字典，而且有可能是我想保留其他鍵

所以我寫了以下內容：

@mock.patch.object("get_countries", "prefix", "P") 
    def test_get_aggregated_performance_records_countries_min_spend(): 
    ... 

我得到一個n錯誤：

AttributeError: get_countries does not have the attribute 'prefix' 

我在做什麼錯？

Answer 1

prefix不是get_countries函數的屬性。這是一個函數範圍的變量，也就是封閉在get_countries，但它不是get_countries :)

一個屬性如果修補os.environ是不可接受的，你可以重構暴露prefix作爲參數：

def get_countries(cities, prefix=os.getenv("PREFIX")): 
    ... 
    ... 

製作爲單元測試提供一個值是微不足道的。

---

此外，如果你patch編os.environ，它應該隻影響當前的測試案例。您需要爲單個測試指定多少個密鑰？