def test_network_exists_with_valid_network(self):
vip = lb_types.VIP()
vip.network_id = uuidutils.generate_uuid()
network = network_models.Network(id=vip.network_id)
with mock.patch('octavia.common.utils.get_network_driver') as net_mock:
net_mock.return_value.get_network.return_value = network
self.assertEqual(
validate.network_exists_optionally_contains_subnet(vip),
network)
def test_network_exists_with_bad_network(self):
vip = lb_types.VIP()
vip.network_id = uuidutils.generate_uuid()
with mock.patch('octavia.common.utils.get_network_driver') as net_mock:
net_mock.return_value.get_network = mock.Mock(
side_effect=network_base.NetworkNotFound('Network not found'))
self.assertRaises(
exceptions.InvalidSubresource,
validate.network_exists_optionally_contains_subnet, vip)
def test_network_exists_with_bad_subnet(self):
vip = lb_types.VIP()
vip.network_id = uuidutils.generate_uuid()
vip.subnet_id = uuidutils.generate_uuid()
network = network_models.Network(id=vip.network_id)
with mock.patch('octavia.common.utils.get_network_driver') as net_mock:
net_mock.return_value.get_network.return_value = network
self.assertRaises(
exceptions.InvalidSubresource,
validate.network_exists_optionally_contains_subnet,
vip.network_id, vip.subnet_id)
