71 lines
2.2 KiB
C
71 lines
2.2 KiB
C
|
#ifndef __TEST_UTIL_H__
|
||
|
|
#define __TEST_UTIL_H__
|
||
|
|
|
||
|
|
#include "util.h"
|
||
|
|
#include <stdarg.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
|
||
|
|
#define RMUtil_Test(f) \
|
||
|
|
if (argc < 2 || RMUtil_ArgExists(__STRING(f), argv, argc, 1)) { \
|
||
|
|
int rc = f(ctx); \
|
||
|
|
if (rc != REDISMODULE_OK) { \
|
||
|
|
RedisModule_ReplyWithError(ctx, "Test " __STRING(f) " FAILED"); \
|
||
|
|
return REDISMODULE_ERR;\
|
||
|
|
}\
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
#define RMUtil_Assert(expr) if (!(expr)) { fprintf (stderr, "Assertion '%s' Failed\n", __STRING(expr)); return REDISMODULE_ERR; }
|
||
|
|
|
||
|
|
#define RMUtil_AssertNullReply(rep) RMUtil_Assert( \
|
||
|
|
RedisModule_CallReplyType(rep) == REDISMODULE_REPLY_NULL || RedisModule_CreateStringFromCallReply(rep) == NULL)
|
||
|
|
|
||
|
|
#define RMUtil_AssertReplyEquals(rep, cstr) RMUtil_Assert( \
|
||
|
|
RMUtil_StringEquals(RedisModule_CreateStringFromCallReply(rep), RedisModule_CreateString(ctx, cstr, strlen(cstr))) \
|
||
|
|
)
|
||
|
|
#
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Create an arg list to pass to a redis command handler manually, based on the format in fmt.
|
||
|
|
* The accepted format specifiers are:
|
||
|
|
* c - for null terminated c strings
|
||
|
|
* s - for RedisModuleString* objects
|
||
|
|
* l - for longs
|
||
|
|
*
|
||
|
|
* Example: RMUtil_MakeArgs(ctx, &argc, "clc", "hello", 1337, "world");
|
||
|
|
*
|
||
|
|
* Returns an array of RedisModuleString pointers. The size of the array is store in argcp
|
||
|
|
*/
|
||
|
|
RedisModuleString **RMUtil_MakeArgs(RedisModuleCtx *ctx, int *argcp, const char *fmt, ...) {
|
||
|
|
|
||
|
|
va_list ap;
|
||
|
|
va_start(ap, fmt);
|
||
|
|
RedisModuleString **argv = calloc(strlen(fmt), sizeof(RedisModuleString*));
|
||
|
|
int argc = 0;
|
||
|
|
const char *p = fmt;
|
||
|
|
while(*p) {
|
||
|
|
if (*p == 'c') {
|
||
|
|
char *cstr = va_arg(ap,char*);
|
||
|
|
argv[argc++] = RedisModule_CreateString(ctx, cstr, strlen(cstr));
|
||
|
|
} else if (*p == 's') {
|
||
|
|
argv[argc++] = va_arg(ap,void*);;
|
||
|
|
} else if (*p == 'l') {
|
||
|
|
long ll = va_arg(ap,long long);
|
||
|
|
argv[argc++] = RedisModule_CreateStringFromLongLong(ctx, ll);
|
||
|
|
} else {
|
||
|
|
goto fmterr;
|
||
|
|
}
|
||
|
|
p++;
|
||
|
|
}
|
||
|
|
*argcp = argc;
|
||
|
|
|
||
|
|
return argv;
|
||
|
|
fmterr:
|
||
|
|
free(argv);
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|