redis_test.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876
  1. package redis
  2. import (
  3. "errors"
  4. "io"
  5. "testing"
  6. "time"
  7. "github.com/alicebob/miniredis/v2"
  8. red "github.com/go-redis/redis"
  9. "github.com/stretchr/testify/assert"
  10. )
  11. func TestRedis_Exists(t *testing.T) {
  12. runOnRedis(t, func(client *Redis) {
  13. _, err := NewRedis(client.Addr, "").Exists("a")
  14. assert.NotNil(t, err)
  15. ok, err := client.Exists("a")
  16. assert.Nil(t, err)
  17. assert.False(t, ok)
  18. assert.Nil(t, client.Set("a", "b"))
  19. ok, err = client.Exists("a")
  20. assert.Nil(t, err)
  21. assert.True(t, ok)
  22. })
  23. }
  24. func TestRedis_Eval(t *testing.T) {
  25. runOnRedis(t, func(client *Redis) {
  26. _, err := NewRedis(client.Addr, "").Eval(`redis.call("EXISTS", KEYS[1])`, []string{"notexist"})
  27. assert.NotNil(t, err)
  28. _, err = client.Eval(`redis.call("EXISTS", KEYS[1])`, []string{"notexist"})
  29. assert.Equal(t, Nil, err)
  30. err = client.Set("key1", "value1")
  31. assert.Nil(t, err)
  32. _, err = client.Eval(`redis.call("EXISTS", KEYS[1])`, []string{"key1"})
  33. assert.Equal(t, Nil, err)
  34. val, err := client.Eval(`return redis.call("EXISTS", KEYS[1])`, []string{"key1"})
  35. assert.Nil(t, err)
  36. assert.Equal(t, int64(1), val)
  37. })
  38. }
  39. func TestRedis_Hgetall(t *testing.T) {
  40. runOnRedis(t, func(client *Redis) {
  41. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  42. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  43. _, err := NewRedis(client.Addr, "").Hgetall("a")
  44. assert.NotNil(t, err)
  45. vals, err := client.Hgetall("a")
  46. assert.Nil(t, err)
  47. assert.EqualValues(t, map[string]string{
  48. "aa": "aaa",
  49. "bb": "bbb",
  50. }, vals)
  51. })
  52. }
  53. func TestRedis_Hvals(t *testing.T) {
  54. runOnRedis(t, func(client *Redis) {
  55. assert.NotNil(t, NewRedis(client.Addr, "").Hset("a", "aa", "aaa"))
  56. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  57. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  58. _, err := NewRedis(client.Addr, "").Hvals("a")
  59. assert.NotNil(t, err)
  60. vals, err := client.Hvals("a")
  61. assert.Nil(t, err)
  62. assert.ElementsMatch(t, []string{"aaa", "bbb"}, vals)
  63. })
  64. }
  65. func TestRedis_Hsetnx(t *testing.T) {
  66. runOnRedis(t, func(client *Redis) {
  67. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  68. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  69. _, err := NewRedis(client.Addr, "").Hsetnx("a", "bb", "ccc")
  70. assert.NotNil(t, err)
  71. ok, err := client.Hsetnx("a", "bb", "ccc")
  72. assert.Nil(t, err)
  73. assert.False(t, ok)
  74. ok, err = client.Hsetnx("a", "dd", "ddd")
  75. assert.Nil(t, err)
  76. assert.True(t, ok)
  77. vals, err := client.Hvals("a")
  78. assert.Nil(t, err)
  79. assert.ElementsMatch(t, []string{"aaa", "bbb", "ddd"}, vals)
  80. })
  81. }
  82. func TestRedis_HdelHlen(t *testing.T) {
  83. runOnRedis(t, func(client *Redis) {
  84. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  85. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  86. _, err := NewRedis(client.Addr, "").Hlen("a")
  87. assert.NotNil(t, err)
  88. num, err := client.Hlen("a")
  89. assert.Nil(t, err)
  90. assert.Equal(t, 2, num)
  91. val, err := client.Hdel("a", "aa")
  92. assert.Nil(t, err)
  93. assert.True(t, val)
  94. vals, err := client.Hvals("a")
  95. assert.Nil(t, err)
  96. assert.ElementsMatch(t, []string{"bbb"}, vals)
  97. })
  98. }
  99. func TestRedis_HIncrBy(t *testing.T) {
  100. runOnRedis(t, func(client *Redis) {
  101. _, err := NewRedis(client.Addr, "").Hincrby("key", "field", 2)
  102. assert.NotNil(t, err)
  103. val, err := client.Hincrby("key", "field", 2)
  104. assert.Nil(t, err)
  105. assert.Equal(t, 2, val)
  106. val, err = client.Hincrby("key", "field", 3)
  107. assert.Nil(t, err)
  108. assert.Equal(t, 5, val)
  109. })
  110. }
  111. func TestRedis_Hkeys(t *testing.T) {
  112. runOnRedis(t, func(client *Redis) {
  113. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  114. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  115. _, err := NewRedis(client.Addr, "").Hkeys("a")
  116. assert.NotNil(t, err)
  117. vals, err := client.Hkeys("a")
  118. assert.Nil(t, err)
  119. assert.ElementsMatch(t, []string{"aa", "bb"}, vals)
  120. })
  121. }
  122. func TestRedis_Hmget(t *testing.T) {
  123. runOnRedis(t, func(client *Redis) {
  124. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  125. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  126. _, err := NewRedis(client.Addr, "").Hmget("a", "aa", "bb")
  127. assert.NotNil(t, err)
  128. vals, err := client.Hmget("a", "aa", "bb")
  129. assert.Nil(t, err)
  130. assert.EqualValues(t, []string{"aaa", "bbb"}, vals)
  131. vals, err = client.Hmget("a", "aa", "no", "bb")
  132. assert.Nil(t, err)
  133. assert.EqualValues(t, []string{"aaa", "", "bbb"}, vals)
  134. })
  135. }
  136. func TestRedis_Hmset(t *testing.T) {
  137. runOnRedis(t, func(client *Redis) {
  138. assert.NotNil(t, NewRedis(client.Addr, "").Hmset("a", nil))
  139. assert.Nil(t, client.Hmset("a", map[string]string{
  140. "aa": "aaa",
  141. "bb": "bbb",
  142. }))
  143. vals, err := client.Hmget("a", "aa", "bb")
  144. assert.Nil(t, err)
  145. assert.EqualValues(t, []string{"aaa", "bbb"}, vals)
  146. })
  147. }
  148. func TestRedis_Incr(t *testing.T) {
  149. runOnRedis(t, func(client *Redis) {
  150. _, err := NewRedis(client.Addr, "").Incr("a")
  151. assert.NotNil(t, err)
  152. val, err := client.Incr("a")
  153. assert.Nil(t, err)
  154. assert.Equal(t, int64(1), val)
  155. val, err = client.Incr("a")
  156. assert.Nil(t, err)
  157. assert.Equal(t, int64(2), val)
  158. })
  159. }
  160. func TestRedis_IncrBy(t *testing.T) {
  161. runOnRedis(t, func(client *Redis) {
  162. _, err := NewRedis(client.Addr, "").Incrby("a", 2)
  163. assert.NotNil(t, err)
  164. val, err := client.Incrby("a", 2)
  165. assert.Nil(t, err)
  166. assert.Equal(t, int64(2), val)
  167. val, err = client.Incrby("a", 3)
  168. assert.Nil(t, err)
  169. assert.Equal(t, int64(5), val)
  170. })
  171. }
  172. func TestRedis_Keys(t *testing.T) {
  173. runOnRedis(t, func(client *Redis) {
  174. err := client.Set("key1", "value1")
  175. assert.Nil(t, err)
  176. err = client.Set("key2", "value2")
  177. assert.Nil(t, err)
  178. _, err = NewRedis(client.Addr, "").Keys("*")
  179. assert.NotNil(t, err)
  180. keys, err := client.Keys("*")
  181. assert.Nil(t, err)
  182. assert.ElementsMatch(t, []string{"key1", "key2"}, keys)
  183. })
  184. }
  185. func TestRedis_HyperLogLog(t *testing.T) {
  186. runOnRedis(t, func(client *Redis) {
  187. client.Ping()
  188. r := NewRedis(client.Addr, "")
  189. _, err := r.Pfadd("key1")
  190. assert.NotNil(t, err)
  191. _, err = r.Pfcount("*")
  192. assert.NotNil(t, err)
  193. err = r.Pfmerge("*")
  194. assert.NotNil(t, err)
  195. })
  196. }
  197. func TestRedis_List(t *testing.T) {
  198. runOnRedis(t, func(client *Redis) {
  199. _, err := NewRedis(client.Addr, "").Lpush("key", "value1", "value2")
  200. assert.NotNil(t, err)
  201. val, err := client.Lpush("key", "value1", "value2")
  202. assert.Nil(t, err)
  203. assert.Equal(t, 2, val)
  204. _, err = NewRedis(client.Addr, "").Rpush("key", "value3", "value4")
  205. assert.NotNil(t, err)
  206. val, err = client.Rpush("key", "value3", "value4")
  207. assert.Nil(t, err)
  208. assert.Equal(t, 4, val)
  209. _, err = NewRedis(client.Addr, "").Llen("key")
  210. assert.NotNil(t, err)
  211. val, err = client.Llen("key")
  212. assert.Nil(t, err)
  213. assert.Equal(t, 4, val)
  214. vals, err := client.Lrange("key", 0, 10)
  215. assert.Nil(t, err)
  216. assert.EqualValues(t, []string{"value2", "value1", "value3", "value4"}, vals)
  217. _, err = NewRedis(client.Addr, "").Lpop("key")
  218. assert.NotNil(t, err)
  219. v, err := client.Lpop("key")
  220. assert.Nil(t, err)
  221. assert.Equal(t, "value2", v)
  222. val, err = client.Lpush("key", "value1", "value2")
  223. assert.Nil(t, err)
  224. assert.Equal(t, 5, val)
  225. val, err = client.Rpush("key", "value3", "value3")
  226. assert.Nil(t, err)
  227. assert.Equal(t, 7, val)
  228. _, err = NewRedis(client.Addr, "").Lrem("key", 2, "value1")
  229. assert.NotNil(t, err)
  230. n, err := client.Lrem("key", 2, "value1")
  231. assert.Nil(t, err)
  232. assert.Equal(t, 2, n)
  233. _, err = NewRedis(client.Addr, "").Lrange("key", 0, 10)
  234. assert.NotNil(t, err)
  235. vals, err = client.Lrange("key", 0, 10)
  236. assert.Nil(t, err)
  237. assert.EqualValues(t, []string{"value2", "value3", "value4", "value3", "value3"}, vals)
  238. n, err = client.Lrem("key", -2, "value3")
  239. assert.Nil(t, err)
  240. assert.Equal(t, 2, n)
  241. vals, err = client.Lrange("key", 0, 10)
  242. assert.Nil(t, err)
  243. assert.EqualValues(t, []string{"value2", "value3", "value4"}, vals)
  244. })
  245. }
  246. func TestRedis_Mget(t *testing.T) {
  247. runOnRedis(t, func(client *Redis) {
  248. err := client.Set("key1", "value1")
  249. assert.Nil(t, err)
  250. err = client.Set("key2", "value2")
  251. assert.Nil(t, err)
  252. _, err = NewRedis(client.Addr, "").Mget("key1", "key0", "key2", "key3")
  253. assert.NotNil(t, err)
  254. vals, err := client.Mget("key1", "key0", "key2", "key3")
  255. assert.Nil(t, err)
  256. assert.EqualValues(t, []string{"value1", "", "value2", ""}, vals)
  257. })
  258. }
  259. func TestRedis_SetBit(t *testing.T) {
  260. runOnRedis(t, func(client *Redis) {
  261. err := NewRedis(client.Addr, "").SetBit("key", 1, 1)
  262. assert.NotNil(t, err)
  263. err = client.SetBit("key", 1, 1)
  264. assert.Nil(t, err)
  265. })
  266. }
  267. func TestRedis_GetBit(t *testing.T) {
  268. runOnRedis(t, func(client *Redis) {
  269. err := client.SetBit("key", 2, 1)
  270. assert.Nil(t, err)
  271. _, err = NewRedis(client.Addr, "").GetBit("key", 2)
  272. assert.NotNil(t, err)
  273. val, err := client.GetBit("key", 2)
  274. assert.Nil(t, err)
  275. assert.Equal(t, 1, val)
  276. })
  277. }
  278. func TestRedis_Persist(t *testing.T) {
  279. runOnRedis(t, func(client *Redis) {
  280. _, err := NewRedis(client.Addr, "").Persist("key")
  281. assert.NotNil(t, err)
  282. ok, err := client.Persist("key")
  283. assert.Nil(t, err)
  284. assert.False(t, ok)
  285. err = client.Set("key", "value")
  286. assert.Nil(t, err)
  287. ok, err = client.Persist("key")
  288. assert.Nil(t, err)
  289. assert.False(t, ok)
  290. err = NewRedis(client.Addr, "").Expire("key", 5)
  291. assert.NotNil(t, err)
  292. err = client.Expire("key", 5)
  293. assert.Nil(t, err)
  294. ok, err = client.Persist("key")
  295. assert.Nil(t, err)
  296. assert.True(t, ok)
  297. err = NewRedis(client.Addr, "").Expireat("key", time.Now().Unix()+5)
  298. assert.NotNil(t, err)
  299. err = client.Expireat("key", time.Now().Unix()+5)
  300. assert.Nil(t, err)
  301. ok, err = client.Persist("key")
  302. assert.Nil(t, err)
  303. assert.True(t, ok)
  304. })
  305. }
  306. func TestRedis_Ping(t *testing.T) {
  307. runOnRedis(t, func(client *Redis) {
  308. ok := client.Ping()
  309. assert.True(t, ok)
  310. })
  311. }
  312. func TestRedis_Scan(t *testing.T) {
  313. runOnRedis(t, func(client *Redis) {
  314. err := client.Set("key1", "value1")
  315. assert.Nil(t, err)
  316. err = client.Set("key2", "value2")
  317. assert.Nil(t, err)
  318. _, _, err = NewRedis(client.Addr, "").Scan(0, "*", 100)
  319. assert.NotNil(t, err)
  320. keys, _, err := client.Scan(0, "*", 100)
  321. assert.Nil(t, err)
  322. assert.ElementsMatch(t, []string{"key1", "key2"}, keys)
  323. })
  324. }
  325. func TestRedis_Sscan(t *testing.T) {
  326. runOnRedis(t, func(client *Redis) {
  327. key := "list"
  328. var list []string
  329. for i := 0; i < 1550; i++ {
  330. list = append(list, randomStr(i))
  331. }
  332. lens, err := client.Sadd(key, list)
  333. assert.Nil(t, err)
  334. assert.Equal(t, lens, 1550)
  335. var cursor uint64 = 0
  336. sum := 0
  337. for {
  338. _, _, err := NewRedis(client.Addr, "").Sscan(key, cursor, "", 100)
  339. assert.NotNil(t, err)
  340. keys, next, err := client.Sscan(key, cursor, "", 100)
  341. assert.Nil(t, err)
  342. sum += len(keys)
  343. if next == 0 {
  344. break
  345. }
  346. cursor = next
  347. }
  348. assert.Equal(t, sum, 1550)
  349. _, err = NewRedis(client.Addr, "").Del(key)
  350. assert.NotNil(t, err)
  351. _, err = client.Del(key)
  352. assert.Nil(t, err)
  353. })
  354. }
  355. func TestRedis_Set(t *testing.T) {
  356. runOnRedis(t, func(client *Redis) {
  357. _, err := NewRedis(client.Addr, "").Sadd("key", 1, 2, 3, 4)
  358. assert.NotNil(t, err)
  359. num, err := client.Sadd("key", 1, 2, 3, 4)
  360. assert.Nil(t, err)
  361. assert.Equal(t, 4, num)
  362. _, err = NewRedis(client.Addr, "").Scard("key")
  363. assert.NotNil(t, err)
  364. val, err := client.Scard("key")
  365. assert.Nil(t, err)
  366. assert.Equal(t, int64(4), val)
  367. _, err = NewRedis(client.Addr, "").Sismember("key", 2)
  368. assert.NotNil(t, err)
  369. ok, err := client.Sismember("key", 2)
  370. assert.Nil(t, err)
  371. assert.True(t, ok)
  372. _, err = NewRedis(client.Addr, "").Srem("key", 3, 4)
  373. assert.NotNil(t, err)
  374. num, err = client.Srem("key", 3, 4)
  375. assert.Nil(t, err)
  376. assert.Equal(t, 2, num)
  377. _, err = NewRedis(client.Addr, "").Smembers("key")
  378. assert.NotNil(t, err)
  379. vals, err := client.Smembers("key")
  380. assert.Nil(t, err)
  381. assert.ElementsMatch(t, []string{"1", "2"}, vals)
  382. _, err = NewRedis(client.Addr, "").Srandmember("key", 1)
  383. assert.NotNil(t, err)
  384. members, err := client.Srandmember("key", 1)
  385. assert.Nil(t, err)
  386. assert.Len(t, members, 1)
  387. assert.Contains(t, []string{"1", "2"}, members[0])
  388. _, err = NewRedis(client.Addr, "").Spop("key")
  389. assert.NotNil(t, err)
  390. member, err := client.Spop("key")
  391. assert.Nil(t, err)
  392. assert.Contains(t, []string{"1", "2"}, member)
  393. _, err = NewRedis(client.Addr, "").Smembers("key")
  394. assert.NotNil(t, err)
  395. vals, err = client.Smembers("key")
  396. assert.Nil(t, err)
  397. assert.NotContains(t, vals, member)
  398. _, err = NewRedis(client.Addr, "").Sadd("key1", 1, 2, 3, 4)
  399. assert.NotNil(t, err)
  400. num, err = client.Sadd("key1", 1, 2, 3, 4)
  401. assert.Nil(t, err)
  402. assert.Equal(t, 4, num)
  403. num, err = client.Sadd("key2", 2, 3, 4, 5)
  404. assert.Nil(t, err)
  405. assert.Equal(t, 4, num)
  406. _, err = NewRedis(client.Addr, "").Sunion("key1", "key2")
  407. assert.NotNil(t, err)
  408. vals, err = client.Sunion("key1", "key2")
  409. assert.Nil(t, err)
  410. assert.ElementsMatch(t, []string{"1", "2", "3", "4", "5"}, vals)
  411. _, err = NewRedis(client.Addr, "").Sunionstore("key3", "key1", "key2")
  412. assert.NotNil(t, err)
  413. num, err = client.Sunionstore("key3", "key1", "key2")
  414. assert.Nil(t, err)
  415. assert.Equal(t, 5, num)
  416. _, err = NewRedis(client.Addr, "").Sdiff("key1", "key2")
  417. assert.NotNil(t, err)
  418. vals, err = client.Sdiff("key1", "key2")
  419. assert.Nil(t, err)
  420. assert.EqualValues(t, []string{"1"}, vals)
  421. _, err = NewRedis(client.Addr, "").Sdiffstore("key4", "key1", "key2")
  422. assert.NotNil(t, err)
  423. num, err = client.Sdiffstore("key4", "key1", "key2")
  424. assert.Nil(t, err)
  425. assert.Equal(t, 1, num)
  426. })
  427. }
  428. func TestRedis_SetGetDel(t *testing.T) {
  429. runOnRedis(t, func(client *Redis) {
  430. err := NewRedis(client.Addr, "").Set("hello", "world")
  431. assert.NotNil(t, err)
  432. err = client.Set("hello", "world")
  433. assert.Nil(t, err)
  434. _, err = NewRedis(client.Addr, "").Get("hello")
  435. assert.NotNil(t, err)
  436. val, err := client.Get("hello")
  437. assert.Nil(t, err)
  438. assert.Equal(t, "world", val)
  439. ret, err := client.Del("hello")
  440. assert.Nil(t, err)
  441. assert.Equal(t, 1, ret)
  442. })
  443. }
  444. func TestRedis_SetExNx(t *testing.T) {
  445. runOnRedis(t, func(client *Redis) {
  446. err := NewRedis(client.Addr, "").Setex("hello", "world", 5)
  447. assert.NotNil(t, err)
  448. err = client.Setex("hello", "world", 5)
  449. assert.Nil(t, err)
  450. _, err = NewRedis(client.Addr, "").Setnx("hello", "newworld")
  451. assert.NotNil(t, err)
  452. ok, err := client.Setnx("hello", "newworld")
  453. assert.Nil(t, err)
  454. assert.False(t, ok)
  455. ok, err = client.Setnx("newhello", "newworld")
  456. assert.Nil(t, err)
  457. assert.True(t, ok)
  458. val, err := client.Get("hello")
  459. assert.Nil(t, err)
  460. assert.Equal(t, "world", val)
  461. val, err = client.Get("newhello")
  462. assert.Nil(t, err)
  463. assert.Equal(t, "newworld", val)
  464. ttl, err := client.Ttl("hello")
  465. assert.Nil(t, err)
  466. assert.True(t, ttl > 0)
  467. _, err = NewRedis(client.Addr, "").SetnxEx("newhello", "newworld", 5)
  468. assert.NotNil(t, err)
  469. ok, err = client.SetnxEx("newhello", "newworld", 5)
  470. assert.Nil(t, err)
  471. assert.False(t, ok)
  472. num, err := client.Del("newhello")
  473. assert.Nil(t, err)
  474. assert.Equal(t, 1, num)
  475. ok, err = client.SetnxEx("newhello", "newworld", 5)
  476. assert.Nil(t, err)
  477. assert.True(t, ok)
  478. val, err = client.Get("newhello")
  479. assert.Nil(t, err)
  480. assert.Equal(t, "newworld", val)
  481. })
  482. }
  483. func TestRedis_SetGetDelHashField(t *testing.T) {
  484. runOnRedis(t, func(client *Redis) {
  485. err := client.Hset("key", "field", "value")
  486. assert.Nil(t, err)
  487. _, err = NewRedis(client.Addr, "").Hget("key", "field")
  488. assert.NotNil(t, err)
  489. val, err := client.Hget("key", "field")
  490. assert.Nil(t, err)
  491. assert.Equal(t, "value", val)
  492. _, err = NewRedis(client.Addr, "").Hexists("key", "field")
  493. assert.NotNil(t, err)
  494. ok, err := client.Hexists("key", "field")
  495. assert.Nil(t, err)
  496. assert.True(t, ok)
  497. _, err = NewRedis(client.Addr, "").Hdel("key", "field")
  498. assert.NotNil(t, err)
  499. ret, err := client.Hdel("key", "field")
  500. assert.Nil(t, err)
  501. assert.True(t, ret)
  502. ok, err = client.Hexists("key", "field")
  503. assert.Nil(t, err)
  504. assert.False(t, ok)
  505. })
  506. }
  507. func TestRedis_SortedSet(t *testing.T) {
  508. runOnRedis(t, func(client *Redis) {
  509. ok, err := client.Zadd("key", 1, "value1")
  510. assert.Nil(t, err)
  511. assert.True(t, ok)
  512. ok, err = client.Zadd("key", 2, "value1")
  513. assert.Nil(t, err)
  514. assert.False(t, ok)
  515. val, err := client.Zscore("key", "value1")
  516. assert.Nil(t, err)
  517. assert.Equal(t, int64(2), val)
  518. _, err = NewRedis(client.Addr, "").Zincrby("key", 3, "value1")
  519. assert.NotNil(t, err)
  520. val, err = client.Zincrby("key", 3, "value1")
  521. assert.Nil(t, err)
  522. assert.Equal(t, int64(5), val)
  523. _, err = NewRedis(client.Addr, "").Zscore("key", "value1")
  524. assert.NotNil(t, err)
  525. val, err = client.Zscore("key", "value1")
  526. assert.Nil(t, err)
  527. assert.Equal(t, int64(5), val)
  528. val, err = NewRedis(client.Addr, "").Zadds("key")
  529. assert.NotNil(t, err)
  530. val, err = client.Zadds("key", Pair{
  531. Key: "value2",
  532. Score: 6,
  533. }, Pair{
  534. Key: "value3",
  535. Score: 7,
  536. })
  537. assert.Nil(t, err)
  538. assert.Equal(t, int64(2), val)
  539. pairs, err := NewRedis(client.Addr, "").ZRevRangeWithScores("key", 1, 3)
  540. assert.NotNil(t, err)
  541. pairs, err = client.ZRevRangeWithScores("key", 1, 3)
  542. assert.Nil(t, err)
  543. assert.EqualValues(t, []Pair{
  544. {
  545. Key: "value2",
  546. Score: 6,
  547. },
  548. {
  549. Key: "value1",
  550. Score: 5,
  551. },
  552. }, pairs)
  553. rank, err := client.Zrank("key", "value2")
  554. assert.Nil(t, err)
  555. assert.Equal(t, int64(1), rank)
  556. rank, err = client.Zrevrank("key", "value1")
  557. assert.Nil(t, err)
  558. assert.Equal(t, int64(2), rank)
  559. _, err = NewRedis(client.Addr, "").Zrank("key", "value4")
  560. assert.NotNil(t, err)
  561. _, err = client.Zrank("key", "value4")
  562. assert.Equal(t, Nil, err)
  563. _, err = NewRedis(client.Addr, "").Zrem("key", "value2", "value3")
  564. assert.NotNil(t, err)
  565. num, err := client.Zrem("key", "value2", "value3")
  566. assert.Nil(t, err)
  567. assert.Equal(t, 2, num)
  568. ok, err = client.Zadd("key", 6, "value2")
  569. assert.Nil(t, err)
  570. assert.True(t, ok)
  571. ok, err = client.Zadd("key", 7, "value3")
  572. assert.Nil(t, err)
  573. assert.True(t, ok)
  574. ok, err = client.Zadd("key", 8, "value4")
  575. assert.Nil(t, err)
  576. assert.True(t, ok)
  577. _, err = NewRedis(client.Addr, "").Zremrangebyscore("key", 6, 7)
  578. assert.NotNil(t, err)
  579. num, err = client.Zremrangebyscore("key", 6, 7)
  580. assert.Nil(t, err)
  581. assert.Equal(t, 2, num)
  582. ok, err = client.Zadd("key", 6, "value2")
  583. assert.Nil(t, err)
  584. assert.True(t, ok)
  585. _, err = NewRedis(client.Addr, "").Zadd("key", 7, "value3")
  586. assert.NotNil(t, err)
  587. ok, err = client.Zadd("key", 7, "value3")
  588. assert.Nil(t, err)
  589. assert.True(t, ok)
  590. _, err = NewRedis(client.Addr, "").Zcount("key", 6, 7)
  591. assert.NotNil(t, err)
  592. num, err = client.Zcount("key", 6, 7)
  593. assert.Nil(t, err)
  594. assert.Equal(t, 2, num)
  595. _, err = NewRedis(client.Addr, "").Zremrangebyrank("key", 1, 2)
  596. assert.NotNil(t, err)
  597. num, err = client.Zremrangebyrank("key", 1, 2)
  598. assert.Nil(t, err)
  599. assert.Equal(t, 2, num)
  600. _, err = NewRedis(client.Addr, "").Zcard("key")
  601. assert.NotNil(t, err)
  602. card, err := client.Zcard("key")
  603. assert.Nil(t, err)
  604. assert.Equal(t, 2, card)
  605. _, err = NewRedis(client.Addr, "").Zrange("key", 0, -1)
  606. assert.NotNil(t, err)
  607. vals, err := client.Zrange("key", 0, -1)
  608. assert.Nil(t, err)
  609. assert.EqualValues(t, []string{"value1", "value4"}, vals)
  610. _, err = NewRedis(client.Addr, "").Zrevrange("key", 0, -1)
  611. assert.NotNil(t, err)
  612. vals, err = client.Zrevrange("key", 0, -1)
  613. assert.Nil(t, err)
  614. assert.EqualValues(t, []string{"value4", "value1"}, vals)
  615. _, err = NewRedis(client.Addr, "").ZrangeWithScores("key", 0, -1)
  616. assert.NotNil(t, err)
  617. pairs, err = client.ZrangeWithScores("key", 0, -1)
  618. assert.Nil(t, err)
  619. assert.EqualValues(t, []Pair{
  620. {
  621. Key: "value1",
  622. Score: 5,
  623. },
  624. {
  625. Key: "value4",
  626. Score: 8,
  627. },
  628. }, pairs)
  629. _, err = NewRedis(client.Addr, "").ZrangebyscoreWithScores("key", 5, 8)
  630. assert.NotNil(t, err)
  631. pairs, err = client.ZrangebyscoreWithScores("key", 5, 8)
  632. assert.Nil(t, err)
  633. assert.EqualValues(t, []Pair{
  634. {
  635. Key: "value1",
  636. Score: 5,
  637. },
  638. {
  639. Key: "value4",
  640. Score: 8,
  641. },
  642. }, pairs)
  643. _, err = NewRedis(client.Addr, "").ZrangebyscoreWithScoresAndLimit(
  644. "key", 5, 8, 1, 1)
  645. assert.NotNil(t, err)
  646. pairs, err = client.ZrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 1)
  647. assert.Nil(t, err)
  648. assert.EqualValues(t, []Pair{
  649. {
  650. Key: "value4",
  651. Score: 8,
  652. },
  653. }, pairs)
  654. pairs, err = client.ZrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 0)
  655. assert.Nil(t, err)
  656. assert.Equal(t, 0, len(pairs))
  657. _, err = NewRedis(client.Addr, "").ZrevrangebyscoreWithScores("key", 5, 8)
  658. assert.NotNil(t, err)
  659. pairs, err = client.ZrevrangebyscoreWithScores("key", 5, 8)
  660. assert.Nil(t, err)
  661. assert.EqualValues(t, []Pair{
  662. {
  663. Key: "value4",
  664. Score: 8,
  665. },
  666. {
  667. Key: "value1",
  668. Score: 5,
  669. },
  670. }, pairs)
  671. _, err = NewRedis(client.Addr, "").ZrevrangebyscoreWithScoresAndLimit(
  672. "key", 5, 8, 1, 1)
  673. assert.NotNil(t, err)
  674. pairs, err = client.ZrevrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 1)
  675. assert.Nil(t, err)
  676. assert.EqualValues(t, []Pair{
  677. {
  678. Key: "value1",
  679. Score: 5,
  680. },
  681. }, pairs)
  682. pairs, err = client.ZrevrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 0)
  683. assert.Nil(t, err)
  684. assert.Equal(t, 0, len(pairs))
  685. _, err = NewRedis(client.Addr, "").Zrevrank("key", "value")
  686. assert.NotNil(t, err)
  687. })
  688. }
  689. func TestRedis_Pipelined(t *testing.T) {
  690. runOnRedis(t, func(client *Redis) {
  691. assert.NotNil(t, NewRedis(client.Addr, "").Pipelined(func(pipeliner Pipeliner) error {
  692. return nil
  693. }))
  694. err := client.Pipelined(
  695. func(pipe Pipeliner) error {
  696. pipe.Incr("pipelined_counter")
  697. pipe.Expire("pipelined_counter", time.Hour)
  698. pipe.ZAdd("zadd", Z{Score: 12, Member: "zadd"})
  699. return nil
  700. },
  701. )
  702. assert.Nil(t, err)
  703. _, err = NewRedis(client.Addr, "").Ttl("pipelined_counter")
  704. assert.NotNil(t, err)
  705. ttl, err := client.Ttl("pipelined_counter")
  706. assert.Nil(t, err)
  707. assert.Equal(t, 3600, ttl)
  708. value, err := client.Get("pipelined_counter")
  709. assert.Nil(t, err)
  710. assert.Equal(t, "1", value)
  711. score, err := client.Zscore("zadd", "zadd")
  712. assert.Nil(t, err)
  713. assert.Equal(t, int64(12), score)
  714. })
  715. }
  716. func TestRedisString(t *testing.T) {
  717. runOnRedis(t, func(client *Redis) {
  718. client.Ping()
  719. _, err := getRedis(NewRedis(client.Addr, ClusterType))
  720. assert.Nil(t, err)
  721. assert.Equal(t, client.Addr, client.String())
  722. assert.NotNil(t, NewRedis(client.Addr, "").Ping())
  723. })
  724. }
  725. func TestRedisScriptLoad(t *testing.T) {
  726. runOnRedis(t, func(client *Redis) {
  727. client.Ping()
  728. _, err := NewRedis(client.Addr, "").scriptLoad("foo")
  729. assert.NotNil(t, err)
  730. _, err = client.scriptLoad("foo")
  731. assert.NotNil(t, err)
  732. })
  733. }
  734. func TestRedisToPairs(t *testing.T) {
  735. pairs := toPairs([]red.Z{
  736. {
  737. Member: 1,
  738. Score: 1,
  739. },
  740. {
  741. Member: 2,
  742. Score: 2,
  743. },
  744. })
  745. assert.EqualValues(t, []Pair{
  746. {
  747. Key: "1",
  748. Score: 1,
  749. },
  750. {
  751. Key: "2",
  752. Score: 2,
  753. },
  754. }, pairs)
  755. }
  756. func TestRedisToStrings(t *testing.T) {
  757. vals := toStrings([]interface{}{1, 2})
  758. assert.EqualValues(t, []string{"1", "2"}, vals)
  759. }
  760. func TestRedisBlpop(t *testing.T) {
  761. runOnRedis(t, func(client *Redis) {
  762. client.Ping()
  763. var node mockedNode
  764. _, err := client.Blpop(nil, "foo")
  765. assert.NotNil(t, err)
  766. _, err = client.Blpop(node, "foo")
  767. assert.NotNil(t, err)
  768. })
  769. }
  770. func TestRedisBlpopEx(t *testing.T) {
  771. runOnRedis(t, func(client *Redis) {
  772. client.Ping()
  773. var node mockedNode
  774. _, _, err := client.BlpopEx(nil, "foo")
  775. assert.NotNil(t, err)
  776. _, _, err = client.BlpopEx(node, "foo")
  777. assert.NotNil(t, err)
  778. })
  779. }
  780. func TestRedisGeo(t *testing.T) {
  781. runOnRedis(t, func(client *Redis) {
  782. client.Ping()
  783. var geoLocation = []*GeoLocation{{Longitude: 13.361389, Latitude: 38.115556, Name: "Palermo"}, {Longitude: 15.087269, Latitude: 37.502669, Name: "Catania"}}
  784. v, err := client.GeoAdd("sicily", geoLocation...)
  785. assert.Nil(t, err)
  786. assert.Equal(t, int64(2), v)
  787. v2, err := client.GeoDist("sicily", "Palermo", "Catania", "m")
  788. assert.Nil(t, err)
  789. assert.Equal(t, 166274, int(v2))
  790. // GeoHash not support
  791. v3, err := client.GeoPos("sicily", "Palermo", "Catania")
  792. assert.Nil(t, err)
  793. assert.Equal(t, int64(v3[0].Longitude), int64(13))
  794. assert.Equal(t, int64(v3[0].Latitude), int64(38))
  795. assert.Equal(t, int64(v3[1].Longitude), int64(15))
  796. assert.Equal(t, int64(v3[1].Latitude), int64(37))
  797. v4, err := client.GeoRadius("sicily", 15, 37, &red.GeoRadiusQuery{WithDist: true, Unit: "km", Radius: 200})
  798. assert.Nil(t, err)
  799. assert.Equal(t, int64(v4[0].Dist), int64(190))
  800. assert.Equal(t, int64(v4[1].Dist), int64(56))
  801. var geoLocation2 = []*GeoLocation{{Longitude: 13.583333, Latitude: 37.316667, Name: "Agrigento"}}
  802. v5, err := client.GeoAdd("sicily", geoLocation2...)
  803. assert.Nil(t, err)
  804. assert.Equal(t, int64(1), v5)
  805. v6, err := client.GeoRadiusByMember("sicily", "Agrigento", &red.GeoRadiusQuery{Unit: "km", Radius: 100})
  806. assert.Nil(t, err)
  807. assert.Equal(t, v6[0].Name, "Agrigento")
  808. assert.Equal(t, v6[1].Name, "Palermo")
  809. })
  810. }
  811. func runOnRedis(t *testing.T, fn func(client *Redis)) {
  812. s, err := miniredis.Run()
  813. assert.Nil(t, err)
  814. defer func() {
  815. client, err := clientManager.GetResource(s.Addr(), func() (io.Closer, error) {
  816. return nil, errors.New("should already exist")
  817. })
  818. if err != nil {
  819. t.Error(err)
  820. }
  821. if client != nil {
  822. client.Close()
  823. }
  824. }()
  825. fn(NewRedis(s.Addr(), NodeType))
  826. }
  827. type mockedNode struct {
  828. RedisNode
  829. }
  830. func (n mockedNode) BLPop(timeout time.Duration, keys ...string) *red.StringSliceCmd {
  831. return red.NewStringSliceCmd("foo", "bar")
  832. }