Unittest.mock을 사용하여 컨텍스트 관리자에서 사용되는 객체를 모의하는 방법은 무엇입니까?

unittest.mock을 사용하여 컨텍스트 관리자에서 사용되는 개체를 모의하는 방법

with 문을 사용하는 코드를 테스트할 때 다음과 같은 경우가 발생할 수 있습니다. 기본 개체를 조롱하는 것이 어렵습니다. 다음 예를 살펴보겠습니다.

def testme(filepath):
    with open(filepath) as f:
        return f.read()

unittest.mock으로 이 함수를 테스트하려면 open 함수를 모의해야 합니다. 방법은 다음과 같습니다.

Python 3:

1. builtins.open을 패치하고 unittest.mock에서 mock_open을 사용합니다.
2. 패치를 컨텍스트로 사용 관리자 또는 장식가. 두 경우 모두 원하는 읽기 데이터와 함께 mock_open을 전달하고 모의 파일이 예상 인수로 호출되는지 확인합니다.

Python 2:

1. buildins.open 대신 __builtin__.open을 패치합니다.
2. mock을 컨텍스트 관리자로 사용하고 이전과 같이 mock_open을 전달합니다.

예:

<code class="python">from unittest.mock import patch, mock_open

@patch("builtins.open", new_callable=mock_open, read_data="data")
def test_patch(mock_file):
    assert open("path/to/open").read() == "data"
    mock_file.assert_called_with("path/to/open")</code>

Python 3에서는 패치가 모의 객체를 테스트 함수에 대한 인수로 전달하는 반면, Python 2에서는 모의 파일을 명시적으로 어설션해야 한다는 점을 기억하세요.

위 내용은 Unittest.mock을 사용하여 컨텍스트 관리자에서 사용되는 객체를 모의하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!