fix more tests

This commit is contained in:
Vishvananda Ishaya
2011-07-26 22:55:58 -07:00
parent fc87e8acff
commit 6a757bbbbd
2 changed files with 14 additions and 7 deletions

View File

@@ -1065,11 +1065,13 @@ class IptablesFirewallTestCase(test.TestCase):
instance = db.instance_get(self.context, inst_id) instance = db.instance_get(self.context, inst_id)
ip = '10.11.12.13' ip = '10.11.12.13'
network_ref = db.project_get_network(self.context, 'fake') network_ref = db.project_get_networks(self.context, 'fake')[0]
fixed_ip = {'address': ip, 'network_id': network_ref['id']} fixed_ip = {'address': ip, 'network_id': network_ref['id']}
db.fixed_ip_create(admin_ctxt, fixed_ip) db.fixed_ip_create(admin_ctxt, fixed_ip)
db.fixed_ip_update(admin_ctxt, ip, {'allocated': True, db.fixed_ip_update(admin_ctxt, ip, {'allocated': True,
'instance_id': inst_id}) 'instance_id': inst_id})
_setup_networking(inst_id, ip)
self.fw.setup_basic_filtering(instance) self.fw.setup_basic_filtering(instance)
self.fw.prepare_instance_filter(instance) self.fw.prepare_instance_filter(instance)
self.fw.apply_instance_filter(instance) self.fw.apply_instance_filter(instance)
@@ -1086,7 +1088,7 @@ class IptablesFirewallTestCase(test.TestCase):
instance_ref = self._create_instance_ref() instance_ref = self._create_instance_ref()
nw_info = _create_network_info(1) nw_info = _create_network_info(1)
ip = '10.11.12.13' ip = '10.11.12.13'
network_ref = db.project_get_network(self.context, 'fake') network_ref = db.project_get_networks(self.context, 'fake')[0]
admin_ctxt = context.get_admin_context() admin_ctxt = context.get_admin_context()
fixed_ip = {'address': ip, 'network_id': network_ref['id']} fixed_ip = {'address': ip, 'network_id': network_ref['id']}
db.fixed_ip_create(admin_ctxt, fixed_ip) db.fixed_ip_create(admin_ctxt, fixed_ip)
@@ -1288,7 +1290,7 @@ class NWFilterTestCase(test.TestCase):
db.fixed_ip_update(admin_ctxt, ip, {'allocated': True, db.fixed_ip_update(admin_ctxt, ip, {'allocated': True,
'instance_id': inst_id}) 'instance_id': inst_id})
self._setup_networking(instance_ref['id'], ip=ip) _setup_networking(instance_ref['id'], ip=ip)
def _ensure_all_called(): def _ensure_all_called():
instance_filter = 'nova-instance-%s-%s' % (instance_ref['name'], instance_filter = 'nova-instance-%s-%s' % (instance_ref['name'],
@@ -1341,7 +1343,7 @@ class NWFilterTestCase(test.TestCase):
instance = db.instance_get(self.context, inst_id) instance = db.instance_get(self.context, inst_id)
ip = '10.11.12.13' ip = '10.11.12.13'
network_ref = db.project_get_network(self.context, 'fake') network_ref = db.project_get_networks(self.context, 'fake')[0]
fixed_ip = {'address': ip, 'network_id': network_ref['id']} fixed_ip = {'address': ip, 'network_id': network_ref['id']}
db.fixed_ip_create(admin_ctxt, fixed_ip) db.fixed_ip_create(admin_ctxt, fixed_ip)
db.fixed_ip_update(admin_ctxt, ip, {'allocated': True, db.fixed_ip_update(admin_ctxt, ip, {'allocated': True,
@@ -1350,6 +1352,7 @@ class NWFilterTestCase(test.TestCase):
self.fw.prepare_instance_filter(instance) self.fw.prepare_instance_filter(instance)
self.fw.apply_instance_filter(instance) self.fw.apply_instance_filter(instance)
original_filter_count = len(fakefilter.filters) original_filter_count = len(fakefilter.filters)
raise Exception(original_filter_count)
self.fw.unfilter_instance(instance) self.fw.unfilter_instance(instance)
# should undefine 2 filters: instance and instance-secgroup # should undefine 2 filters: instance and instance-secgroup

View File

@@ -25,6 +25,7 @@ import netaddr
from nova import context from nova import context
from nova import db from nova import db
from nova import exception
from nova import flags from nova import flags
from nova import ipv6 from nova import ipv6
from nova import utils from nova import utils
@@ -55,10 +56,13 @@ def get_network_info(instance):
# we should cache network_info # we should cache network_info
admin_context = context.get_admin_context() admin_context = context.get_admin_context()
try:
fixed_ips = db.fixed_ip_get_by_instance(admin_context, instance['id']) fixed_ips = db.fixed_ip_get_by_instance(admin_context, instance['id'])
except exception.FixedIpNotFoundForInstance:
fixed_ips = []
vifs = db.virtual_interface_get_by_instance(admin_context, instance['id']) vifs = db.virtual_interface_get_by_instance(admin_context, instance['id'])
networks = db.network_get_all_by_instance(admin_context,
instance['id'])
flavor = db.instance_type_get(admin_context, flavor = db.instance_type_get(admin_context,
instance['instance_type_id']) instance['instance_type_id'])
network_info = [] network_info = []