tests/utils.py: contains utility functions, etc that will be used in test modules.

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)