+++ /dev/null
-// -*- mode:C; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
-// vim: ts=8 sw=2 smarttab
-
-#include "include/types.h"
-#include "cls/statelog/cls_statelog_types.h"
-#include "cls/statelog/cls_statelog_client.h"
-
-#include "include/utime.h"
-#include "common/Clock.h"
-#include "global/global_context.h"
-
-#include "gtest/gtest.h"
-#include "test/librados/test.h"
-
-#include <errno.h>
-#include <string>
-#include <vector>
-
-static librados::ObjectWriteOperation *new_op() {
- return new librados::ObjectWriteOperation();
-}
-
-static librados::ObjectReadOperation *new_rop() {
- return new librados::ObjectReadOperation();
-}
-
-static void reset_op(librados::ObjectWriteOperation **pop) {
- delete *pop;
- *pop = new_op();
-}
-static void reset_rop(librados::ObjectReadOperation **pop) {
- delete *pop;
- *pop = new_rop();
-}
-
-void add_log(librados::ObjectWriteOperation *op, const string& client_id, const string& op_id, string& obj, uint32_t state)
-{
- bufferlist bl;
- ::encode(state, bl);
-
- utime_t ts = ceph_clock_now();
-
- cls_statelog_add(*op, client_id, op_id, obj, ts, state, bl);
-}
-
-void next_op_id(string& op_id, int *id)
-{
- char buf[16];
- snprintf(buf, sizeof(buf), "%d", *id);
- op_id = buf;
- (*id)++;
-}
-
-static string get_obj_name(int num)
-{
- char buf[16];
- snprintf(buf, sizeof(buf), "obj-%d", num);
- return string(buf);
-}
-
-static void get_entries_by_object(librados::IoCtx& ioctx, string& oid,
- list<cls_statelog_entry>& entries, string& object, string& op_id, int expected)
-{
- /* search everything */
- string empty_str, marker;
-
- librados::ObjectReadOperation *rop = new_rop();
- bufferlist obl;
- bool truncated;
- cls_statelog_list(*rop, empty_str, op_id, object, marker, 0, entries, &marker, &truncated);
- ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
- ASSERT_EQ(expected, (int)entries.size());
- delete rop;
-}
-
-static void get_entries_by_client_id(librados::IoCtx& ioctx, string& oid,
- list<cls_statelog_entry>& entries, string& client_id, string& op_id, int expected)
-{
- /* search everything */
- string empty_str, marker;
-
- librados::ObjectReadOperation *rop = new_rop();
- bufferlist obl;
- bool truncated;
- cls_statelog_list(*rop, client_id, op_id, empty_str, marker, 0, entries, &marker, &truncated);
- ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
- ASSERT_EQ(expected, (int)entries.size());
- delete rop;
-}
-
-static void get_all_entries(librados::IoCtx& ioctx, string& oid, list<cls_statelog_entry>& entries, int expected)
-{
- /* search everything */
- string object, op_id;
- get_entries_by_object(ioctx, oid, entries, object, op_id, expected);
-}
-
-TEST(cls_rgw, test_statelog_basic)
-{
- librados::Rados rados;
- librados::IoCtx ioctx;
- string pool_name = get_temp_pool_name();
-
- /* create pool */
- ASSERT_EQ("", create_one_pool_pp(pool_name, rados));
- ASSERT_EQ(0, rados.ioctx_create(pool_name.c_str(), ioctx));
-
- string oid = "obj";
-
- /* create object */
- ASSERT_EQ(0, ioctx.create(oid, true));
-
- int id = 0;
- string client_id[] = { "client-1", "client-2" };
-
- const int num_ops = 10;
- string op_ids[num_ops];
-
- librados::ObjectWriteOperation *op = new_op();
-
- for (int i = 0; i < num_ops; i++) {
- next_op_id(op_ids[i], &id);
- string obj = get_obj_name(i / 2);
- string cid = client_id[i / (num_ops / 2)];
- add_log(op, cid, op_ids[i], obj, i /* just for testing */);
- }
- ASSERT_EQ(0, ioctx.operate(oid, op));
-
- librados::ObjectReadOperation *rop = new_rop();
-
- list<cls_statelog_entry> entries;
- bool truncated;
-
- /* check list by client_id */
-
- int total_count = 0;
- for (int j = 0; j < 2; j++) {
- string marker;
- string obj;
- string cid = client_id[j];
- string op_id;
-
- bufferlist obl;
-
- cls_statelog_list(*rop, cid, op_id, obj, marker, 1, entries, &marker, &truncated);
- ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
- ASSERT_EQ(1, (int)entries.size());
-
- reset_rop(&rop);
- marker.clear();
- cls_statelog_list(*rop, cid, op_id, obj, marker, 0, entries, &marker, &truncated);
- obl.clear();
- ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
-
- ASSERT_EQ(5, (int)entries.size());
- ASSERT_EQ(0, (int)truncated);
-
- map<string, string> emap;
- for (list<cls_statelog_entry>::iterator iter = entries.begin(); iter != entries.end(); ++iter) {
- ASSERT_EQ(cid, iter->client_id);
- emap[iter->op_id] = iter->object;
- }
- ASSERT_EQ(5, (int)emap.size());
- /* verify correct object */
- for (int i = 0; i < num_ops / 2; i++, total_count++) {
- string ret_obj = emap[op_ids[total_count]];
- string obj = get_obj_name(total_count / 2);
- ASSERT_EQ(0, ret_obj.compare(obj));
- }
- }
-
- entries.clear();
- /* now search by object */
- total_count = 0;
- for (int i = 0; i < num_ops; i++) {
- string marker;
- string obj = get_obj_name(i / 2);
- string cid;
- string op_id;
- bufferlist obl;
-
- reset_rop(&rop);
- cls_statelog_list(*rop, cid, op_id, obj, marker, 0, entries, &marker, &truncated);
- ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
- ASSERT_EQ(2, (int)entries.size());
- }
-
- /* search everything */
-
- get_all_entries(ioctx, oid, entries, 10);
-
- /* now remove an entry */
- cls_statelog_entry e = entries.front();
- entries.pop_front();
-
- reset_op(&op);
- cls_statelog_remove_by_client(*op, e.client_id, e.op_id);
- ASSERT_EQ(0, ioctx.operate(oid, op));
-
- get_all_entries(ioctx, oid, entries, 9);
-
- get_entries_by_object(ioctx, oid, entries, e.object, e.op_id, 0);
- get_entries_by_client_id(ioctx, oid, entries, e.client_id, e.op_id, 0);
-
- string empty_str;
- get_entries_by_client_id(ioctx, oid, entries, e.client_id, empty_str, 4);
- get_entries_by_object(ioctx, oid, entries, e.object, empty_str, 1);
- delete op;
- delete rop;
-}
-