1 files changed, 145 insertions, 0 deletions
diff --git a/src/test/common/test_context.cc b/src/test/common/test_context.cc
new file mode 100644
index 000000000..2c846a9ae
--- /dev/null
+++ b/src/test/common/test_context.cc
@@ -0,0 +1,145 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2014 Cloudwatt <libre.licensing@cloudwatt.com>
+ *
+ * Author: Loic Dachary <loic@dachary.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Library Public License as published by
+ * the Free Software Foundation; either version 2, or (at your option)
+ * any later version.
+ *
+ * This program 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 Library Public License for more details.
+ *
+ *
+ */
+#include "gtest/gtest.h"
+#include "include/types.h"
+#include "include/msgr.h"
+#include "common/ceph_context.h"
+#include "common/config_proxy.h"
+#include "log/Log.h"
+
+using namespace std;
+
+TEST(CephContext, do_command)
+{
+  CephContext *cct = (new CephContext(CEPH_ENTITY_TYPE_CLIENT))->get();
+
+  cct->_conf->cluster = "ceph";
+
+  string key("key");
+  string value("value");
+  cct->_conf.set_val(key.c_str(), value.c_str());
+  cmdmap_t cmdmap;
+  cmdmap["var"] = key;
+
+  {
+    stringstream ss;
+    bufferlist out;
+    std::unique_ptr<Formatter> f{Formatter::create_unique("xml", "xml")};
+    cct->do_command("config get", cmdmap, f.get(), ss, &out);
+    f->flush(out);
+    string s(out.c_str(), out.length());
+    EXPECT_EQ("<config_get><key>" + value + "</key></config_get>", s);
+  }
+
+  {
+    stringstream ss;
+    bufferlist out;
+    cmdmap_t bad_cmdmap; // no 'var' field
+    std::unique_ptr<Formatter> f{Formatter::create_unique("xml", "xml")};
+    int r = cct->do_command("config get", bad_cmdmap, f.get(), ss, &out);
+    if (r >= 0) {
+      f->flush(out);
+    }
+    string s(out.c_str(), out.length());
+    EXPECT_EQ(-EINVAL, r);
+    EXPECT_EQ("", s);
+    EXPECT_EQ("", ss.str()); // no error string :/
+  }
+  {
+    stringstream ss;
+    bufferlist out;
+    cmdmap_t bad_cmdmap;
+    bad_cmdmap["var"] = string("doesnotexist123");
+    std::unique_ptr<Formatter> f{Formatter::create_unique("xml", "xml")};
+    int r = cct->do_command("config help", bad_cmdmap, f.get(), ss, &out);
+    if (r >= 0) {
+      f->flush(out);
+    }
+    string s(out.c_str(), out.length());
+    EXPECT_EQ(-ENOENT, r);
+    EXPECT_EQ("", s);
+    EXPECT_EQ("Setting not found: 'doesnotexist123'", ss.str());
+  }
+
+  {
+    stringstream ss;
+    bufferlist out;
+    std::unique_ptr<Formatter> f{Formatter::create_unique("xml", "xml")};
+    cct->do_command("config diff get", cmdmap, f.get(), ss, &out);
+    f->flush(out);
+    string s(out.c_str(), out.length());
+    EXPECT_EQ("<config_diff_get><diff><key><default></default><override>" + value + "</override><final>value</final></key><rbd_default_features><default>61</default><final>61</final></rbd_default_features><rbd_qos_exclude_ops><default>0</default><final>0</final></rbd_qos_exclude_ops></diff></config_diff_get>", s);
+  }
+  cct->put();
+}
+
+TEST(CephContext, experimental_features)
+{
+  CephContext *cct = (new CephContext(CEPH_ENTITY_TYPE_CLIENT))->get();
+
+  cct->_conf->cluster = "ceph";
+
+  ASSERT_FALSE(cct->check_experimental_feature_enabled("foo"));
+  ASSERT_FALSE(cct->check_experimental_feature_enabled("bar"));
+  ASSERT_FALSE(cct->check_experimental_feature_enabled("baz"));
+
+  cct->_conf.set_val("enable_experimental_unrecoverable_data_corrupting_features",
+		      "foo,bar");
+  cct->_conf.apply_changes(&cout);
+  ASSERT_TRUE(cct->check_experimental_feature_enabled("foo"));
+  ASSERT_TRUE(cct->check_experimental_feature_enabled("bar"));
+  ASSERT_FALSE(cct->check_experimental_feature_enabled("baz"));
+
+  cct->_conf.set_val("enable_experimental_unrecoverable_data_corrupting_features",
+		      "foo bar");
+  cct->_conf.apply_changes(&cout);
+  ASSERT_TRUE(cct->check_experimental_feature_enabled("foo"));
+  ASSERT_TRUE(cct->check_experimental_feature_enabled("bar"));
+  ASSERT_FALSE(cct->check_experimental_feature_enabled("baz"));
+
+  cct->_conf.set_val("enable_experimental_unrecoverable_data_corrupting_features",
+		      "baz foo");
+  cct->_conf.apply_changes(&cout);
+  ASSERT_TRUE(cct->check_experimental_feature_enabled("foo"));
+  ASSERT_FALSE(cct->check_experimental_feature_enabled("bar"));
+  ASSERT_TRUE(cct->check_experimental_feature_enabled("baz"));
+
+  cct->_conf.set_val("enable_experimental_unrecoverable_data_corrupting_features",
+		      "*");
+  cct->_conf.apply_changes(&cout);
+  ASSERT_TRUE(cct->check_experimental_feature_enabled("foo"));
+  ASSERT_TRUE(cct->check_experimental_feature_enabled("bar"));
+  ASSERT_TRUE(cct->check_experimental_feature_enabled("baz"));
+
+  cct->_log->flush();
+}
+
+/*
+ * Local Variables:
+ * compile-command: "cd ../.. ;
+ *   make unittest_context &&
+ *    valgrind \
+ *    --max-stackframe=20000000 --tool=memcheck \
+ *   ./unittest_context # --gtest_filter=CephContext.*
+ * "
+ * End:
+ */