--- /dev/null
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2015 CERN
+ *
+ * Author: Joaquim Rocha <joaquim.rocha@cern.ch>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ */
+
+#include <iostream>
+#include <errno.h>
+#include <set>
+#include <sstream>
+#include <string>
+
+#include "cls/numops/cls_numops_client.h"
+#include "gtest/gtest.h"
+#include "include/rados/librados.hpp"
+#include "test/librados/test.h"
+
+using namespace librados;
+
+TEST(ClsNumOps, Add) {
+ Rados cluster;
+ std::string pool_name = get_temp_pool_name();
+ ASSERT_EQ("", create_one_pool_pp(pool_name, cluster));
+ IoCtx ioctx;
+ cluster.ioctx_create(pool_name.c_str(), ioctx);
+
+ // exec numops add method with an empty bufferlist
+
+ bufferlist in, out;
+
+ ASSERT_EQ(-EINVAL, ioctx.exec("myobject", "numops", "add", in, out));
+
+ // add a number to a non-existing key
+
+ std::string key = "my-key";
+ double value_in = 0.5;
+
+ std::stringstream stream;
+ stream << value_in;
+
+ ASSERT_EQ(0, rados::cls::numops::add(&ioctx, "myobject", key, value_in));
+
+ // check that the omap entry was set and the value matches
+
+ std::set<std::string> keys;
+ std::map<std::string, bufferlist> omap;
+ keys.insert(key);
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ std::map<std::string, bufferlist>::iterator it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bufferlist bl = (*it).second;
+ std::string value_out(bl.c_str(), bl.length());
+
+ EXPECT_EQ(stream.str(), value_out);
+
+ // add another value to the existing one
+
+ double new_value_in = 3.001;
+
+ ASSERT_EQ(0, rados::cls::numops::add(&ioctx, "myobject", key, new_value_in));
+
+ // check that the omap entry's value matches
+
+ omap.clear();
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bl = (*it).second;
+ value_out.assign(bl.c_str(), bl.length());
+
+ stream.str("");
+ stream << (value_in + new_value_in);
+
+ EXPECT_EQ(stream.str(), value_out);
+
+ // set the omap entry with some non-numeric value
+
+ omap.clear();
+
+ std::string non_numeric_value("some-non-numeric-text");
+ omap[key].append(non_numeric_value);
+
+ ASSERT_EQ(0, ioctx.omap_set("myobject", omap));
+
+ // check that adding a number does not succeed
+
+ omap.clear();
+
+ ASSERT_EQ(-EBADMSG, rados::cls::numops::add(&ioctx, "myobject", key, 2.0));
+
+ // check that the omap entry was not changed
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bl = (*it).second;
+ value_out.assign(bl.c_str(), bl.length());
+
+ EXPECT_EQ(non_numeric_value, value_out);
+
+ ASSERT_EQ(0, destroy_one_pool_pp(pool_name, cluster));
+}
+
+TEST(ClsNumOps, Sub) {
+ Rados cluster;
+ std::string pool_name = get_temp_pool_name();
+ ASSERT_EQ("", create_one_pool_pp(pool_name, cluster));
+ IoCtx ioctx;
+ cluster.ioctx_create(pool_name.c_str(), ioctx);
+
+ // subtract a number from a non-existing key
+
+ std::string key = "my-key";
+ double value_in = 0.5;
+
+ std::stringstream stream;
+ stream << value_in;
+
+ ASSERT_EQ(0, rados::cls::numops::sub(&ioctx, "myobject", key, value_in));
+
+ // check that the omap entry was set and the value matches
+
+ std::set<std::string> keys;
+ std::map<std::string, bufferlist> omap;
+ keys.insert(key);
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ std::map<std::string, bufferlist>::iterator it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bufferlist bl = (*it).second;
+ std::string value_out(bl.c_str(), bl.length());
+
+ EXPECT_EQ("-" + stream.str(), value_out);
+
+ // subtract another value to the existing one
+
+ double new_value_in = 3.001;
+
+ ASSERT_EQ(0, rados::cls::numops::sub(&ioctx, "myobject", key, new_value_in));
+
+ // check that the omap entry's value matches
+
+ omap.clear();
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bl = (*it).second;
+ value_out.assign(bl.c_str(), bl.length());
+
+ stream.str("");
+ stream << -(value_in + new_value_in);
+
+ EXPECT_EQ(stream.str(), value_out);
+
+ // set the omap entry with some non-numeric value
+
+ omap.clear();
+
+ std::string non_numeric_value("some-non-numeric-text");
+ omap[key].append(non_numeric_value);
+
+ ASSERT_EQ(0, ioctx.omap_set("myobject", omap));
+
+ // check that subtracting a number does not succeed
+
+ omap.clear();
+
+ ASSERT_EQ(-EBADMSG, rados::cls::numops::sub(&ioctx, "myobject", key, 2.0));
+
+ // check that the omap entry was not changed
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bl = (*it).second;
+ value_out.assign(bl.c_str(), bl.length());
+
+ EXPECT_EQ(non_numeric_value, value_out);
+
+ ASSERT_EQ(0, destroy_one_pool_pp(pool_name, cluster));
+}
+
+TEST(ClsNumOps, Mul) {
+ Rados cluster;
+ std::string pool_name = get_temp_pool_name();
+ ASSERT_EQ("", create_one_pool_pp(pool_name, cluster));
+ IoCtx ioctx;
+ cluster.ioctx_create(pool_name.c_str(), ioctx);
+
+ // exec numops mul method with an empty bufferlist
+
+ bufferlist in, out;
+
+ ASSERT_EQ(-EINVAL, ioctx.exec("myobject", "numops", "mul", in, out));
+
+ // multiply a number to a non-existing key
+
+ std::string key = "my-key";
+ double value_in = 0.5;
+
+ std::stringstream stream;
+ stream << value_in;
+
+ ASSERT_EQ(0, rados::cls::numops::mul(&ioctx, "myobject", key, value_in));
+
+ // check that the omap entry was set and the value is zero
+
+ std::set<std::string> keys;
+ std::map<std::string, bufferlist> omap;
+ keys.insert(key);
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ std::map<std::string, bufferlist>::iterator it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bufferlist bl = (*it).second;
+ std::string value_out(bl.c_str(), bl.length());
+
+ EXPECT_EQ("0", value_out);
+
+ // set a non-zero value so we can effectively test multiplications
+
+ omap.clear();
+
+ omap[key].append(stream.str());
+
+ ASSERT_EQ(0, ioctx.omap_set("myobject", omap));
+
+ // multiply another value to the existing one
+
+ double new_value_in = 3.001;
+
+ ASSERT_EQ(0, rados::cls::numops::mul(&ioctx, "myobject", key, new_value_in));
+
+ // check that the omap entry's value matches
+
+ omap.clear();
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bl = (*it).second;
+ value_out.assign(bl.c_str(), bl.length());
+
+ stream.str("");
+ stream << (value_in * new_value_in);
+
+ EXPECT_EQ(stream.str(), value_out);
+
+ // set the omap entry with some non-numeric value
+
+ omap.clear();
+
+ std::string non_numeric_value("some-non-numeric-text");
+ omap[key].append(non_numeric_value);
+
+ ASSERT_EQ(0, ioctx.omap_set("myobject", omap));
+
+ // check that adding a number does not succeed
+
+ ASSERT_EQ(-EBADMSG, rados::cls::numops::mul(&ioctx, "myobject", key, 2.0));
+
+ // check that the omap entry was not changed
+
+ omap.clear();
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bl = (*it).second;
+ value_out.assign(bl.c_str(), bl.length());
+
+ EXPECT_EQ(non_numeric_value, value_out);
+
+ ASSERT_EQ(0, destroy_one_pool_pp(pool_name, cluster));
+}
+
+TEST(ClsNumOps, Div) {
+ Rados cluster;
+ std::string pool_name = get_temp_pool_name();
+ ASSERT_EQ("", create_one_pool_pp(pool_name, cluster));
+ IoCtx ioctx;
+ cluster.ioctx_create(pool_name.c_str(), ioctx);
+
+ // divide a non-existing key by a number
+
+ std::string key = "my-key";
+ double value_in = 0.5;
+
+ std::stringstream stream;
+ stream << value_in;
+
+ ASSERT_EQ(0, rados::cls::numops::div(&ioctx, "myobject", key, value_in));
+
+ // check that the omap entry was set and the value is zero
+
+ std::set<std::string> keys;
+ std::map<std::string, bufferlist> omap;
+ keys.insert(key);
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ std::map<std::string, bufferlist>::iterator it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bufferlist bl = (*it).second;
+ std::string value_out(bl.c_str(), bl.length());
+
+ EXPECT_EQ("0", value_out);
+
+ // check that division by zero is not allowed
+
+ ASSERT_EQ(-EINVAL, rados::cls::numops::div(&ioctx, "myobject", key, 0));
+
+ // set a non-zero value so we can effectively test divisions
+
+ omap.clear();
+
+ omap[key].append(stream.str());
+
+ ASSERT_EQ(0, ioctx.omap_set("myobject", omap));
+
+ // divide another value to the existing one
+
+ double new_value_in = 3.001;
+
+ ASSERT_EQ(0, rados::cls::numops::div(&ioctx, "myobject", key, new_value_in));
+
+ // check that the omap entry's value matches
+
+ omap.clear();
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bl = (*it).second;
+ value_out.assign(bl.c_str(), bl.length());
+
+ stream.str("");
+ stream << (value_in / new_value_in);
+
+ EXPECT_EQ(stream.str(), value_out);
+
+ omap.clear();
+
+ // set the omap entry with some non-numeric value
+
+ std::string non_numeric_value("some-non-numeric-text");
+ omap[key].append(non_numeric_value);
+
+ ASSERT_EQ(0, ioctx.omap_set("myobject", omap));
+
+ // check that adding a number does not succeed
+
+ ASSERT_EQ(-EBADMSG, rados::cls::numops::div(&ioctx, "myobject", key, 2.0));
+
+ // check that the omap entry was not changed
+
+ omap.clear();
+
+ ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap));
+
+ it = omap.find(key);
+
+ ASSERT_NE(omap.end(), it);
+
+ bl = (*it).second;
+ value_out.assign(bl.c_str(), bl.length());
+
+ EXPECT_EQ(non_numeric_value, value_out);
+
+ ASSERT_EQ(0, destroy_one_pool_pp(pool_name, cluster));
+}
Code Review / stor4nfv.git / blobdiff