summaryrefslogtreecommitdiffstats
path: root/tests/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/utils.py')
-rw-r--r--tests/utils.py57
1 files changed, 57 insertions, 0 deletions
diff --git a/tests/utils.py b/tests/utils.py
new file mode 100644
index 0000000..2b644e5
--- /dev/null
+++ b/tests/utils.py
@@ -0,0 +1,57 @@
+# Copyright (C) 2015 Combox author(s). See AUTHORS.
+#
+# This file is part of Combox.
+#
+# Combox is free software: you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Combox is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Combox (see COPYING). If not, see
+# <http://www.gnu.org/licenses/>.
+
+import yaml
+
+from nose.tools import *
+from os import path, remove, rmdir
+
+from combox.config import (config_cb, get_secret, get_stdin)
+
+
+def get_input_func():
+ """Returns the input function
+
+ The input function is meant to be passed to the config_cb function
+ in the combox.config module.
+
+ """
+ config_dir = path.join('tests', 'test-config')
+ combox_dir = path.join('tests', 'files')
+ node_dir_0 = path.join('tests', 'shard-dir-0')
+ node_dir_1 = path.join('tests', 'shard-dir-1')
+
+ # sample config info.
+ config_info = ['testbox', combox_dir, config_dir, '2',
+ 'node-0', node_dir_0, '1024',
+ 'node-1', node_dir_1, '1024']
+
+ config_info_iter = iter(config_info)
+ input_func = lambda(x): next(config_info_iter)
+
+ return input_func
+
+
+def get_config():
+ """Constructs test config dict and returns it."""
+
+ config_dir = path.join('tests', 'test-config')
+ pass_func = lambda: 'topsecret'
+ input_func = get_input_func()
+
+ return config_cb(config_dir, pass_func, input_func, write=False)