如何在python中的方法內部模擬導入

Question

這是一個非常簡單的問題。 如果我在方法中有導入，如何模擬該特定導入？ 例如：

def myFunction(self):
    auth_token_path = self.authTokenPath()
    import json

    if os.path.exists(auth_token_path):
      auth_token = json.load(open(auth_token_path, 'r'))
      return auth_token

我該如何模擬這個“導入json”？ 有什么辦法可以解決這個問題？ 我試圖將其導入測試函數中，但我認為這不是正確的方法（並且也不起作用）。

Answer 1

模擬此功能

def myFunction(self):
auth_token_path = self.authTokenPath()
import json

if os.path.exists(auth_token_path):
  auth_token = json.load(open(auth_token_path, 'r'))
  return auth_token

我們需要做的就是添加與需要模擬的導入功能相關的裝飾器PATCH（）：

@patch('json.load')
def myFunction(self,json):
auth_token_path = self.authTokenPath()
import json

if os.path.exists(auth_token_path):
  auth_token = json.load(open(auth_token_path, 'r'))
  return auth_token