Add support for the image chmod method (python-oca#49)

mguezuraga · sysradium · commit 469abd871553 · 2017-02-18T13:05:55.000+02:00
diff --git a/oca/image.py b/oca/image.py
@@ -10,6 +10,7 @@ class Image(PoolElement):
         'update': 'image.update',
         'enable': 'image.enable',
         'publish': 'image.publish',
+        'chmod': 'image.chmod',
         'chown': 'image.chown',
         'persistent': 'image.persistent',
         'clone': 'image.clone',
@@ -152,6 +153,34 @@ def chown(self, uid, gid):
         """
         self.client.call(self.METHODS['chown'], self.id, uid, gid)
 
+    def chmod(self, owner_u, owner_m, owner_a, group_u, group_m, group_a, other_u, other_m, other_a):
+        """
+        Changes the permission bits
+
+        Arguments
+
+        ``owner_u``
+            User USE bit. Set to -1 to leave current value
+        ``owner_m``
+            User MANAGE bit. Set to -1 to leave current value
+        ``owner_a``
+            User ADMIN bit. Set to -1 to leave current value
+        ``group_u``
+            Group USE bit. Set to -1 to leave current value
+        ``group_m``
+            Group MANAGE bit. Set to -1 to leave current value
+        ``group_a``
+            Group ADMIN bit. Set to -1 to leave current value
+        ``other_u``
+            Other USE bit. Set to -1 to leave current value
+        ``other_m``
+            Other MANAGE bit. Set to -1 to leave current value
+        ``other_a``
+            Other ADMIN bit. Set to -1 to leave current value
+        """
+        self.client.call(self.METHODS['chmod'], self.id, owner_u, owner_m, owner_a, group_u, group_m, group_a, other_u,
+                         other_m, other_a)
+
     def clone(self, name='', datastore_id=-1):
         """
         Creates a clone of an image
diff --git a/oca/tests/test_image.py b/oca/tests/test_image.py
@@ -100,3 +100,10 @@ def test_chown(self):
         h.chown(10, 10)
         self.client.call.assert_called_once_with('image.chown',
                                                  '1', 10, 10)
+
+    def test_chmod(self):
+        self.client.call = Mock(return_value='')
+        h = oca.Image(self.xml, self.client)
+        h.chmod(1, 0, 0, -1, -1, -1, -1, -1, -1)
+        self.client.call.assert_called_once_with('image.chmod',
+                                                 '1', 1, 0, 0, -1, -1, -1, -1, -1, -1)
