def test_get_network_by_name_failure(mock_get_networks):
mock_get_networks.return_value = []
mock_client = mock.create_autospec(VCloudClient)
driver = NetworkDriver(mock_client)
with nose.tools.assert_raises(errors.VCloudNotFoundError):
driver.get_network_by_name('test-network')
def test_get_network_by_name(mock_get_networks):
mock_get_networks.return_value = networks()
mock_client = mock.create_autospec(VCloudClient)
driver = NetworkDriver(mock_client)
name = 'test-org-network'
network = driver.get_network_by_name(name)
assert network.get('name') == name
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()
