redis_test.go 22 KB

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