From e6918187568dbd01842d8d1d2c808ce16a894239 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 21 Apr 2024 13:54:28 +0200 Subject: Adding upstream version 18.2.2. Signed-off-by: Daniel Baumann --- src/test/cls_numops/test_cls_numops.cc | 414 +++++++++++++++++++++++++++++++++ 1 file changed, 414 insertions(+) create mode 100644 src/test/cls_numops/test_cls_numops.cc (limited to 'src/test/cls_numops/test_cls_numops.cc') diff --git a/src/test/cls_numops/test_cls_numops.cc b/src/test/cls_numops/test_cls_numops.cc new file mode 100644 index 000000000..ebb225620 --- /dev/null +++ b/src/test/cls_numops/test_cls_numops.cc @@ -0,0 +1,414 @@ +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2015 CERN + * + * Author: Joaquim Rocha + * + * 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 +#include +#include +#include +#include + +#include "cls/numops/cls_numops_client.h" +#include "gtest/gtest.h" +#include "include/rados/librados.hpp" +#include "test/librados/test_cxx.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 keys; + std::map omap; + keys.insert(key); + + ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap)); + + std::map::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 keys; + std::map omap; + keys.insert(key); + + ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap)); + + std::map::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 keys; + std::map omap; + keys.insert(key); + + ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap)); + + std::map::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 keys; + std::map omap; + keys.insert(key); + + ASSERT_EQ(0, ioctx.omap_get_vals_by_keys("myobject", keys, &omap)); + + std::map::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)); +} -- cgit v1.2.3