@@ -6716,7 +6716,7 @@ def test_get_guest_config_with_video_driver_vram(self):
67166716
67176717 self.assertEqual(cfg.devices[4].type, "spice")
67186718 self.assertEqual(cfg.devices[5].type, "qxl")
6719- self.assertEqual(cfg.devices[5].vram, 64 * units.Mi / units.Ki )
6719+ self.assertEqual(cfg.devices[5].vram, 65536 )
67206720
67216721 def _test_add_video_driver(self, model):
67226722 self.flags(virt_type='kvm', group='libvirt')
@@ -6727,15 +6727,19 @@ def _test_add_video_driver(self, model):
67276727
67286728 drvr = libvirt_driver.LibvirtDriver(fake.FakeVirtAPI(), True)
67296729 guest = vconfig.LibvirtConfigGuest()
6730- instance_ref = objects.Instance(**self.test_instance)
6731- flavor = instance_ref.get_flavor( )
6730+ flavor = objects.Flavor(
6731+ extra_specs={'hw_video:ram_max_mb': '512'} )
67326732 image_meta = objects.ImageMeta.from_dict({
6733- 'properties': {'hw_video_model': model}})
6733+ 'properties': {
6734+ 'hw_video_model': model,
6735+ 'hw_video_ram': 8,
6736+ },
6737+ })
67346738
67356739 self.assertTrue(drvr._guest_add_video_device(guest))
6736- video = drvr._add_video_driver(guest, image_meta,
6737- flavor)
6740+ video = drvr._add_video_driver(guest, image_meta, flavor)
67386741 self.assertEqual(model, video.type)
6742+ self.assertEqual(8192, video.vram) # should be in bytes
67396743
67406744 def test__add_video_driver(self):
67416745 self._test_add_video_driver('qxl')
0 commit comments