diff --git a/dragonflow/tests/fullstack/test_objects.py b/dragonflow/tests/fullstack/test_objects.py index 95aee2004..6c6e89be3 100644 --- a/dragonflow/tests/fullstack/test_objects.py +++ b/dragonflow/tests/fullstack/test_objects.py @@ -289,8 +289,16 @@ class VMTestObj(object): self.nova = clients.get_nova_client_from_cloud_config() def create(self, network=None, script=None, security_groups=None, - net_address=None): - image = self.nova.glance.find_image("cirros-0.4.0-x86_64-disk") + net_address=None, vm_image=None): + if vm_image: + image = self.nova.glance.find_image(vm_image) + else: + # As we are running in test environment, it would be safe to assume + # we only have one image, which is the image we want to use. + # Taking the first image is safe enough in this case + images = self.nova.glance.list() + assert images + image = images[0] self.parent.assertIsNotNone(image) flavor = self.nova.flavors.find(name="m1.tiny") self.parent.assertIsNotNone(flavor)