redis_test.go 20 KB

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