Esempio n. 1
0
def test_get_networks():
    mock_org_response = mock.create_autospec(requests.Response)
    mock_org_response.status_code = 200
    mock_org_response.content = org_networks()

    mock_external_response = mock.create_autospec(requests.Response)
    mock_external_response.status_code = 200
    mock_external_response.content = external_networks()

    mock_client = mock.create_autospec(VCloudClient)
    mock_client.request.side_effect = [
        mock_org_response,
        mock_external_response,
    ]

    driver = NetworkDriver(mock_client)
    assert driver.get_networks()