add tests for utils module

onedns/tests/test_shell.py

@@ -0,0 +1,33 @@
+import sys
+
+import mock
+
+from testfixtures import LogCapture
+
+IPY = mock.MagicMock()
+IPY.embed = mock.MagicMock()
+
+IPY_MODULES = {
+  'IPython': IPY,
+}
+
+
+@mock.patch.dict('sys.modules', IPY_MODULES)
+def test_with_ipython():
+    from onedns import utils
+    ns = dict(test=True)
+    utils.shell(local_ns=ns)
+    IPY.embed.assert_called_once_with(user_ns=ns)
+
+
+@mock.patch.object(sys, 'path', [])
+def test_without_ipython():
+    from onedns import utils
+    with LogCapture() as log:
+        utils.shell()
+    log.check(
+        ('onedns', 'ERROR',
+         'Unable to load IPython:\n\nNo module named IPython\n'),
+        ('onedns', 'ERROR',
+         'Please check that IPython is installed and working.'),
+    )

onedns/tests/test_utils.py

@@ -0,0 +1,24 @@
+import os
+
+import mock
+
+from onedns import utils
+
+
+ONE_XMLRPC = 'https://controller:2633/RPC2'
+
+
+def test_get_kwargs_from_dict():
+    d = dict(ONE_XMLRPC=ONE_XMLRPC)
+    kwargs = utils.get_kwargs_from_dict(d, prefix='ONE_')
+    assert kwargs['XMLRPC'] == ONE_XMLRPC
+    kwargs = utils.get_kwargs_from_dict(d, prefix='ONE_', lower=True)
+    assert kwargs['xmlrpc'] == ONE_XMLRPC
+
+
+@mock.patch.dict(os.environ, {'ONE_XMLRPC': ONE_XMLRPC})
+def test_get_kwargs_from_env():
+    kwargs = utils.get_kwargs_from_env(prefix='ONE_')
+    assert kwargs['XMLRPC'] == ONE_XMLRPC
+    kwargs = utils.get_kwargs_from_env(prefix='ONE_', lower=True)
+    assert kwargs['xmlrpc'] == ONE_XMLRPC