X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fceph%2Fsrc%2Ftest%2Ftest_rgw_admin_opstate.cc;fp=src%2Fceph%2Fsrc%2Ftest%2Ftest_rgw_admin_opstate.cc;h=1a41df30f1aa08a3bb03292cdd6a8a9890016824;hb=812ff6ca9fcd3e629e49d4328905f33eee8ca3f5;hp=0000000000000000000000000000000000000000;hpb=15280273faafb77777eab341909a3f495cf248d9;p=stor4nfv.git diff --git a/src/ceph/src/test/test_rgw_admin_opstate.cc b/src/ceph/src/test/test_rgw_admin_opstate.cc new file mode 100644 index 0000000..1a41df3 --- /dev/null +++ b/src/ceph/src/test/test_rgw_admin_opstate.cc @@ -0,0 +1,833 @@ +// -*- 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) 2013 eNovance SAS + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +extern "C"{ +#include +} +#include "common/ceph_crypto.h" +#include "include/str_list.h" +#include "common/ceph_json.h" +#include "common/code_environment.h" +#include "common/ceph_argparse.h" +#include "common/Finisher.h" +#include "global/global_init.h" +#include "rgw/rgw_common.h" +#include "rgw/rgw_bucket.h" +#include "rgw/rgw_rados.h" +#include "include/utime.h" +#include "include/object.h" +#include "cls/statelog/cls_statelog_types.h" +#define GTEST +#ifdef GTEST +#include +#else +#define TEST(x, y) void y() +#define ASSERT_EQ(v, s) if(v != s)cout << "Error at " << __LINE__ << "(" << #v << "!= " << #s << "\n"; \ + else cout << "(" << #v << "==" << #s << ") PASSED\n"; +#define EXPECT_EQ(v, s) ASSERT_EQ(v, s) +#define ASSERT_TRUE(c) if(c)cout << "Error at " << __LINE__ << "(" << #c << ")" << "\n"; \ + else cout << "(" << #c << ") PASSED\n"; +#define EXPECT_TRUE(c) ASSERT_TRUE(c) +#endif +using namespace std; + +#define CURL_VERBOSE 0 +#define HTTP_RESPONSE_STR "RespCode" +#define CEPH_CRYPTO_HMACSHA1_DIGESTSIZE 20 +#define RGW_ADMIN_RESP_PATH "/tmp/.test_rgw_admin_resp" + +static string uid = "ceph"; +static string display_name = "CEPH"; + +extern "C" int ceph_armor(char *dst, const char *dst_end, + const char *src, const char *end); +static void print_usage(char *exec){ + cout << "Usage: " << exec << " \n"; + cout << "Options:\n" + "-g - The ip address of the gateway\n" + "-p - The port number of the gateway\n" + "-c - Absolute path of ceph config file\n" + "-rgw-admin - radosgw-admin absolute path\n"; +} + +namespace admin_log { +class test_helper { + private: + string host; + string port; + string creds; + string rgw_admin_path; + string conf_path; + CURL *curl_inst; + map response; + list extra_hdrs; + string *resp_data; + unsigned resp_code; + public: + test_helper() : curl_inst(NULL), resp_data(NULL), resp_code(0) { + curl_global_init(CURL_GLOBAL_ALL); + } + ~test_helper(){ + curl_global_cleanup(); + } + int send_request(string method, string uri, + size_t (*function)(void *,size_t,size_t,void *) = 0, + void *ud = 0, size_t length = 0); + int extract_input(int argc, char *argv[]); + string& get_response(string hdr){ + return response[hdr]; + } + void set_extra_header(string hdr){ + extra_hdrs.push_back(hdr); + } + void set_response(char *val); + void set_response_data(char *data, size_t len){ + if(resp_data) delete resp_data; + resp_data = new string(data, len); + } + string& get_rgw_admin_path() { + return rgw_admin_path; + } + string& get_ceph_conf_path() { + return conf_path; + } + void set_creds(string& c) { + creds = c; + } + const string *get_response_data(){return resp_data;} + unsigned get_resp_code(){return resp_code;} +}; + +int test_helper::extract_input(int argc, char *argv[]){ +#define ERR_CHECK_NEXT_PARAM(o) \ + if(((int)loop + 1) >= argc)return -1; \ + else o = argv[loop+1]; + + for(unsigned loop = 1;loop < (unsigned)argc; loop += 2){ + if(strcmp(argv[loop], "-g") == 0){ + ERR_CHECK_NEXT_PARAM(host); + }else if(strcmp(argv[loop],"-p") == 0){ + ERR_CHECK_NEXT_PARAM(port); + }else if(strcmp(argv[loop], "-c") == 0){ + ERR_CHECK_NEXT_PARAM(conf_path); + }else if(strcmp(argv[loop], "-rgw-admin") == 0){ + ERR_CHECK_NEXT_PARAM(rgw_admin_path); + }else return -1; + } + if(host.length() <= 0 || + rgw_admin_path.length() <= 0) + return -1; + return 0; +} + +void test_helper::set_response(char *r){ + string sr(r), h, v; + size_t off = sr.find(": "); + if(off != string::npos){ + h.assign(sr, 0, off); + v.assign(sr, off + 2, sr.find("\r\n") - (off+2)); + }else{ + /*Could be the status code*/ + if(sr.find("HTTP/") != string::npos){ + h.assign(HTTP_RESPONSE_STR); + off = sr.find(" "); + v.assign(sr, off + 1, sr.find("\r\n") - (off + 1)); + resp_code = atoi((v.substr(0, 3)).c_str()); + } + } + response[h] = v; +} + +size_t write_header(void *ptr, size_t size, size_t nmemb, void *ud){ + test_helper *h = static_cast(ud); + h->set_response((char *)ptr); + return size*nmemb; +} + +size_t write_data(void *ptr, size_t size, size_t nmemb, void *ud){ + test_helper *h = static_cast(ud); + h->set_response_data((char *)ptr, size*nmemb); + return size*nmemb; +} + +static inline void buf_to_hex(const unsigned char *buf, int len, char *str) +{ + int i; + str[0] = '\0'; + for (i = 0; i < len; i++) { + sprintf(&str[i*2], "%02x", (int)buf[i]); + } +} + +static void calc_hmac_sha1(const char *key, int key_len, + const char *msg, int msg_len, char *dest) +/* destination should be CEPH_CRYPTO_HMACSHA1_DIGESTSIZE bytes long */ +{ + ceph::crypto::HMACSHA1 hmac((const unsigned char *)key, key_len); + hmac.Update((const unsigned char *)msg, msg_len); + hmac.Final((unsigned char *)dest); + + char hex_str[(CEPH_CRYPTO_HMACSHA1_DIGESTSIZE * 2) + 1]; + admin_log::buf_to_hex((unsigned char *)dest, CEPH_CRYPTO_HMACSHA1_DIGESTSIZE, hex_str); +} + +static int get_s3_auth(string method, string creds, string date, string res, string& out){ + string aid, secret, auth_hdr; + string tmp_res; + size_t off = creds.find(":"); + out = ""; + if(off != string::npos){ + aid.assign(creds, 0, off); + secret.assign(creds, off + 1, string::npos); + + /*sprintf(auth_hdr, "%s\n\n\n%s\n%s", req_type, date, res);*/ + char hmac_sha1[CEPH_CRYPTO_HMACSHA1_DIGESTSIZE]; + char b64[65]; /* 64 is really enough */ + size_t off = res.find("?"); + if(off == string::npos) + tmp_res = res; + else + tmp_res.assign(res, 0, off); + auth_hdr.append(method + string("\n\n\n") + date + string("\n") + tmp_res); + admin_log::calc_hmac_sha1(secret.c_str(), secret.length(), + auth_hdr.c_str(), auth_hdr.length(), hmac_sha1); + int ret = ceph_armor(b64, b64 + 64, hmac_sha1, + hmac_sha1 + CEPH_CRYPTO_HMACSHA1_DIGESTSIZE); + if (ret < 0) { + cout << "ceph_armor failed\n"; + return -1; + } + b64[ret] = 0; + out.append(aid + string(":") + b64); + }else return -1; + return 0; +} + +void get_date(string& d){ + struct timeval tv; + char date[64]; + struct tm tm; + char *days[] = {(char *)"Sun", (char *)"Mon", (char *)"Tue", + (char *)"Wed", (char *)"Thu", (char *)"Fri", + (char *)"Sat"}; + char *months[] = {(char *)"Jan", (char *)"Feb", (char *)"Mar", + (char *)"Apr", (char *)"May", (char *)"Jun", + (char *)"Jul",(char *) "Aug", (char *)"Sep", + (char *)"Oct", (char *)"Nov", (char *)"Dec"}; + gettimeofday(&tv, NULL); + gmtime_r(&tv.tv_sec, &tm); + sprintf(date, "%s, %d %s %d %d:%d:%d GMT", + days[tm.tm_wday], + tm.tm_mday, months[tm.tm_mon], + tm.tm_year + 1900, + tm.tm_hour, tm.tm_min, tm.tm_sec); + d = date; +} + +int test_helper::send_request(string method, string res, + size_t (*read_function)( void *,size_t,size_t,void *), + void *ud, + size_t length){ + string url; + string auth, date; + url.append(string("http://") + host); + if(port.length() > 0)url.append(string(":") + port); + url.append(res); + curl_inst = curl_easy_init(); + if(curl_inst){ + curl_easy_setopt(curl_inst, CURLOPT_URL, url.c_str()); + curl_easy_setopt(curl_inst, CURLOPT_CUSTOMREQUEST, method.c_str()); + curl_easy_setopt(curl_inst, CURLOPT_VERBOSE, CURL_VERBOSE); + curl_easy_setopt(curl_inst, CURLOPT_HEADERFUNCTION, admin_log::write_header); + curl_easy_setopt(curl_inst, CURLOPT_WRITEHEADER, (void *)this); + curl_easy_setopt(curl_inst, CURLOPT_WRITEFUNCTION, admin_log::write_data); + curl_easy_setopt(curl_inst, CURLOPT_WRITEDATA, (void *)this); + if(read_function){ + curl_easy_setopt(curl_inst, CURLOPT_READFUNCTION, read_function); + curl_easy_setopt(curl_inst, CURLOPT_READDATA, (void *)ud); + curl_easy_setopt(curl_inst, CURLOPT_UPLOAD, 1L); + curl_easy_setopt(curl_inst, CURLOPT_INFILESIZE_LARGE, (curl_off_t)length); + } + + get_date(date); + string http_date; + http_date.append(string("Date: ") + date); + + string s3auth; + if (admin_log::get_s3_auth(method, creds, date, res, s3auth) < 0) + return -1; + auth.append(string("Authorization: AWS ") + s3auth); + + struct curl_slist *slist = NULL; + slist = curl_slist_append(slist, auth.c_str()); + slist = curl_slist_append(slist, http_date.c_str()); + for(list::iterator it = extra_hdrs.begin(); + it != extra_hdrs.end(); ++it){ + slist = curl_slist_append(slist, (*it).c_str()); + } + if(read_function) + curl_slist_append(slist, "Expect:"); + curl_easy_setopt(curl_inst, CURLOPT_HTTPHEADER, slist); + + response.erase(response.begin(), response.end()); + extra_hdrs.erase(extra_hdrs.begin(), extra_hdrs.end()); + CURLcode res = curl_easy_perform(curl_inst); + if(res != CURLE_OK){ + cout << "Curl perform failed for " << url << ", res: " << + curl_easy_strerror(res) << "\n"; + return -1; + } + curl_slist_free_all(slist); + } + curl_easy_cleanup(curl_inst); + return 0; +} +}; + +admin_log::test_helper *g_test; +Finisher *finisher; +RGWRados *store; + +int run_rgw_admin(string& cmd, string& resp) { + pid_t pid; + pid = fork(); + if (pid == 0) { + /* child */ + list l; + get_str_list(cmd, " \t", l); + char *argv[l.size()]; + unsigned loop = 1; + + argv[0] = (char *)"radosgw-admin"; + for (list::iterator it = l.begin(); + it != l.end(); ++it) { + argv[loop++] = (char *)(*it).c_str(); + } + argv[loop] = NULL; + if (!freopen(RGW_ADMIN_RESP_PATH, "w+", stdout)) { + cout << "Unable to open stdout file" << std::endl; + } + execv((g_test->get_rgw_admin_path()).c_str(), argv); + } else if (pid > 0) { + int status; + waitpid(pid, &status, 0); + if (WIFEXITED(status)) { + if(WEXITSTATUS(status) != 0) { + cout << "Child exited with status " << WEXITSTATUS(status) << std::endl; + return -1; + } + } + ifstream in; + struct stat st; + + if (stat(RGW_ADMIN_RESP_PATH, &st) < 0) { + cout << "Error stating the admin response file, errno " << errno << std::endl; + return -1; + } else { + char *data = (char *)malloc(st.st_size + 1); + in.open(RGW_ADMIN_RESP_PATH); + in.read(data, st.st_size); + in.close(); + data[st.st_size] = 0; + resp = data; + free(data); + unlink(RGW_ADMIN_RESP_PATH); + /* cout << "radosgw-admin " << cmd << ": " << resp << std::endl; */ + } + } else + return -1; + return 0; +} + +int get_creds(string& json, string& creds) { + JSONParser parser; + if(!parser.parse(json.c_str(), json.length())) { + cout << "Error parsing create user response" << std::endl; + return -1; + } + + RGWUserInfo info; + decode_json_obj(info, &parser); + creds = ""; + for(map::iterator it = info.access_keys.begin(); + it != info.access_keys.end(); ++it) { + RGWAccessKey _k = it->second; + /*cout << "accesskeys [ " << it->first << " ] = " << + "{ " << _k.id << ", " << _k.key << ", " << _k.subuser << "}" << std::endl;*/ + creds.append(it->first + string(":") + _k.key); + break; + } + return 0; +} + +int user_create(string& uid, string& display_name, bool set_creds = true) { + stringstream ss; + string creds; + ss << "-c " << g_test->get_ceph_conf_path() << " user create --uid=" << uid + << " --display-name=" << display_name; + + string out; + string cmd = ss.str(); + if(run_rgw_admin(cmd, out) != 0) { + cout << "Error creating user" << std::endl; + return -1; + } + get_creds(out, creds); + if(set_creds) + g_test->set_creds(creds); + return 0; +} + +int user_info(string& uid, string& display_name, RGWUserInfo& uinfo) { + stringstream ss; + ss << "-c " << g_test->get_ceph_conf_path() << " user info --uid=" << uid + << " --display-name=" << display_name; + + string out; + string cmd = ss.str(); + if(run_rgw_admin(cmd, out) != 0) { + cout << "Error reading user information" << std::endl; + return -1; + } + JSONParser parser; + if(!parser.parse(out.c_str(), out.length())) { + cout << "Error parsing create user response" << std::endl; + return -1; + } + decode_json_obj(uinfo, &parser); + return 0; +} + +int user_rm(string& uid, string& display_name) { + stringstream ss; + ss << "-c " << g_test->get_ceph_conf_path() << + " metadata rm --metadata-key=user:" << uid; + + string out; + string cmd = ss.str(); + if(run_rgw_admin(cmd, out) != 0) { + cout << "Error removing user" << std::endl; + return -1; + } + return 0; +} + +int caps_add(const char * name, const char *perm) { + stringstream ss; + + ss << "-c " << g_test->get_ceph_conf_path() << " caps add --caps=" << + name << "=" << perm << " --uid=" << uid; + string out; + string cmd = ss.str(); + if(run_rgw_admin(cmd, out) != 0) { + cout << "Error creating user" << std::endl; + return -1; + } + return 0; +} + +int caps_rm(const char * name, const char *perm) { + stringstream ss; + + ss << "-c " << g_test->get_ceph_conf_path() << " caps rm --caps=" << + name << "=" << perm << " --uid=" << uid; + string out; + string cmd = ss.str(); + if(run_rgw_admin(cmd, out) != 0) { + cout << "Error creating user" << std::endl; + return -1; + } + return 0; +} + +size_t read_dummy_post(void *ptr, size_t s, size_t n, void *ud) { + int dummy = 0; + memcpy(ptr, &dummy, sizeof(dummy)); + return sizeof(dummy); +} + +int parse_json_resp(JSONParser &parser) { + string *resp; + resp = (string *)g_test->get_response_data(); + if(!resp) + return -1; + if(!parser.parse(resp->c_str(), resp->length())) { + cout << "Error parsing create user response" << std::endl; + return -1; + } + return 0; +} + +static int decode_json(JSONObj *obj, cls_statelog_entry& ret) { + JSONDecoder::decode_json("op_id", ret.op_id, obj); + JSONDecoder::decode_json("client_id", ret.client_id, obj); + JSONDecoder::decode_json("object", ret.object, obj); + string state; + JSONDecoder::decode_json("state", state, obj); + RGWOpState oc(store); + return oc.state_from_str(state, (RGWOpState::OpState *)&ret.state); +} + +static int get_opstate_list(list &entries) { + JSONParser parser; + if (parse_json_resp(parser) != 0) + return -1; + if (!parser.is_array()) + return -1; + + vector l; + l = parser.get_array_elements(); + int loop = 0; + for(vector::iterator it = l.begin(); + it != l.end(); ++it, loop++) { + JSONParser jp; + cls_statelog_entry entry; + + if(!jp.parse((*it).c_str(), (*it).length())) { + cerr << "Error parsing log json object" << std::endl; + return -1; + } + EXPECT_EQ(decode_json((JSONObj *)&jp, entry), 0); + entries.push_back(entry); + } + return 0; +} + + +TEST(TestRGWAdmin, opstate_set_list_delete) { + const char *cname = "opstate", + *perm = "*"; + string rest_req; + stringstream ss; + list entries; + string cid_1 = "cid_1", cid_2 = "cid_2"; + string oid_1 = "operation", oid_2 = "operation"; + string obj_1 = "obj1", obj_2 = "obj2"; + string state; + + ASSERT_EQ(user_create(uid, display_name), 0); + ASSERT_EQ(0, caps_add(cname, perm)); + + rest_req = "/admin/opstate?client-id=1&op-id=1&object=o"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing state*/ + + rest_req = "/admin/opstate?client-id=1&op-id=1&state=in-progress"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing object*/ + + rest_req = "/admin/opstate?client-id=1&state=in-progress&object=o"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing op-id*/ + + rest_req = "/admin/opstate?state=in-progress&op-id=1&object=o"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing client-id*/ + + rest_req = "/admin/opstate?state=invalid&op-id=1&object=o&client-id=1"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Invalid state*/ + + state = "in-progress"; + entries.clear(); + ss << "/admin/opstate?client-id=" << cid_1 << "&op-id=" << oid_1 + << "&object=" << obj_1 << "&state=" << state; + rest_req = ss.str(); + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(200U, g_test->get_resp_code()); + + ss.str(""); + entries.clear(); + rest_req = "/admin/opstate"; + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 1U); + + if (entries.size() == 1U) { + list::iterator it = entries.begin(); + + EXPECT_TRUE((*it).client_id.compare(cid_1) == 0); + EXPECT_TRUE((*it).op_id.compare(oid_1) == 0); + EXPECT_TRUE((*it).object.compare(obj_1) == 0); + EXPECT_EQ((*it).state, (uint32_t)RGWOpState::OPSTATE_IN_PROGRESS); + } + + state = "complete"; + ss.str(""); + entries.clear(); + ss << "/admin/opstate?client-id=" << cid_1 << "&op-id=" << oid_1 + << "&object=" << obj_1 << "&state=" << state; + rest_req = ss.str(); + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(200U, g_test->get_resp_code()); + + entries.clear(); + rest_req = "/admin/opstate"; + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 1U); + + if (entries.size() == 1U) { + list::iterator it = entries.begin(); + + EXPECT_TRUE((*it).client_id.compare(cid_1) == 0); + EXPECT_TRUE((*it).op_id.compare(oid_1) == 0); + EXPECT_TRUE((*it).object.compare(obj_1) == 0); + EXPECT_EQ((*it).state, (uint32_t)RGWOpState::OPSTATE_COMPLETE); + } + + ss.str(""); + entries.clear(); + ss << "/admin/opstate?client-id=" << cid_2 << "&op-id=" << oid_2 + << "&object=" << obj_2 << "&state=" << state; + rest_req = ss.str(); + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(200U, g_test->get_resp_code()); + + entries.clear(); + rest_req = "/admin/opstate"; + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 2U); + + if (entries.size() == 1U) { + list::iterator it = entries.begin(); + + EXPECT_TRUE((*it).client_id.compare(cid_1) == 0); + EXPECT_TRUE((*it).op_id.compare(oid_1) == 0); + EXPECT_TRUE((*it).object.compare(obj_1) == 0); + EXPECT_EQ((*it).state, (uint32_t)RGWOpState::OPSTATE_COMPLETE); + + ++it; + EXPECT_TRUE((*it).client_id.compare(cid_2) == 0); + EXPECT_TRUE((*it).op_id.compare(oid_2) == 0); + EXPECT_TRUE((*it).object.compare(obj_2) == 0); + EXPECT_EQ((*it).state, (uint32_t)RGWOpState::OPSTATE_COMPLETE); + } + + entries.clear(); + rest_req = "/admin/opstate?max-entries=1"; + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 1U); + + ss.str(""); + entries.clear(); + ss << "/admin/opstate?client-id=" << cid_2 << "&op-id=" << oid_2 << "&object=" << obj_2; + rest_req = ss.str(); + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 1U); + + ss.str(""); + entries.clear(); + ss << "/admin/opstate?client-id=" << cid_2; + rest_req = ss.str(); + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 1U); + + ss.str(""); + entries.clear(); + ss << "/admin/opstate?op-id=" << oid_2; + rest_req = ss.str(); + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 2U); + + ss.str(""); + entries.clear(); + ss << "/admin/opstate?op-id=" << oid_2 << "&object=" << obj_1 << "&client-id=" << cid_2 ; + rest_req = ss.str(); + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 0U); + + rest_req = "/admin/opstate?client-id=1&op-id=1"; + g_test->send_request(string("DELETE"), rest_req); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing object*/ + + rest_req = "/admin/opstate?client-id=1&object=1"; + g_test->send_request(string("DELETE"), rest_req); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing op-id*/ + + rest_req = "/admin/opstate?object=1&op-id=1"; + g_test->send_request(string("DELETE"), rest_req); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing client-id*/ + + rest_req = "/admin/opstate?object=1&op-id=1&client-id=1"; + g_test->send_request(string("DELETE"), rest_req); + EXPECT_EQ(404U, g_test->get_resp_code()); + + ss.str(""); + ss << "/admin/opstate?client-id=" << cid_1 << "&op-id=" << oid_1 + << "&object=" << obj_1; + rest_req = ss.str(); + g_test->send_request(string("DELETE"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + + ss.str(""); + entries.clear(); + ss << "/admin/opstate"; + rest_req = ss.str(); + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 1U); + + ss.str(""); + ss << "/admin/opstate?client-id=" << cid_2 << "&op-id=" << oid_2 + << "&object=" << obj_2; + rest_req = ss.str(); + g_test->send_request(string("DELETE"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + + ss.str(""); + entries.clear(); + ss << "/admin/opstate"; + rest_req = ss.str(); + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + EXPECT_EQ(get_opstate_list(entries), 0); + EXPECT_EQ(entries.size(), 0U); + + ASSERT_EQ(caps_rm(cname, perm), 0); + ASSERT_EQ(user_rm(uid, display_name), 0); +} + +TEST(TestRGWAdmin, opstate_renew) { + const char *cname = "opstate", + *perm = "*"; + string rest_req; + stringstream ss; + string cid_1 = "cid_1"; + string oid_1 = "operation"; + string obj_1 = "obj1"; + string state; + + ASSERT_EQ(user_create(uid, display_name), 0); + ASSERT_EQ(0, caps_add(cname, perm)); + + rest_req = "/admin/opstate?renew&client-id=1&op-id=1&object=o"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing state*/ + + rest_req = "/admin/opstate?renew&client-id=1&op-id=1&state=in-progress"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing object*/ + + rest_req = "/admin/opstate?renew&client-id=1&state=in-progress&object=o"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing op-id*/ + + rest_req = "/admin/opstate?renew&state=in-progress&op-id=1&object=o"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Missing client-id*/ + + rest_req = "/admin/opstate?&renew&state=invalid&op-id=1&object=o&client-id=1"; + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(400U, g_test->get_resp_code()); /*Invalid state*/ + + state = "complete"; + ss.str(""); + ss << "/admin/opstate?client-id=" << cid_1 << "&op-id=" << oid_1 + << "&object=" << obj_1 << "&state=" << state; + rest_req = ss.str(); + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(200U, g_test->get_resp_code()); + + rest_req = "/admin/opstate"; + g_test->send_request(string("GET"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + + ss.str(""); + ss << "/admin/opstate?renew&client-id=" << cid_1 << "&op-id=1" + << "&object=" << obj_1 << "&state=" << state; + rest_req = ss.str(); + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(404U, g_test->get_resp_code()); + + ss.str(""); + ss << "/admin/opstate?renew&client-id=" << cid_1 << "&op-id=" << oid_1 + << "&object=" << obj_1 << "&state=" << state; + rest_req = ss.str(); + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(200U, g_test->get_resp_code()); + + ss.str(""); + ss << "/admin/opstate?renew&client-id=" << cid_1 << "&op-id=" << oid_1 + << "&object=" << obj_1 << "&state=in-progress"; + rest_req = ss.str(); + g_test->send_request(string("POST"), rest_req, read_dummy_post, NULL, sizeof(int)); + EXPECT_EQ(500U, g_test->get_resp_code()); + + ss.str(""); + ss << "/admin/opstate?client-id=" << cid_1 << "&op-id=" << oid_1 + << "&object=" << obj_1; + rest_req = ss.str(); + g_test->send_request(string("DELETE"), rest_req); + EXPECT_EQ(200U, g_test->get_resp_code()); + + ASSERT_EQ(caps_rm(cname, perm), 0); + ASSERT_EQ(user_rm(uid, display_name), 0); +} + +int main(int argc, char *argv[]){ + vector args; + argv_to_vec(argc, (const char **)argv, args); + + auto cct = global_init(NULL, args, CEPH_ENTITY_TYPE_CLIENT, + CODE_ENVIRONMENT_UTILITY, 0); + common_init_finish(g_ceph_context); + store = RGWStoreManager::get_storage(g_ceph_context, false, false, false, false, false); + g_test = new admin_log::test_helper(); + finisher = new Finisher(g_ceph_context); +#ifdef GTEST + ::testing::InitGoogleTest(&argc, argv); +#endif + finisher->start(); + + if(g_test->extract_input(argc, argv) < 0){ + print_usage(argv[0]); + return -1; + } +#ifdef GTEST + int r = RUN_ALL_TESTS(); + if (r >= 0) { + cout << "There are no failures in the test case\n"; + } else { + cout << "There are some failures\n"; + } +#endif + finisher->stop(); + return 0; +}