1 // -*- mode:C; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
2 // vim: ts=8 sw=2 smarttab
4 #include "include/types.h"
5 #include "cls/statelog/cls_statelog_types.h"
6 #include "cls/statelog/cls_statelog_client.h"
8 #include "include/utime.h"
9 #include "common/Clock.h"
10 #include "global/global_context.h"
12 #include "gtest/gtest.h"
13 #include "test/librados/test.h"
19 static librados::ObjectWriteOperation *new_op() {
20 return new librados::ObjectWriteOperation();
23 static librados::ObjectReadOperation *new_rop() {
24 return new librados::ObjectReadOperation();
27 static void reset_op(librados::ObjectWriteOperation **pop) {
31 static void reset_rop(librados::ObjectReadOperation **pop) {
36 void add_log(librados::ObjectWriteOperation *op, const string& client_id, const string& op_id, string& obj, uint32_t state)
41 utime_t ts = ceph_clock_now();
43 cls_statelog_add(*op, client_id, op_id, obj, ts, state, bl);
46 void next_op_id(string& op_id, int *id)
49 snprintf(buf, sizeof(buf), "%d", *id);
54 static string get_obj_name(int num)
57 snprintf(buf, sizeof(buf), "obj-%d", num);
61 static void get_entries_by_object(librados::IoCtx& ioctx, string& oid,
62 list<cls_statelog_entry>& entries, string& object, string& op_id, int expected)
64 /* search everything */
65 string empty_str, marker;
67 librados::ObjectReadOperation *rop = new_rop();
70 cls_statelog_list(*rop, empty_str, op_id, object, marker, 0, entries, &marker, &truncated);
71 ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
72 ASSERT_EQ(expected, (int)entries.size());
76 static void get_entries_by_client_id(librados::IoCtx& ioctx, string& oid,
77 list<cls_statelog_entry>& entries, string& client_id, string& op_id, int expected)
79 /* search everything */
80 string empty_str, marker;
82 librados::ObjectReadOperation *rop = new_rop();
85 cls_statelog_list(*rop, client_id, op_id, empty_str, marker, 0, entries, &marker, &truncated);
86 ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
87 ASSERT_EQ(expected, (int)entries.size());
91 static void get_all_entries(librados::IoCtx& ioctx, string& oid, list<cls_statelog_entry>& entries, int expected)
93 /* search everything */
95 get_entries_by_object(ioctx, oid, entries, object, op_id, expected);
98 TEST(cls_rgw, test_statelog_basic)
100 librados::Rados rados;
101 librados::IoCtx ioctx;
102 string pool_name = get_temp_pool_name();
105 ASSERT_EQ("", create_one_pool_pp(pool_name, rados));
106 ASSERT_EQ(0, rados.ioctx_create(pool_name.c_str(), ioctx));
111 ASSERT_EQ(0, ioctx.create(oid, true));
114 string client_id[] = { "client-1", "client-2" };
116 const int num_ops = 10;
117 string op_ids[num_ops];
119 librados::ObjectWriteOperation *op = new_op();
121 for (int i = 0; i < num_ops; i++) {
122 next_op_id(op_ids[i], &id);
123 string obj = get_obj_name(i / 2);
124 string cid = client_id[i / (num_ops / 2)];
125 add_log(op, cid, op_ids[i], obj, i /* just for testing */);
127 ASSERT_EQ(0, ioctx.operate(oid, op));
129 librados::ObjectReadOperation *rop = new_rop();
131 list<cls_statelog_entry> entries;
134 /* check list by client_id */
137 for (int j = 0; j < 2; j++) {
140 string cid = client_id[j];
145 cls_statelog_list(*rop, cid, op_id, obj, marker, 1, entries, &marker, &truncated);
146 ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
147 ASSERT_EQ(1, (int)entries.size());
151 cls_statelog_list(*rop, cid, op_id, obj, marker, 0, entries, &marker, &truncated);
153 ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
155 ASSERT_EQ(5, (int)entries.size());
156 ASSERT_EQ(0, (int)truncated);
158 map<string, string> emap;
159 for (list<cls_statelog_entry>::iterator iter = entries.begin(); iter != entries.end(); ++iter) {
160 ASSERT_EQ(cid, iter->client_id);
161 emap[iter->op_id] = iter->object;
163 ASSERT_EQ(5, (int)emap.size());
164 /* verify correct object */
165 for (int i = 0; i < num_ops / 2; i++, total_count++) {
166 string ret_obj = emap[op_ids[total_count]];
167 string obj = get_obj_name(total_count / 2);
168 ASSERT_EQ(0, ret_obj.compare(obj));
173 /* now search by object */
175 for (int i = 0; i < num_ops; i++) {
177 string obj = get_obj_name(i / 2);
183 cls_statelog_list(*rop, cid, op_id, obj, marker, 0, entries, &marker, &truncated);
184 ASSERT_EQ(0, ioctx.operate(oid, rop, &obl));
185 ASSERT_EQ(2, (int)entries.size());
188 /* search everything */
190 get_all_entries(ioctx, oid, entries, 10);
192 /* now remove an entry */
193 cls_statelog_entry e = entries.front();
197 cls_statelog_remove_by_client(*op, e.client_id, e.op_id);
198 ASSERT_EQ(0, ioctx.operate(oid, op));
200 get_all_entries(ioctx, oid, entries, 9);
202 get_entries_by_object(ioctx, oid, entries, e.object, e.op_id, 0);
203 get_entries_by_client_id(ioctx, oid, entries, e.client_id, e.op_id, 0);
206 get_entries_by_client_id(ioctx, oid, entries, e.client_id, empty_str, 4);
207 get_entries_by_object(ioctx, oid, entries, e.object, empty_str, 1);