Pythonとunittestライブラリを使用して関数呼び出しをテストする方法

Question

次の関数のテストを書く方法について質問がありましたか？ここでは、カバーされている部分の私のテストですが、私はどのように印刷文をカバーし、get_employee_name関数を呼び出すテストを変更するか分からない。どんな助けもありがとう！ここで

被覆部分をテストするために私のコードです：

DATA = { 
"employee_name": "Brian Weber", 
"minutes": 120, 
"task_name": "Surfing", 
"notes": "These are my notes.", 
"date": "2016-12-25" 
} 
​ 
​ 
class WorkLogTests(unittest.TestCase): 
    def test_get_employee_name(self): 
     with mock.patch('builtins.input', 
      return_value=DATA["employee_name"]): 
      assert worklog.get_employee_name() == DATA["employee_name"] 

Answer 1

ユーザーが何も入力しなかった場合、私は再帰を使用していたていた最初の問題。だから私はコードをリファクタリングしてwhileループを使用してを使用してください。ユーザ入力がない場合はに進みます。 新しいコードとすべての行を対象とするテストは次のとおりです。

def get_employee_name(): 
"""Prompt the employee for their name.""" 
while True: 
    employee_name = input("Enter employee name: ") 
    if len(employee_name) == 0: 
     print("\nYou must enter your name!\n") 
     continue 
    else: 
     return employee_name 

def test_get_employee_name(self): 
    with mock.patch('builtins.input', side_effect=["", "Brian Weber"], 
     return_value=DATA["employee_name"]): 
     assert worklog.get_employee_name() == DATA["employee_name"]