diff --git a/nova/tests/unit/virt/libvirt/fakelibvirt.py b/nova/tests/unit/virt/libvirt/fakelibvirt.py index 306f1150b8e2..c64ad32ef643 100644 --- a/nova/tests/unit/virt/libvirt/fakelibvirt.py +++ b/nova/tests/unit/virt/libvirt/fakelibvirt.py @@ -776,6 +776,9 @@ class Domain(object): def migrateSetMaxDowntime(self, downtime): pass + def migrateSetMaxSpeed(self, bandwidth): + pass + def attachDevice(self, xml): disk_info = _parse_disk_info(etree.fromstring(xml)) disk_info['_attached'] = True diff --git a/nova/tests/unit/virt/libvirt/test_guest.py b/nova/tests/unit/virt/libvirt/test_guest.py index 245e0f39d58c..18367fd2456b 100644 --- a/nova/tests/unit/virt/libvirt/test_guest.py +++ b/nova/tests/unit/virt/libvirt/test_guest.py @@ -654,6 +654,10 @@ class GuestTestCase(test.NoDBTestCase): self.guest.migrate_configure_max_downtime(1000) self.domain.migrateSetMaxDowntime.assert_called_once_with(1000) + def test_migrate_configure_max_speed(self): + self.guest.migrate_configure_max_speed(1000) + self.domain.migrateSetMaxSpeed.assert_called_once_with(1000) + class GuestBlockTestCase(test.NoDBTestCase): diff --git a/nova/virt/libvirt/guest.py b/nova/virt/libvirt/guest.py index f533edf43112..d66284d63998 100644 --- a/nova/virt/libvirt/guest.py +++ b/nova/virt/libvirt/guest.py @@ -663,6 +663,13 @@ class Guest(object): """ self._domain.migrateSetMaxDowntime(mstime) + def migrate_configure_max_speed(self, bandwidth): + """The maximum bandwidth that will be used to do migration + + :param bw: Bandwidth in MiB/s + """ + self._domain.migrateSetMaxSpeed(bandwidth) + def migrate_start_postcopy(self): """Switch running live migration to post-copy mode""" self._domain.migrateStartPostCopy()