[python-oca] create new repo

2023-11-16 22:29:06 +09:00 · 2023-11-16 22:29:06 +09:00 · b202172a30
commit b202172a30
78 changed files with 4703 additions and 0 deletions
--- a/oca/tests/test_template.py
+++ b/oca/tests/test_template.py
@ -0,0 +1,85 @@
+# -*- coding: UTF-8 -*-
+import os
+import unittest
+
+from mock import Mock
+
+import oca
+
+
+class TestVmTemplate(unittest.TestCase):
+    def setUp(self):
+        self.client = oca.Client('test:test')
+        self.xml = open(os.path.join(os.path.dirname(oca.__file__),
+                                     'tests/fixtures/template.xml')).read()
+
+    def test_allocate(self):
+        self.client.call = Mock(return_value=3)
+        assert oca.VmTemplate.allocate(self.client, 'name=a') == 3
+
+    def test_delete(self):
+        self.client.call = Mock()
+        template = oca.VmTemplate(self.xml, self.client)
+        template.delete()
+        self.client.call.assert_called_once_with('template.delete', '1')
+
+    def test_update(self):
+        self.client.call = Mock()
+        template = oca.VmTemplate(self.xml, self.client)
+        template.update('name=b')
+        self.client.call.assert_called_once_with('template.update',
+                                                 '1', 'name=b', 0)
+
+    def test_publish(self):
+        self.client.call = Mock()
+        template = oca.VmTemplate(self.xml, self.client)
+        template.publish()
+        self.client.call.assert_called_once_with('template.publish',
+                                                 '1', True)
+
+    def test_unpublish(self):
+        self.client.call = Mock()
+        template = oca.VmTemplate(self.xml, self.client)
+        template.unpublish()
+        self.client.call.assert_called_once_with('template.publish',
+                                                 '1', False)
+
+    def test_chown(self):
+        self.client.call = Mock()
+        template = oca.VmTemplate(self.xml, self.client)
+        template.chown(2, 3)
+        self.client.call.assert_called_once_with('template.chown',
+                                                 '1', 2, 3)
+
+    def test_instantiate_with_default_name(self):
+        self.client.call = Mock(return_value=4)
+        template = oca.VmTemplate(self.xml, self.client)
+        assert template.instantiate() == 4
+        self.client.call.assert_called_once_with('template.instantiate',
+                                                 '1', '', False, '')
+
+    def test_instantiate_with_custom_name(self):
+        self.client.call = Mock(return_value=5)
+        template = oca.VmTemplate(self.xml, self.client)
+        assert template.instantiate('asd') == 5
+        self.client.call.assert_called_once_with('template.instantiate',
+                                                 '1', 'asd', False, '')
+
+    def test_instantiate_with_default_name_and_context(self):
+        self.client.call = Mock(return_value=6)
+        template = oca.VmTemplate(self.xml, self.client)
+        assert template.instantiate('', False, 'VCPU=4') == 6
+        self.client.call.assert_called_once_with('template.instantiate',
+                                                 '1', '', False, 'VCPU=4')
+
+    def test_instantiate_with_custom_name_and_context(self):
+        self.client.call = Mock(return_value=7)
+        template = oca.VmTemplate(self.xml, self.client)
+        assert template.instantiate('asd', False, 'VCPU=4') == 7
+        self.client.call.assert_called_once_with('template.instantiate',
+                                                 '1', 'asd', False, 'VCPU=4')
+
+    def test_repr(self):
+        self.client.call = Mock()
+        template = oca.VmTemplate(self.xml, self.client)
+        assert repr(template) == '<oca.VmTemplate("test1")>'