2:15 p.m.
From: Daniel Henrique Barboza <danielhb(a)linux.vnet.ibm.com>
This patch added new tests in test_api.py and test_utils.py
for the new /config/plugins API.
Signed-off-by: Daniel Henrique Barboza <danielhb(a)linux.vnet.ibm.com>
---
tests/test_api.py | 43 ++++++++++++++++++++++++++++++++++++++++++-
tests/test_utils.py | 51 ++++++++++++++++++++++++++++++++++++++++++++++++++-
2 files changed, 92 insertions(+), 2 deletions(-)
diff --git a/tests/test_api.py b/tests/test_api.py
index 23c263d..1358151 100644
--- a/tests/test_api.py
+++ b/tests/test_api.py
@@ -1,7 +1,7 @@
#
# Project Wok
#
-# Copyright IBM Corp, 2016
+# Copyright IBM Corp, 2016-2017
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
@@ -25,6 +25,8 @@ import utils
from functools import partial
from wok.asynctask import AsyncTask
+from wok.utils import set_plugin_state
+from wok.rollbackcontext import RollbackContext
test_server = None
model = None
@@ -53,6 +55,45 @@ class APITests(unittest.TestCase):
"server_root"]
self.assertEquals(sorted(keys), sorted(conf.keys()))
+ def test_config_plugins(self):
+ resp = self.request('/config/plugins')
+ self.assertEquals(200, resp.status)
+
+ plugins = json.loads(resp.read())
+ if len(plugins) == 0:
+ return
+
+ plugin = plugins[0]
+ plugin_state = plugin.get('enabled')
+ plugin_name = plugin.get('name')
+
+ with RollbackContext() as rollback:
+ rollback.prependDefer(set_plugin_state, plugin_name,
+ plugin_state)
+
+ resp = self.request('/config/plugins/%s' % plugin_name)
+ self.assertEquals(200, resp.status)
+
+ resp = self.request('/config/plugins/%s/enable' % plugin_name,
+ '{}', 'POST')
+ self.assertEquals(200, resp.status)
+
+ resp = self.request('/config/plugins')
+ self.assertEquals(200, resp.status)
+ plugins = json.loads(resp.read())
+ plugin = plugins[0]
+ self.assertEqual(plugin.get('enabled'), True)
+
+ resp = self.request('/config/plugins/%s/disable' % plugin_name,
+ '{}', 'POST')
+ self.assertEquals(200, resp.status)
+
+ resp = self.request('/config/plugins')
+ self.assertEquals(200, resp.status)
+ plugins = json.loads(resp.read())
+ plugin = plugins[0]
+ self.assertEqual(plugin.get('enabled'), False)
+
def test_user_log(self):
# Login and logout to make sure there there are entries in user log
hdrs = {'AUTHORIZATION': '',
diff --git a/tests/test_utils.py b/tests/test_utils.py
index e7fd264..c0e1009 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -19,10 +19,14 @@
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+import mock
+import os
+import tempfile
import unittest
from wok.exception import InvalidParameter
-from wok.utils import convert_data_size
+from wok.rollbackcontext import RollbackContext
+from wok.utils import convert_data_size, set_plugin_state
class UtilsTests(unittest.TestCase):
@@ -69,3 +73,48 @@ class UtilsTests(unittest.TestCase):
for d in success_data:
self.assertEquals(d['got'], d['want'])
+
+ def _get_fake_config_file_content(self, enable=True):
+ return """\
+[wok]
+# Enable plugin on Wok server (values: True|False)
+enable = %s
+
+[fakeplugin]
+# Yet another comment on this config file
+very_interesting_option = True
+""" % str(enable)
+
+ def _create_fake_config_file(self):
+ _, tmp_file_name = tempfile.mkstemp(suffix='.conf')
+
+ config_contents = self._get_fake_config_file_content()
+ with open(tmp_file_name, 'w') as f:
+ f.writelines(config_contents)
+
+ return tmp_file_name
+
+ @mock.patch('wok.utils.get_plugin_config_file')
+ def test_set_plugin_state(self, mock_config_file):
+ with RollbackContext() as rollback:
+
+ config_file_name = self._create_fake_config_file()
+ rollback.prependDefer(os.remove, config_file_name)
+
+ mock_config_file.return_value = config_file_name
+
+ set_plugin_state('pluginA', False)
+ with open(config_file_name, 'r') as f:
+ updated_conf = f.read()
+ self.assertEqual(
+ updated_conf,
+ self._get_fake_config_file_content(enable=False)
+ )
+
+ set_plugin_state('pluginA', True)
+ with open(config_file_name, 'r') as f:
+ updated_conf = f.read()
+ self.assertEqual(
+ updated_conf,
+ self._get_fake_config_file_content(enable=True)
+ )
--
2.7.4