Ejemplos de MagicMock en Python

Ejemplo n.º 1

def test_close():
    transport = mock.MagicMock()
    client = SecretClient(vault_url="https://localhost",
                          credential=object(),
                          transport=transport)
    client.close()
    assert transport.__enter__.call_count == 0
    assert transport.__exit__.call_count == 1

Ejemplo n.º 2

def test_crypto_client_close():
    transport = mock.MagicMock()
    client = CryptographyClient(key="https://localhost/a/b/c",
                                credential=object(),
                                transport=transport)
    client.close()
    assert transport.__enter__.call_count == 0
    assert transport.__exit__.call_count == 1

Ejemplo n.º 3

Archivo: test_context_manager.py Proyecto: zolvarga/azure-sdk-for-python

def test_context_manager():
    transport = mock.MagicMock()
    client = CertificateClient(vault_url="https://localhost", credential=object(), transport=transport)

    with client:
        assert transport.__enter__.call_count == 1
    assert transport.__enter__.call_count == 1
    assert transport.__exit__.call_count == 1

Ejemplo n.º 4

def test_crypto_client_context_manager():
    transport = mock.MagicMock()
    client = CryptographyClient(key="https://localhost/a/b/c",
                                credential=object(),
                                transport=transport)

    with client:
        assert transport.__enter__.call_count == 1
    assert transport.__enter__.call_count == 1
    assert transport.__exit__.call_count == 1