1 files changed, 124 insertions, 0 deletions
diff --git a/juniper/src/test/testenv.cpp b/juniper/src/test/testenv.cpp
new file mode 100644
index 00000000000..40f35072948
--- /dev/null
+++ b/juniper/src/test/testenv.cpp
@@ -0,0 +1,124 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+/* Setup and parameter parsing for static Juniper environment to reuse
+ * within test framework
+ */
+
+#include <vespa/fastos/fastos.h>
+#include <vespa/log/log.h>
+LOG_SETUP("");
+#include "testenv.h"
+#include <vespa/juniper/propreader.h>
+
+
+namespace juniper
+{
+
+bool color_highlight = false;
+// Easy access in tests..
+Config* TestConfig;
+Juniper * _Juniper;
+
+
+TestEnv::TestEnv(FastOS_Application* app, const char* propfile) :
+    _props(), _config(), _juniper(), _wordFolder()
+{
+    char c;
+    const char* oarg = NULL;
+    int oind = 1;
+
+    while ((c = app->GetOpt("d:hcm:", oarg, oind)) != EOF)
+    {
+        switch (c)
+        {
+	case 'd':
+#ifdef FASTOS_DEBUG
+            debug_level = strtol(oarg, NULL, 0);
+#else
+            fprintf(stderr, "This version of Juniper compiled without debug\n");
+#endif
+            break;
+	case 'c':
+            color_highlight = true;
+            break;
+	case 'm':
+            // option handled by test framework
+            break;
+	case 'h':
+	default:
+            Usage(app->_argv[0]);
+            return;
+        }
+    }
+
+    int expected_args = 0;
+
+    if (app->_argc - oind < expected_args)
+    {
+        Usage(app->_argv[0]);
+        return;
+    }
+
+    _props.reset(new PropReader(propfile));
+
+    if (color_highlight)
+    {
+        _props->UpdateProperty("juniper.dynsum.highlight_on", "\\1b[1;31m");
+        _props->UpdateProperty("juniper.dynsum.highlight_off", "\\1b[0m");
+    }
+
+    _juniper.reset(new Juniper(_props.get(), &_wordFolder));
+    _Juniper = _juniper.get();
+    _config = _juniper->CreateConfig();
+    TestConfig = _config.get();
+}
+
+TestEnv::~TestEnv()
+{
+}
+
+void TestEnv::Usage(char* s)
+{
+    fprintf(stderr, "Usage: %s [options]\n", s);
+    fprintf(stderr, "Available options:\n");
+    fprintf(stderr, "  -d<debugmask>: Turn on debugging\n");
+    fprintf(stderr, "  -h: This help\n");
+}
+
+
+TestQuery::TestQuery(const char* qexp, const char* options) :
+    _qparser(qexp),
+    _qhandle(_qparser, options, _Juniper->getModifier())
+{ }
+
+
+PropertyMap::PropertyMap()
+    : _map()
+{
+}
+
+
+PropertyMap::~PropertyMap()
+{
+}
+
+
+PropertyMap &
+PropertyMap::set(const char *name, const char *value)
+{
+    _map[std::string(name)] = std::string(value);
+    return *this;
+}
+
+
+const char *
+PropertyMap::GetProperty(const char* name, const char* def)
+{
+    std::map<std::string, std::string>::iterator res = _map.find(std::string(name));
+    if (res != _map.end()) {
+        return res->second.c_str();
+    }
+    return def;
+}
+
+
+} // end namespace juniper