fix tests

This commit is contained in:
Vishvananda Ishaya
2010-10-12 19:27:48 -07:00
parent a03d53436f
commit 62dba2f6b3
2 changed files with 12 additions and 6 deletions

View File

@@ -58,7 +58,8 @@ class NetworkTestCase(test.TrialTestCase):
user_context = context.APIRequestContext(project=self.projects[i], user_context = context.APIRequestContext(project=self.projects[i],
user=self.user) user=self.user)
network_ref = self.network.get_network(user_context) network_ref = self.network.get_network(user_context)
self.network.set_network_host(user_context, network_ref['id']) self.network.set_network_host(context.get_admin_context(),
network_ref['id'])
instance_ref = self._create_instance(0) instance_ref = self._create_instance(0)
self.instance_id = instance_ref['id'] self.instance_id = instance_ref['id']
instance_ref = self._create_instance(1) instance_ref = self._create_instance(1)

View File

@@ -20,21 +20,22 @@ from xml.dom.minidom import parseString as xml_to_dom
from nova import db from nova import db
from nova import flags from nova import flags
from nova import test from nova import test
from nova import utils
from nova.api import context from nova.api import context
from nova.api.ec2 import cloud from nova.api.ec2 import cloud
from nova.auth import manager from nova.auth import manager
# Needed to get FLAGS.instances_path defined:
from nova.compute import manager as compute_manager
from nova.virt import libvirt_conn from nova.virt import libvirt_conn
FLAGS = flags.FLAGS FLAGS = flags.FLAGS
flags.DECLARE('instances_path', 'nova.compute.manager')
class LibvirtConnTestCase(test.TrialTestCase): class LibvirtConnTestCase(test.TrialTestCase):
def setUp(self): def setUp(self):
super(LibvirtConnTestCase, self).setUp()
self.manager = manager.AuthManager() self.manager = manager.AuthManager()
self.user = self.manager.create_user('fake', 'fake', 'fake', admin=True) self.user = self.manager.create_user('fake', 'fake', 'fake', admin=True)
self.project = self.manager.create_project('fake', 'fake', 'fake') self.project = self.manager.create_project('fake', 'fake', 'fake')
self.network = utils.import_object(FLAGS.network_manager)
FLAGS.instances_path = '' FLAGS.instances_path = ''
def test_get_uri_and_template(self): def test_get_uri_and_template(self):
@@ -51,11 +52,15 @@ class LibvirtConnTestCase(test.TrialTestCase):
'instance_type' : 'm1.small'} 'instance_type' : 'm1.small'}
instance_ref = db.instance_create(None, instance) instance_ref = db.instance_create(None, instance)
network_ref = db.project_get_network(None, self.project.id) user_context = context.APIRequestContext(project=self.project,
user=self.user)
network_ref = self.network.get_network(user_context)
self.network.set_network_host(context.get_admin_context(),
network_ref['id'])
fixed_ip = { 'address' : ip, fixed_ip = { 'address' : ip,
'network_id' : network_ref['id'] } 'network_id' : network_ref['id'] }
fixed_ip_ref = db.fixed_ip_create(None, fixed_ip) fixed_ip_ref = db.fixed_ip_create(None, fixed_ip)
db.fixed_ip_update(None, ip, { 'allocated' : True, db.fixed_ip_update(None, ip, { 'allocated' : True,
'instance_id' : instance_ref['id'] }) 'instance_id' : instance_ref['id'] })