[python-oca] create new repo

oca/tests/test_client.py

@@ -0,0 +1,107 @@
+# -*- coding: UTF-8 -*-
+import os
+import socket
+import unittest
+
+from mock import Mock
+from nose.tools import raises
+
+import oca
+
+
+class TestClient(unittest.TestCase):
+    def setUp(self):
+        try:
+            del os.environ["ONE_AUTH"]
+        except KeyError:
+            pass
+
+    def test_secret(self):
+        c = oca.Client('test:test')
+        assert c.one_auth == 'test:test'
+
+    def test_one_auth(self):
+        os.environ["ONE_AUTH"] = os.path.join(os.path.dirname(oca.__file__),
+                                              'tests/fixtures/one_auth')
+        try:
+            c = oca.Client()
+            secret = 'test:test'
+            assert c.one_auth == secret
+        finally:
+            os.environ["ONE_AUTH"] = ''
+
+    def test_default_user_path(self):
+        os.environ["ONE_AUTH"] = os.path.join(os.path.dirname(oca.__file__),
+                                              'tests/fixtures/one_auth')
+        c = oca.Client()
+        assert c.one_auth == 'test:test'
+
+    @raises(oca.OpenNebulaException)
+    def test_wrong_default_user_path(self):
+        oca.Client.DEFAULT_ONE_AUTH = '/ad/ads/a/das/d/sad/sad/sa/d/one_auth'
+        c = oca.Client()
+
+    @raises(oca.OpenNebulaException)
+    def test_invalid_secret(self):
+        os.environ["ONE_AUTH"] = os.path.join(os.path.dirname(oca.__file__),
+                                              'tests/fixtures/one_auth')
+        c = oca.Client('testtest')
+
+    def test_addres(self):
+        c = oca.Client('test:test', "http://8.8.8.8:2633/RPC2")
+        assert c.one_address == "http://8.8.8.8:2633/RPC2"
+
+    def test_one_xml_rpc(self):
+        os.environ["ONE_AUTH"] = os.path.join(os.path.dirname(oca.__file__),
+                                              'tests/fixtures/one_auth')
+        os.environ["ONE_XMLRPC"] = "http://8.8.8.8:2633/RPC2"
+        try:
+            c = oca.Client()
+            assert c.one_address == "http://8.8.8.8:2633/RPC2"
+        finally:
+            os.environ["ONE_XMLRPC"] = ''
+
+    def test_defaul_xmlrpc(self):
+        c = oca.Client('test:test')
+        assert c.one_address == oca.Client.DEFAULT_ONE_ADDRESS
+
+    def test_version(self):
+        c = oca.Client('test:test')
+        assert c.one_version is None
+        c.call = Mock(return_value='1.0.0')
+        assert c.version() == '1.0.0'
+        c.call.assert_called_once_with('system.version')
+        assert c.one_version == '1.0.0'
+
+    def test_return_two_values_call(self):
+        c = oca.Client('test:test')
+        c.server.one = Mock()
+        c.server.one.test_method = lambda x: [True, '2', 0]
+        assert c.call('test_method') == '2'
+
+    def test_return_one_value_call(self):
+        c = oca.Client('test:test')
+        c.server.one = Mock()
+        c.server.one.test_method = lambda x: [True, '', 0]
+        assert c.call('test_method') == ''
+
+    @raises(oca.OpenNebulaException)
+    def test_retrurn_error_code_0_call(self):
+        c = oca.Client('test:test')
+        c.server.one = Mock()
+        c.server.one.test_method = lambda x: [False, '2', 1]
+        c.call('test_method')
+
+    @raises(oca.OpenNebulaException)
+    def test_invalid_call(self):
+        c = oca.Client('test:test')
+        c.server.one = Mock()
+        c.server.one.test_method = lambda x: [False, '2']
+        c.call('test_method')
+
+    @raises(socket.error)
+    def test_connection_error(self):
+        c = oca.Client('test:test')
+        c.server.one = Mock()
+        c.server.one.test_method = Mock(side_effect=socket.error(1))
+        c.call('test_method')