[python-oca] create new repo

oca/tests/test_image.py

@@ -0,0 +1,109 @@
+# -*- coding: UTF-8 -*-
+import os
+import unittest
+
+from mock import Mock
+
+import oca
+import oca.pool
+
+IMAGE_TEMPLATE = """NAME          = "Ubuntu Desktop"
+PATH          = /home/cloud/images/ubuntu-desktop/disk.0
+PUBLIC        = YES
+DESCRIPTION   = "Ubuntu 10.04 desktop for students.\""""
+
+DEFAULT_IMG_DATASTORE = 1
+
+
+class TestImage(unittest.TestCase):
+    def setUp(self):
+        self.client = oca.Client('test:test')
+        self.xml = open(os.path.join(os.path.dirname(oca.__file__),
+                                     'tests/fixtures/image.xml')).read()
+
+    def test_allocate(self):
+        self.client.call = Mock(return_value=2)
+        assert oca.Image.allocate(self.client, IMAGE_TEMPLATE, DEFAULT_IMG_DATASTORE) == 2
+
+    def test_enable(self):
+        self.client.call = Mock(return_value='')
+        h = oca.Image(self.xml, self.client)
+        h.enable()
+        self.client.call.assert_called_once_with('image.enable', '1', True)
+
+    def test_disable(self):
+        self.client.call = Mock(return_value='')
+        h = oca.Image(self.xml, self.client)
+        h.disable()
+        self.client.call.assert_called_once_with('image.enable', '1', False)
+
+    def test_update(self):
+        self.client.call = Mock(return_value='')
+        h = oca.Image(self.xml, self.client)
+        new_content = 'DEV_PREFIX=hd\nNAME=Debian\nTYPE=OS'
+        h.update(new_content)
+        self.client.call.assert_called_once_with('image.update', '1',
+                                                 new_content)
+
+    def test_publish(self):
+        self.client.call = Mock(return_value='')
+        h = oca.Image(self.xml, self.client)
+        assert h.publish() is None
+
+    def test_unpublish(self):
+        self.client.call = Mock(return_value='')
+        h = oca.Image(self.xml, self.client)
+        assert h.unpublish() is None
+
+    def test_set_persistent(self):
+        self.client.call = Mock(return_value='')
+        h = oca.Image(self.xml, self.client)
+        h.set_persistent()
+        self.client.call.assert_called_once_with('image.persistent',
+                                                 '1', True)
+
+    def test_set_nonpersistent(self):
+        self.client.call = Mock(return_value='')
+        h = oca.Image(self.xml, self.client)
+        h.set_nonpersistent()
+        self.client.call.assert_called_once_with('image.persistent',
+                                                 '1', False)
+
+    def test_states(self):
+        for i in range(len(oca.Image.IMAGE_STATES)):
+            h = oca.Image('<IMAGE><ID>2</ID><STATE>%s</STATE></IMAGE>' % i,
+                          self.client)
+            assert h.str_state == oca.Image.IMAGE_STATES[i]
+            short_image_state = oca.Image.SHORT_IMAGE_STATES[oca.Image.IMAGE_STATES[i]]
+            assert h.short_state == short_image_state
+
+    def test_repr(self):
+        h = oca.Image(self.xml, self.client)
+        assert h.__repr__() == '<oca.Image("MATLAB install CD")>'
+
+    def test_types(self):
+        for i in range(len(oca.Image.IMAGE_TYPES)):
+            h = oca.Image('<IMAGE><ID>2</ID><TYPE>%s</TYPE></IMAGE>' % i,
+                          self.client)
+            assert h.str_type == oca.Image.IMAGE_TYPES[i]
+            short_image_type = oca.Image.SHORT_IMAGE_TYPES[oca.Image.IMAGE_TYPES[i]]
+            assert h.short_type == short_image_type
+
+    def test_template(self):
+        i = oca.Image(self.xml, self.client)
+        i._convert_types()
+        assert isinstance(i.template, oca.pool.Template)
+
+    def test_chown(self):
+        self.client.call = Mock(return_value='')
+        h = oca.Image(self.xml, self.client)
+        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)