return setPubStringCommon(ctx, &setParams, &pubParams);
}
+int SetMPub_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc)
+{
+ if (argc < 7 || (argc % 2) == 0)
+ return RedisModule_WrongArity(ctx);
+
+ long long setPairsCount, pubPairsCount;
+ RedisModule_StringToLongLong(argv[1], &setPairsCount);
+ RedisModule_StringToLongLong(argv[2], &pubPairsCount);
+ if (setPairsCount < 1 || pubPairsCount < 1)
+ return RedisModule_ReplyWithError(ctx, "ERR SET_PAIR_COUNT and PUB_PAIR_COUNT must be greater than zero");
+
+ long long setLen, pubLen;
+ setLen = 2*setPairsCount;
+ pubLen = 2*pubPairsCount;
+ if (setLen + pubLen + 3 != argc)
+ return RedisModule_ReplyWithError(ctx, "ERR SET_PAIR_COUNT or PUB_PAIR_COUNT do not match the total pair count");
+
+ SetParams setParams = {
+ .key_val_pairs = argv + 3,
+ .length = setLen
+ };
+ PubParams pubParams = {
+ .channel_msg_pairs = argv + 3 + setParams.length,
+ .length = pubLen
+ };
+
+ return setPubStringCommon(ctx, &setParams, &pubParams);
+}
+
int setIENEPubStringCommon(RedisModuleCtx *ctx, RedisModuleString **argv, int argc, int flag)
{
if (argc < 6 || (argc % 2) != 0)
SetPub_RedisCommand,"write deny-oom",1,1,1) == REDISMODULE_ERR)
return REDISMODULE_ERR;
+ if (RedisModule_CreateCommand(ctx,"msetmpub",
+ SetMPub_RedisCommand,"write deny-oom pubsub",1,1,1) == REDISMODULE_ERR)
+ return REDISMODULE_ERR;
+
if (RedisModule_CreateCommand(ctx,"setiepub",
SetIEPub_RedisCommand,"write deny-oom",1,1,1) == REDISMODULE_ERR)
return REDISMODULE_ERR;
delete []redisStrVec;
}
+TEST(exstring, setmpub)
+{
+ RedisModuleCtx ctx;
+ RedisModuleString ** redisStrVec = new (RedisModuleString*[11]);
+
+ for (int i = 0 ; i < 11 ; ++i)
+ redisStrVec[i] = (RedisModuleString *)1;
+
+ mock().setData("RedisModule_OpenKey_have", 1);
+ mock().setData("RedisModule_KeyType_str", 1);
+ mock().setData("RedisModule_String_same", 1);
+ mock().setData("RedisModule_CallReplyType_null", 1);
+ mock().setData("RedisModule_StringToLongLongCallCount", 0);
+ mock().setData("RedisModule_StringToLongLongCall_1", 2);
+ mock().setData("RedisModule_StringToLongLongCall_2", 2);
+
+ int ret = SetMPub_RedisCommand(&ctx, redisStrVec, 11);
+ CHECK_EQUAL(ret, REDISMODULE_OK);
+ mock().checkExpectations();
+ delete []redisStrVec;
+}
+
TEST(exstring, setxxpub)
{
RedisModuleCtx ctx;
ret = SetPub_RedisCommand(&ctx, 0, 8);
CHECK_EQUAL(ret, REDISMODULE_ERR);
+ ret = 0;
+ ret = SetMPub_RedisCommand(&ctx, 0, 2);
+ CHECK_EQUAL(ret, REDISMODULE_ERR);
+
+ ret = 0;
+ ret = SetMPub_RedisCommand(&ctx, 0, 8);
+ CHECK_EQUAL(ret, REDISMODULE_ERR);
+
ret = 0;
ret = SetXXPub_RedisCommand(&ctx, 0, 3);
CHECK_EQUAL(ret, REDISMODULE_ERR);
}
+TEST(exstring, setmpub_command_no_key_replynull)
+{
+ RedisModuleCtx ctx;
+ RedisModuleString ** redisStrVec = new (RedisModuleString*[9]);
+
+ redisStrVec[0] = (RedisModuleString *)0;
+ redisStrVec[1] = (RedisModuleString *)1;
+ redisStrVec[2] = (RedisModuleString *)2;
+ redisStrVec[3] = (RedisModuleString *)3;
+ redisStrVec[4] = (RedisModuleString *)4;
+ redisStrVec[5] = (RedisModuleString *)5;
+ redisStrVec[6] = (RedisModuleString *)6;
+ redisStrVec[7] = (RedisModuleString *)7;
+ redisStrVec[8] = (RedisModuleString *)8;
+
+ mock().setData("RedisModule_KeyType_empty", 1);
+ mock().setData("RedisModule_CallReplyType_null", 1);
+ mock().setData("RedisModule_StringToLongLongCall_1", 1);
+ mock().setData("RedisModule_StringToLongLongCall_2", 2);
+
+ int ret = SetMPub_RedisCommand(&ctx, redisStrVec, 9);
+ CHECK_EQUAL(ret, REDISMODULE_OK);
+ mock().checkExpectations();
+ CHECK_EQUAL(0, mock().getData("GET").getIntValue());
+ CHECK_EQUAL(1, mock().getData("MSET").getIntValue());
+ CHECK_EQUAL(0, mock().getData("PUBLISH").getIntValue());
+ CHECK_EQUAL(1, mock().getData("RedisModule_FreeCallReply").getIntValue());
+
+ delete []redisStrVec;
+
+}
+
+TEST(exstring, setmpub_command_negative_key_val_count)
+{
+ RedisModuleCtx ctx;
+ RedisModuleString ** redisStrVec = new (RedisModuleString*[7]);
+
+ redisStrVec[0] = (RedisModuleString *)0;
+ redisStrVec[1] = (RedisModuleString *)1;
+ redisStrVec[2] = (RedisModuleString *)2;
+ redisStrVec[3] = (RedisModuleString *)3;
+ redisStrVec[4] = (RedisModuleString *)4;
+ redisStrVec[5] = (RedisModuleString *)5;
+ redisStrVec[6] = (RedisModuleString *)6;
+
+ mock().setData("RedisModule_KeyType_empty", 1);
+ mock().setData("RedisModule_CallReplyType_str", 1);
+ mock().setData("RedisModule_StringToLongLongCall_1", -1);
+ mock().setData("RedisModule_StringToLongLongCall_2", 1);
+
+ int ret = SetMPub_RedisCommand(&ctx, redisStrVec, 7);
+ CHECK_EQUAL(ret, REDISMODULE_OK);
+ mock().checkExpectations();
+ CHECK_EQUAL(0, mock().getData("GET").getIntValue());
+ CHECK_EQUAL(0, mock().getData("MSET").getIntValue());
+ CHECK_EQUAL(0, mock().getData("PUBLISH").getIntValue());
+ CHECK_EQUAL(1, mock().getData("RedisModule_ReplyWithError").getIntValue());
+ CHECK_EQUAL(0, mock().getData("RedisModule_FreeCallReply").getIntValue());
+
+ delete []redisStrVec;
+
+}
+
+TEST(exstring, setmpub_command_negative_chan_msg_count)
+{
+ RedisModuleCtx ctx;
+ RedisModuleString ** redisStrVec = new (RedisModuleString*[7]);
+
+ redisStrVec[0] = (RedisModuleString *)0;
+ redisStrVec[1] = (RedisModuleString *)1;
+ redisStrVec[2] = (RedisModuleString *)2;
+ redisStrVec[3] = (RedisModuleString *)3;
+ redisStrVec[4] = (RedisModuleString *)4;
+ redisStrVec[5] = (RedisModuleString *)5;
+ redisStrVec[6] = (RedisModuleString *)6;
+
+ mock().setData("RedisModule_KeyType_empty", 1);
+ mock().setData("RedisModule_CallReplyType_str", 1);
+ mock().setData("RedisModule_StringToLongLongCall_1", 1);
+ mock().setData("RedisModule_StringToLongLongCall_2", -1);
+
+ int ret = SetMPub_RedisCommand(&ctx, redisStrVec, 7);
+ CHECK_EQUAL(ret, REDISMODULE_OK);
+ mock().checkExpectations();
+ CHECK_EQUAL(0, mock().getData("GET").getIntValue());
+ CHECK_EQUAL(0, mock().getData("MSET").getIntValue());
+ CHECK_EQUAL(0, mock().getData("PUBLISH").getIntValue());
+ CHECK_EQUAL(1, mock().getData("RedisModule_ReplyWithError").getIntValue());
+ CHECK_EQUAL(0, mock().getData("RedisModule_FreeCallReply").getIntValue());
+
+ delete []redisStrVec;
+
+}
+
+TEST(exstring, setmpub_command_invalid_total_count)
+{
+ RedisModuleCtx ctx;
+ RedisModuleString ** redisStrVec = new (RedisModuleString*[7]);
+
+ redisStrVec[0] = (RedisModuleString *)0;
+ redisStrVec[1] = (RedisModuleString *)1;
+ redisStrVec[2] = (RedisModuleString *)2;
+ redisStrVec[3] = (RedisModuleString *)3;
+ redisStrVec[4] = (RedisModuleString *)4;
+ redisStrVec[5] = (RedisModuleString *)5;
+ redisStrVec[6] = (RedisModuleString *)6;
+
+ mock().setData("RedisModule_KeyType_empty", 1);
+ mock().setData("RedisModule_CallReplyType_str", 1);
+ mock().setData("RedisModule_StringToLongLongCall_1", 100);
+ mock().setData("RedisModule_StringToLongLongCall_2", 100);
+
+ int ret = SetMPub_RedisCommand(&ctx, redisStrVec, 7);
+ CHECK_EQUAL(ret, REDISMODULE_OK);
+ mock().checkExpectations();
+ CHECK_EQUAL(0, mock().getData("GET").getIntValue());
+ CHECK_EQUAL(0, mock().getData("MSET").getIntValue());
+ CHECK_EQUAL(0, mock().getData("PUBLISH").getIntValue());
+ CHECK_EQUAL(1, mock().getData("RedisModule_ReplyWithError").getIntValue());
+ CHECK_EQUAL(0, mock().getData("RedisModule_FreeCallReply").getIntValue());
+
+ delete []redisStrVec;
+
+}
+
+TEST(exstring, setmpub_command_set)
+{
+ RedisModuleCtx ctx;
+ RedisModuleString ** redisStrVec = new (RedisModuleString*[7]);
+
+ redisStrVec[0] = (RedisModuleString *)0;
+ redisStrVec[1] = (RedisModuleString *)1;
+ redisStrVec[2] = (RedisModuleString *)2;
+ redisStrVec[3] = (RedisModuleString *)3;
+ redisStrVec[4] = (RedisModuleString *)4;
+ redisStrVec[5] = (RedisModuleString *)5;
+ redisStrVec[6] = (RedisModuleString *)6;
+
+ mock().setData("RedisModule_KeyType_empty", 1);
+ mock().setData("RedisModule_CallReplyType_str", 1);
+ mock().setData("RedisModule_StringToLongLongCall_1", 1);
+ mock().setData("RedisModule_StringToLongLongCall_2", 1);
+
+ int ret = SetMPub_RedisCommand(&ctx, redisStrVec, 7);
+ CHECK_EQUAL(ret, REDISMODULE_OK);
+ mock().checkExpectations();
+ CHECK_EQUAL(0, mock().getData("GET").getIntValue());
+ CHECK_EQUAL(1, mock().getData("MSET").getIntValue());
+ CHECK_EQUAL(1, mock().getData("PUBLISH").getIntValue());
+ CHECK_EQUAL(2, mock().getData("RedisModule_FreeCallReply").getIntValue());
+
+ delete []redisStrVec;
+
+}
+
+TEST(exstring, setmpub_command_set_multipub)
+{
+ RedisModuleCtx ctx;
+ RedisModuleString ** redisStrVec = new (RedisModuleString*[9]);
+
+ redisStrVec[0] = (RedisModuleString *)0;
+ redisStrVec[1] = (RedisModuleString *)1;
+ redisStrVec[2] = (RedisModuleString *)2;
+ redisStrVec[3] = (RedisModuleString *)3;
+ redisStrVec[4] = (RedisModuleString *)4;
+ redisStrVec[5] = (RedisModuleString *)5;
+ redisStrVec[6] = (RedisModuleString *)6;
+ redisStrVec[7] = (RedisModuleString *)7;
+ redisStrVec[8] = (RedisModuleString *)8;
+
+ mock().setData("RedisModule_KeyType_empty", 1);
+ mock().setData("RedisModule_CallReplyType_str", 1);
+ mock().setData("RedisModule_StringToLongLongCall_1", 1);
+ mock().setData("RedisModule_StringToLongLongCall_2", 2);
+
+ int ret = SetMPub_RedisCommand(&ctx, redisStrVec, 9);
+ CHECK_EQUAL(ret, REDISMODULE_OK);
+ mock().checkExpectations();
+ CHECK_EQUAL(0, mock().getData("GET").getIntValue());
+ CHECK_EQUAL(1, mock().getData("MSET").getIntValue());
+ CHECK_EQUAL(2, mock().getData("PUBLISH").getIntValue());
+ CHECK_EQUAL(3, mock().getData("RedisModule_FreeCallReply").getIntValue());
+
+ delete []redisStrVec;
+
+}
+
TEST(exstring, setxxpub_command_has_no_key)
{
RedisModuleCtx ctx;