redis_test.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839
  1. package redis
  2. import (
  3. "errors"
  4. "io"
  5. "testing"
  6. "time"
  7. "github.com/alicebob/miniredis"
  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. _, err = NewRedis(client.Addr, "").Zrank("key", "value4")
  557. assert.NotNil(t, err)
  558. _, err = client.Zrank("key", "value4")
  559. assert.Equal(t, Nil, err)
  560. _, err = NewRedis(client.Addr, "").Zrem("key", "value2", "value3")
  561. assert.NotNil(t, err)
  562. num, err := client.Zrem("key", "value2", "value3")
  563. assert.Nil(t, err)
  564. assert.Equal(t, 2, num)
  565. ok, err = client.Zadd("key", 6, "value2")
  566. assert.Nil(t, err)
  567. assert.True(t, ok)
  568. ok, err = client.Zadd("key", 7, "value3")
  569. assert.Nil(t, err)
  570. assert.True(t, ok)
  571. ok, err = client.Zadd("key", 8, "value4")
  572. assert.Nil(t, err)
  573. assert.True(t, ok)
  574. _, err = NewRedis(client.Addr, "").Zremrangebyscore("key", 6, 7)
  575. assert.NotNil(t, err)
  576. num, err = client.Zremrangebyscore("key", 6, 7)
  577. assert.Nil(t, err)
  578. assert.Equal(t, 2, num)
  579. ok, err = client.Zadd("key", 6, "value2")
  580. assert.Nil(t, err)
  581. assert.True(t, ok)
  582. _, err = NewRedis(client.Addr, "").Zadd("key", 7, "value3")
  583. assert.NotNil(t, err)
  584. ok, err = client.Zadd("key", 7, "value3")
  585. assert.Nil(t, err)
  586. assert.True(t, ok)
  587. _, err = NewRedis(client.Addr, "").Zcount("key", 6, 7)
  588. assert.NotNil(t, err)
  589. num, err = client.Zcount("key", 6, 7)
  590. assert.Nil(t, err)
  591. assert.Equal(t, 2, num)
  592. _, err = NewRedis(client.Addr, "").Zremrangebyrank("key", 1, 2)
  593. assert.NotNil(t, err)
  594. num, err = client.Zremrangebyrank("key", 1, 2)
  595. assert.Nil(t, err)
  596. assert.Equal(t, 2, num)
  597. _, err = NewRedis(client.Addr, "").Zcard("key")
  598. assert.NotNil(t, err)
  599. card, err := client.Zcard("key")
  600. assert.Nil(t, err)
  601. assert.Equal(t, 2, card)
  602. _, err = NewRedis(client.Addr, "").Zrange("key", 0, -1)
  603. assert.NotNil(t, err)
  604. vals, err := client.Zrange("key", 0, -1)
  605. assert.Nil(t, err)
  606. assert.EqualValues(t, []string{"value1", "value4"}, vals)
  607. _, err = NewRedis(client.Addr, "").Zrevrange("key", 0, -1)
  608. assert.NotNil(t, err)
  609. vals, err = client.Zrevrange("key", 0, -1)
  610. assert.Nil(t, err)
  611. assert.EqualValues(t, []string{"value4", "value1"}, vals)
  612. _, err = NewRedis(client.Addr, "").ZrangeWithScores("key", 0, -1)
  613. assert.NotNil(t, err)
  614. pairs, err = client.ZrangeWithScores("key", 0, -1)
  615. assert.Nil(t, err)
  616. assert.EqualValues(t, []Pair{
  617. {
  618. Key: "value1",
  619. Score: 5,
  620. },
  621. {
  622. Key: "value4",
  623. Score: 8,
  624. },
  625. }, pairs)
  626. _, err = NewRedis(client.Addr, "").ZrangebyscoreWithScores("key", 5, 8)
  627. assert.NotNil(t, err)
  628. pairs, err = client.ZrangebyscoreWithScores("key", 5, 8)
  629. assert.Nil(t, err)
  630. assert.EqualValues(t, []Pair{
  631. {
  632. Key: "value1",
  633. Score: 5,
  634. },
  635. {
  636. Key: "value4",
  637. Score: 8,
  638. },
  639. }, pairs)
  640. _, err = NewRedis(client.Addr, "").ZrangebyscoreWithScoresAndLimit(
  641. "key", 5, 8, 1, 1)
  642. assert.NotNil(t, err)
  643. pairs, err = client.ZrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 1)
  644. assert.Nil(t, err)
  645. assert.EqualValues(t, []Pair{
  646. {
  647. Key: "value4",
  648. Score: 8,
  649. },
  650. }, pairs)
  651. pairs, err = client.ZrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 0)
  652. assert.Nil(t, err)
  653. assert.Equal(t, 0, len(pairs))
  654. _, err = NewRedis(client.Addr, "").ZrevrangebyscoreWithScores("key", 5, 8)
  655. assert.NotNil(t, err)
  656. pairs, err = client.ZrevrangebyscoreWithScores("key", 5, 8)
  657. assert.Nil(t, err)
  658. assert.EqualValues(t, []Pair{
  659. {
  660. Key: "value4",
  661. Score: 8,
  662. },
  663. {
  664. Key: "value1",
  665. Score: 5,
  666. },
  667. }, pairs)
  668. _, err = NewRedis(client.Addr, "").ZrevrangebyscoreWithScoresAndLimit(
  669. "key", 5, 8, 1, 1)
  670. assert.NotNil(t, err)
  671. pairs, err = client.ZrevrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 1)
  672. assert.Nil(t, err)
  673. assert.EqualValues(t, []Pair{
  674. {
  675. Key: "value1",
  676. Score: 5,
  677. },
  678. }, pairs)
  679. pairs, err = client.ZrevrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 0)
  680. assert.Nil(t, err)
  681. assert.Equal(t, 0, len(pairs))
  682. })
  683. }
  684. func TestRedis_Pipelined(t *testing.T) {
  685. runOnRedis(t, func(client *Redis) {
  686. assert.NotNil(t, NewRedis(client.Addr, "").Pipelined(func(pipeliner Pipeliner) error {
  687. return nil
  688. }))
  689. err := client.Pipelined(
  690. func(pipe Pipeliner) error {
  691. pipe.Incr("pipelined_counter")
  692. pipe.Expire("pipelined_counter", time.Hour)
  693. pipe.ZAdd("zadd", Z{Score: 12, Member: "zadd"})
  694. return nil
  695. },
  696. )
  697. assert.Nil(t, err)
  698. _, err = NewRedis(client.Addr, "").Ttl("pipelined_counter")
  699. assert.NotNil(t, err)
  700. ttl, err := client.Ttl("pipelined_counter")
  701. assert.Nil(t, err)
  702. assert.Equal(t, 3600, ttl)
  703. value, err := client.Get("pipelined_counter")
  704. assert.Nil(t, err)
  705. assert.Equal(t, "1", value)
  706. score, err := client.Zscore("zadd", "zadd")
  707. assert.Nil(t, err)
  708. assert.Equal(t, int64(12), score)
  709. })
  710. }
  711. func TestRedisString(t *testing.T) {
  712. runOnRedis(t, func(client *Redis) {
  713. client.Ping()
  714. _, err := getRedis(NewRedis(client.Addr, ClusterType))
  715. assert.Nil(t, err)
  716. assert.Equal(t, client.Addr, client.String())
  717. assert.NotNil(t, NewRedis(client.Addr, "").Ping())
  718. })
  719. }
  720. func TestRedisScriptLoad(t *testing.T) {
  721. runOnRedis(t, func(client *Redis) {
  722. client.Ping()
  723. _, err := NewRedis(client.Addr, "").scriptLoad("foo")
  724. assert.NotNil(t, err)
  725. _, err = client.scriptLoad("foo")
  726. assert.NotNil(t, err)
  727. })
  728. }
  729. func TestRedisToPairs(t *testing.T) {
  730. pairs := toPairs([]red.Z{
  731. {
  732. Member: 1,
  733. Score: 1,
  734. },
  735. {
  736. Member: 2,
  737. Score: 2,
  738. },
  739. })
  740. assert.EqualValues(t, []Pair{
  741. {
  742. Key: "1",
  743. Score: 1,
  744. },
  745. {
  746. Key: "2",
  747. Score: 2,
  748. },
  749. }, pairs)
  750. }
  751. func TestRedisToStrings(t *testing.T) {
  752. vals := toStrings([]interface{}{1, 2})
  753. assert.EqualValues(t, []string{"1", "2"}, vals)
  754. }
  755. func TestRedisBlpop(t *testing.T) {
  756. runOnRedis(t, func(client *Redis) {
  757. client.Ping()
  758. var node mockedNode
  759. _, err := client.Blpop(nil, "foo")
  760. assert.NotNil(t, err)
  761. _, err = client.Blpop(node, "foo")
  762. assert.NotNil(t, err)
  763. })
  764. }
  765. func TestRedisBlpopEx(t *testing.T) {
  766. runOnRedis(t, func(client *Redis) {
  767. client.Ping()
  768. var node mockedNode
  769. _, _, err := client.BlpopEx(nil, "foo")
  770. assert.NotNil(t, err)
  771. _, _, err = client.BlpopEx(node, "foo")
  772. assert.NotNil(t, err)
  773. })
  774. }
  775. func runOnRedis(t *testing.T, fn func(client *Redis)) {
  776. s, err := miniredis.Run()
  777. assert.Nil(t, err)
  778. defer func() {
  779. client, err := clientManager.GetResource(s.Addr(), func() (io.Closer, error) {
  780. return nil, errors.New("should already exist")
  781. })
  782. if err != nil {
  783. t.Error(err)
  784. }
  785. if client != nil {
  786. client.Close()
  787. }
  788. }()
  789. fn(NewRedis(s.Addr(), NodeType))
  790. }
  791. type mockedNode struct {
  792. RedisNode
  793. }
  794. func (n mockedNode) BLPop(timeout time.Duration, keys ...string) *red.StringSliceCmd {
  795. return red.NewStringSliceCmd("foo", "bar")
  796. }