From eae0983d2d5f13e0897494f2604a0df691c1678d Mon Sep 17 00:00:00 2001
From: sundb <sundbcn@gmail.com>
Date: Fri, 18 Jun 2021 02:45:21 +0800
Subject: [PATCH] Fix leak and double free issues in datatype2 module test
 (#9102)

* Add missing call for RedisModule_DictDel in datatype2 test
* Fix memory leak in datatype2 test
---
 tests/modules/datatype2.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/modules/datatype2.c b/tests/modules/datatype2.c
index aba66b124..0c4279ae5 100644
--- a/tests/modules/datatype2.c
+++ b/tests/modules/datatype2.c
@@ -220,6 +220,8 @@ void flushdbCallback(RedisModuleCtx *ctx, RedisModuleEvent e, uint64_t sub, void
     int i;
     RedisModuleFlushInfo *fi = data;
 
+    RedisModule_AutoMemory(ctx);
+
     if (sub == REDISMODULE_SUBEVENT_FLUSHDB_START) {
         if (fi->dbnum != -1) {
            MemPoolFreeDb(ctx, fi->dbnum);
@@ -298,6 +300,7 @@ int MemFree_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc
     int nokey;
     struct MemBlock *mem = (struct MemBlock *)RedisModule_DictGet(mem_pool[RedisModule_GetSelectedDb(ctx)], argv[1], &nokey);
     if (!nokey && mem) {
+        RedisModule_DictDel(mem_pool[RedisModule_GetSelectedDb(ctx)], argv[1], NULL);
         MemBlockFree(mem);
         o->used = 0;
         o->size = 0;