add vm action methods for resched and unresched (python-oca#39)

zwopir · sysradium · commit 29358dde8965 · 2016-11-04T20:54:48.000+03:00
diff --git a/oca/tests/test_virtualmachine.py b/oca/tests/test_virtualmachine.py
@@ -90,8 +90,9 @@ def test_save_disk(self):
     def test_actions(self):
         oca.client = oca.Client('test:test')
         vm = oca.VirtualMachine(self.xml, self.client)
-        for action in ['shutdown', 'shutdown_hard', 'poweroff', 'poweroff_hard', 'hold', 'release', 'stop', 'cancel',
-                'suspend', 'resume', 'reboot', 'finalize', 'delete']:
+        for action in ['shutdown', 'shutdown_hard', 'poweroff', 'poweroff_hard',
+                       'hold', 'release', 'stop', 'cancel', 'suspend', 'resume',
+                       'reboot', 'finalize', 'delete', 'resched', 'unresched']:
             self.client.call = Mock(return_value='')
             getattr(vm, action)()
             if action in ('shutdown_hard', 'poweroff_hard', 'undeploy_hard'):
diff --git a/oca/vm.py b/oca/vm.py
@@ -306,6 +306,18 @@ def delete(self):
         '''
         self._action('delete')
 
+    def resched(self):
+        '''
+        Set the rescheduling flag of the VM.
+        '''
+        self._action('resched')
+
+    def unresched(self):
+        '''
+        Remove the rescheduling flag of the VM.
+        '''
+        self._action('unresched')
+
     def _action(self, action):
         self.client.call(self.METHODS['action'], action, self.id)
 
