#include #include #include #include #include #include "hiredis.h" /* The following lines make up our testing "framework" :) */ #define test(_s) { printf("#%02d ", ++tests); printf(_s); } #define test_cond(_c) if(_c) printf("PASSED\n"); else {printf("FAILED\n"); fails++;} static long long usec(void) { struct timeval tv; gettimeofday(&tv,NULL); return (((long long)tv.tv_sec)*1000000)+tv.tv_usec; } static void __connect(int *fd) { redisReply *reply = redisConnect(fd, "127.0.0.1", 6379); if (reply != NULL) { printf("Connection error: %s", reply->reply); exit(1); } } int main(void) { int fd; int i, tests = 0, fails = 0; long long t1, t2; redisReply *reply; void *reader; __connect(&fd); test("Returns I/O error when the connection is lost: "); reply = redisCommand(fd,"QUIT"); test_cond(reply->type == REDIS_PROTOCOL_ERROR && strcasecmp(reply->reply,"i/o error") == 0); freeReplyObject(reply); __connect(&fd); /* reconnect */ test("Is able to deliver commands: "); reply = redisCommand(fd,"PING"); test_cond(reply->type == REDIS_REPLY_STRING && strcasecmp(reply->reply,"pong") == 0) freeReplyObject(reply); /* Switch to DB 9 for testing, now that we know we can chat. */ reply = redisCommand(fd,"SELECT 9"); freeReplyObject(reply); /* Make sure the DB is emtpy */ reply = redisCommand(fd,"DBSIZE"); if (reply->type != REDIS_REPLY_INTEGER || reply->integer != 0) { printf("Sorry DB 9 is not empty, test can not continue\n"); exit(1); } else { printf("DB 9 is empty... test can continue\n"); } freeReplyObject(reply); test("Is a able to send commands verbatim: "); reply = redisCommand(fd,"SET foo bar"); test_cond (reply->type == REDIS_REPLY_STRING && strcasecmp(reply->reply,"ok") == 0) freeReplyObject(reply); test("%%s String interpolation works: "); reply = redisCommand(fd,"SET %s %s","foo","hello world"); freeReplyObject(reply); reply = redisCommand(fd,"GET foo"); test_cond(reply->type == REDIS_REPLY_STRING && strcmp(reply->reply,"hello world") == 0); freeReplyObject(reply); test("%%b String interpolation works: "); reply = redisCommand(fd,"SET %b %b","foo",3,"hello\x00world",11); freeReplyObject(reply); reply = redisCommand(fd,"GET foo"); test_cond(reply->type == REDIS_REPLY_STRING && memcmp(reply->reply,"hello\x00world",11) == 0) test("Binary reply length is correct: "); test_cond(sdslen(reply->reply) == 11) freeReplyObject(reply); test("Can parse nil replies: "); reply = redisCommand(fd,"GET nokey"); test_cond(reply->type == REDIS_REPLY_NIL) freeReplyObject(reply); /* test 7 */ test("Can parse integer replies: "); reply = redisCommand(fd,"INCR mycounter"); test_cond(reply->type == REDIS_REPLY_INTEGER && reply->integer == 1) freeReplyObject(reply); test("Can parse multi bulk replies: "); freeReplyObject(redisCommand(fd,"LPUSH mylist foo")); freeReplyObject(redisCommand(fd,"LPUSH mylist bar")); reply = redisCommand(fd,"LRANGE mylist 0 -1"); test_cond(reply->type == REDIS_REPLY_ARRAY && reply->elements == 2 && !memcmp(reply->element[0]->reply,"bar",3) && !memcmp(reply->element[1]->reply,"foo",3)) freeReplyObject(reply); /* m/e with multi bulk reply *before* other reply. * specifically test ordering of reply items to parse. */ test("Can handle nested multi bulk replies: "); freeReplyObject(redisCommand(fd,"MULTI")); freeReplyObject(redisCommand(fd,"LRANGE mylist 0 -1")); freeReplyObject(redisCommand(fd,"PING")); reply = (redisCommand(fd,"EXEC")); test_cond(reply->type == REDIS_REPLY_ARRAY && reply->elements == 2 && reply->element[0]->type == REDIS_REPLY_ARRAY && reply->element[0]->elements == 2 && !memcmp(reply->element[0]->element[0]->reply,"bar",3) && !memcmp(reply->element[0]->element[1]->reply,"foo",3) && reply->element[1]->type == REDIS_REPLY_STRING && strcasecmp(reply->element[1]->reply,"pong") == 0); freeReplyObject(reply); test("Error handling in reply parser: "); reader = redisCreateReplyReader(); reply = redisFeedReplyReader(reader,(char*)"@foo\r\n",6); test_cond(reply->type == REDIS_PROTOCOL_ERROR && strcasecmp(reply->reply,"protocol error, got \"@\" as reply type byte") == 0); freeReplyObject(reply); redisFreeReplyReader(reader); /* when the reply already contains multiple items, they must be free'd * on an error. valgrind will bark when this doesn't happen. */ test("Memory cleanup in reply parser: "); reader = redisCreateReplyReader(); redisFeedReplyReader(reader,(char*)"*2\r\n",4); redisFeedReplyReader(reader,(char*)"$5\r\nhello\r\n",11); reply = redisFeedReplyReader(reader,(char*)"@foo\r\n",6); test_cond(reply->type == REDIS_PROTOCOL_ERROR && strcasecmp(reply->reply,"protocol error, got \"@\" as reply type byte") == 0); freeReplyObject(reply); redisFreeReplyReader(reader); test("Throughput:\n"); for (i = 0; i < 500; i++) freeReplyObject(redisCommand(fd,"LPUSH mylist foo")); t1 = usec(); for (i = 0; i < 1000; i++) freeReplyObject(redisCommand(fd,"PING")); t2 = usec(); printf("\t(1000x PING: %.2fs)\n", (t2-t1)/1000000.0); t1 = usec(); for (i = 0; i < 1000; i++) freeReplyObject(redisCommand(fd,"LRANGE mylist 0 499")); t2 = usec(); printf("\t(1000x LRANGE with 500 elements: %.2fs)\n", (t2-t1)/1000000.0); /* Clean DB 9 */ reply = redisCommand(fd,"FLUSHDB"); freeReplyObject(reply); if (fails == 0) { printf("ALL TESTS PASSED\n"); } else { printf("*** %d TESTS FAILED ***\n", fails); } return 0; }