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()