redis_test.go 24 KB

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