redis_test.go 47 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728
  1. package redis
  2. import (
  3. "context"
  4. "crypto/tls"
  5. "errors"
  6. "io"
  7. "strconv"
  8. "testing"
  9. "time"
  10. "github.com/alicebob/miniredis/v2"
  11. red "github.com/go-redis/redis/v8"
  12. "github.com/stretchr/testify/assert"
  13. "github.com/zeromicro/go-zero/core/logx"
  14. "github.com/zeromicro/go-zero/core/stringx"
  15. )
  16. func TestRedis_Decr(t *testing.T) {
  17. runOnRedis(t, func(client *Redis) {
  18. _, err := New(client.Addr, badType()).Decr("a")
  19. assert.NotNil(t, err)
  20. val, err := client.Decr("a")
  21. assert.Nil(t, err)
  22. assert.Equal(t, int64(-1), val)
  23. val, err = client.Decr("a")
  24. assert.Nil(t, err)
  25. assert.Equal(t, int64(-2), val)
  26. })
  27. }
  28. func TestRedis_DecrBy(t *testing.T) {
  29. runOnRedis(t, func(client *Redis) {
  30. _, err := New(client.Addr, badType()).Decrby("a", 2)
  31. assert.NotNil(t, err)
  32. val, err := client.Decrby("a", 2)
  33. assert.Nil(t, err)
  34. assert.Equal(t, int64(-2), val)
  35. val, err = client.Decrby("a", 3)
  36. assert.Nil(t, err)
  37. assert.Equal(t, int64(-5), val)
  38. })
  39. }
  40. func TestRedis_Exists(t *testing.T) {
  41. runOnRedis(t, func(client *Redis) {
  42. _, err := New(client.Addr, badType()).Exists("a")
  43. assert.NotNil(t, err)
  44. ok, err := client.Exists("a")
  45. assert.Nil(t, err)
  46. assert.False(t, ok)
  47. assert.Nil(t, client.Set("a", "b"))
  48. ok, err = client.Exists("a")
  49. assert.Nil(t, err)
  50. assert.True(t, ok)
  51. })
  52. }
  53. func TestRedisTLS_Exists(t *testing.T) {
  54. runOnRedisTLS(t, func(client *Redis) {
  55. _, err := New(client.Addr, badType()).Exists("a")
  56. assert.NotNil(t, err)
  57. ok, err := client.Exists("a")
  58. assert.NotNil(t, err)
  59. assert.False(t, ok)
  60. assert.NotNil(t, client.Set("a", "b"))
  61. ok, err = client.Exists("a")
  62. assert.NotNil(t, err)
  63. assert.False(t, ok)
  64. })
  65. }
  66. func TestRedis_Eval(t *testing.T) {
  67. runOnRedis(t, func(client *Redis) {
  68. _, err := New(client.Addr, badType()).Eval(`redis.call("EXISTS", KEYS[1])`, []string{"notexist"})
  69. assert.NotNil(t, err)
  70. _, err = client.Eval(`redis.call("EXISTS", KEYS[1])`, []string{"notexist"})
  71. assert.Equal(t, Nil, err)
  72. err = client.Set("key1", "value1")
  73. assert.Nil(t, err)
  74. _, err = client.Eval(`redis.call("EXISTS", KEYS[1])`, []string{"key1"})
  75. assert.Equal(t, Nil, err)
  76. val, err := client.Eval(`return redis.call("EXISTS", KEYS[1])`, []string{"key1"})
  77. assert.Nil(t, err)
  78. assert.Equal(t, int64(1), val)
  79. })
  80. }
  81. func TestRedis_GeoHash(t *testing.T) {
  82. runOnRedis(t, func(client *Redis) {
  83. _, err := client.GeoHash("parent", "child1", "child2")
  84. assert.Error(t, err)
  85. _, err = New(client.Addr, badType()).GeoHash("parent", "child1", "child2")
  86. assert.Error(t, err)
  87. })
  88. }
  89. func TestRedis_Hgetall(t *testing.T) {
  90. runOnRedis(t, func(client *Redis) {
  91. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  92. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  93. _, err := New(client.Addr, badType()).Hgetall("a")
  94. assert.NotNil(t, err)
  95. vals, err := client.Hgetall("a")
  96. assert.Nil(t, err)
  97. assert.EqualValues(t, map[string]string{
  98. "aa": "aaa",
  99. "bb": "bbb",
  100. }, vals)
  101. })
  102. }
  103. func TestRedis_Hvals(t *testing.T) {
  104. runOnRedis(t, func(client *Redis) {
  105. assert.NotNil(t, New(client.Addr, badType()).Hset("a", "aa", "aaa"))
  106. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  107. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  108. _, err := New(client.Addr, badType()).Hvals("a")
  109. assert.NotNil(t, err)
  110. vals, err := client.Hvals("a")
  111. assert.Nil(t, err)
  112. assert.ElementsMatch(t, []string{"aaa", "bbb"}, vals)
  113. })
  114. }
  115. func TestRedis_Hsetnx(t *testing.T) {
  116. runOnRedis(t, func(client *Redis) {
  117. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  118. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  119. _, err := New(client.Addr, badType()).Hsetnx("a", "bb", "ccc")
  120. assert.NotNil(t, err)
  121. ok, err := client.Hsetnx("a", "bb", "ccc")
  122. assert.Nil(t, err)
  123. assert.False(t, ok)
  124. ok, err = client.Hsetnx("a", "dd", "ddd")
  125. assert.Nil(t, err)
  126. assert.True(t, ok)
  127. vals, err := client.Hvals("a")
  128. assert.Nil(t, err)
  129. assert.ElementsMatch(t, []string{"aaa", "bbb", "ddd"}, vals)
  130. })
  131. }
  132. func TestRedis_HdelHlen(t *testing.T) {
  133. runOnRedis(t, func(client *Redis) {
  134. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  135. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  136. _, err := New(client.Addr, badType()).Hlen("a")
  137. assert.NotNil(t, err)
  138. num, err := client.Hlen("a")
  139. assert.Nil(t, err)
  140. assert.Equal(t, 2, num)
  141. val, err := client.Hdel("a", "aa")
  142. assert.Nil(t, err)
  143. assert.True(t, val)
  144. vals, err := client.Hvals("a")
  145. assert.Nil(t, err)
  146. assert.ElementsMatch(t, []string{"bbb"}, vals)
  147. })
  148. }
  149. func TestRedis_HIncrBy(t *testing.T) {
  150. runOnRedis(t, func(client *Redis) {
  151. _, err := New(client.Addr, badType()).Hincrby("key", "field", 2)
  152. assert.NotNil(t, err)
  153. val, err := client.Hincrby("key", "field", 2)
  154. assert.Nil(t, err)
  155. assert.Equal(t, 2, val)
  156. val, err = client.Hincrby("key", "field", 3)
  157. assert.Nil(t, err)
  158. assert.Equal(t, 5, val)
  159. })
  160. }
  161. func TestRedis_Hkeys(t *testing.T) {
  162. runOnRedis(t, func(client *Redis) {
  163. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  164. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  165. _, err := New(client.Addr, badType()).Hkeys("a")
  166. assert.NotNil(t, err)
  167. vals, err := client.Hkeys("a")
  168. assert.Nil(t, err)
  169. assert.ElementsMatch(t, []string{"aa", "bb"}, vals)
  170. })
  171. }
  172. func TestRedis_Hmget(t *testing.T) {
  173. runOnRedis(t, func(client *Redis) {
  174. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  175. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  176. _, err := New(client.Addr, badType()).Hmget("a", "aa", "bb")
  177. assert.NotNil(t, err)
  178. vals, err := client.Hmget("a", "aa", "bb")
  179. assert.Nil(t, err)
  180. assert.EqualValues(t, []string{"aaa", "bbb"}, vals)
  181. vals, err = client.Hmget("a", "aa", "no", "bb")
  182. assert.Nil(t, err)
  183. assert.EqualValues(t, []string{"aaa", "", "bbb"}, vals)
  184. })
  185. }
  186. func TestRedis_Hmset(t *testing.T) {
  187. runOnRedis(t, func(client *Redis) {
  188. assert.NotNil(t, New(client.Addr, badType()).Hmset("a", nil))
  189. assert.Nil(t, client.Hmset("a", map[string]string{
  190. "aa": "aaa",
  191. "bb": "bbb",
  192. }))
  193. vals, err := client.Hmget("a", "aa", "bb")
  194. assert.Nil(t, err)
  195. assert.EqualValues(t, []string{"aaa", "bbb"}, vals)
  196. })
  197. }
  198. func TestRedis_Hscan(t *testing.T) {
  199. t.Run("scan", func(t *testing.T) {
  200. runOnRedis(t, func(client *Redis) {
  201. key := "hash:test"
  202. fieldsAndValues := make(map[string]string)
  203. for i := 0; i < 1550; i++ {
  204. fieldsAndValues["filed_"+strconv.Itoa(i)] = stringx.Randn(i)
  205. }
  206. err := client.Hmset(key, fieldsAndValues)
  207. assert.Nil(t, err)
  208. var cursor uint64 = 0
  209. sum := 0
  210. for {
  211. _, _, err := New(client.Addr, badType()).Hscan(key, cursor, "*", 100)
  212. assert.NotNil(t, err)
  213. reMap, next, err := client.Hscan(key, cursor, "*", 100)
  214. assert.Nil(t, err)
  215. sum += len(reMap)
  216. if next == 0 {
  217. break
  218. }
  219. cursor = next
  220. }
  221. assert.Equal(t, sum, 3100)
  222. _, err = New(client.Addr, badType()).Del(key)
  223. assert.Error(t, err)
  224. _, err = client.Del(key)
  225. assert.NoError(t, err)
  226. })
  227. })
  228. t.Run("scan with error", func(t *testing.T) {
  229. runOnRedisWithError(t, func(client *Redis) {
  230. _, err := client.Del("hash:test")
  231. assert.Error(t, err)
  232. })
  233. })
  234. }
  235. func TestRedis_Incr(t *testing.T) {
  236. runOnRedis(t, func(client *Redis) {
  237. _, err := New(client.Addr, badType()).Incr("a")
  238. assert.NotNil(t, err)
  239. val, err := client.Incr("a")
  240. assert.Nil(t, err)
  241. assert.Equal(t, int64(1), val)
  242. val, err = client.Incr("a")
  243. assert.Nil(t, err)
  244. assert.Equal(t, int64(2), val)
  245. })
  246. }
  247. func TestRedis_IncrBy(t *testing.T) {
  248. runOnRedis(t, func(client *Redis) {
  249. _, err := New(client.Addr, badType()).Incrby("a", 2)
  250. assert.NotNil(t, err)
  251. val, err := client.Incrby("a", 2)
  252. assert.Nil(t, err)
  253. assert.Equal(t, int64(2), val)
  254. val, err = client.Incrby("a", 3)
  255. assert.Nil(t, err)
  256. assert.Equal(t, int64(5), val)
  257. })
  258. }
  259. func TestRedis_Keys(t *testing.T) {
  260. runOnRedis(t, func(client *Redis) {
  261. err := client.Set("key1", "value1")
  262. assert.Nil(t, err)
  263. err = client.Set("key2", "value2")
  264. assert.Nil(t, err)
  265. _, err = New(client.Addr, badType()).Keys("*")
  266. assert.NotNil(t, err)
  267. keys, err := client.Keys("*")
  268. assert.Nil(t, err)
  269. assert.ElementsMatch(t, []string{"key1", "key2"}, keys)
  270. })
  271. }
  272. func TestRedis_HyperLogLog(t *testing.T) {
  273. t.Run("hyperloglog", func(t *testing.T) {
  274. runOnRedis(t, func(client *Redis) {
  275. client.Ping()
  276. r := New(client.Addr)
  277. _, err := New(client.Addr, badType()).Pfadd("key1", "val1")
  278. assert.Error(t, err)
  279. ok, err := r.Pfadd("key1", "val1")
  280. assert.Nil(t, err)
  281. assert.True(t, ok)
  282. _, err = New(client.Addr, badType()).Pfcount("key1")
  283. assert.Error(t, err)
  284. val, err := r.Pfcount("key1")
  285. assert.Nil(t, err)
  286. assert.Equal(t, int64(1), val)
  287. ok, err = r.Pfadd("key2", "val2")
  288. assert.Nil(t, err)
  289. assert.True(t, ok)
  290. val, err = r.Pfcount("key2")
  291. assert.Nil(t, err)
  292. assert.Equal(t, int64(1), val)
  293. err = New(client.Addr, badType()).Pfmerge("key3", "key1", "key2")
  294. assert.Error(t, err)
  295. err = r.Pfmerge("key1", "key2")
  296. assert.Nil(t, err)
  297. val, err = r.Pfcount("key1")
  298. assert.Nil(t, err)
  299. assert.Equal(t, int64(2), val)
  300. })
  301. })
  302. t.Run("hyperloglog with error", func(t *testing.T) {
  303. runOnRedisWithError(t, func(client *Redis) {
  304. _, err := client.Pfadd("key1", "val1")
  305. assert.Error(t, err)
  306. })
  307. })
  308. }
  309. func TestRedis_List(t *testing.T) {
  310. t.Run("list", func(t *testing.T) {
  311. runOnRedis(t, func(client *Redis) {
  312. _, err := New(client.Addr, badType()).Lpush("key", "value1", "value2")
  313. assert.NotNil(t, err)
  314. val, err := client.Lpush("key", "value1", "value2")
  315. assert.Nil(t, err)
  316. assert.Equal(t, 2, val)
  317. _, err = New(client.Addr, badType()).Rpush("key", "value3", "value4")
  318. assert.NotNil(t, err)
  319. val, err = client.Rpush("key", "value3", "value4")
  320. assert.Nil(t, err)
  321. assert.Equal(t, 4, val)
  322. _, err = New(client.Addr, badType()).Llen("key")
  323. assert.NotNil(t, err)
  324. val, err = client.Llen("key")
  325. assert.Nil(t, err)
  326. assert.Equal(t, 4, val)
  327. _, err = New(client.Addr, badType()).Lindex("key", 1)
  328. assert.NotNil(t, err)
  329. value, err := client.Lindex("key", 0)
  330. assert.Nil(t, err)
  331. assert.Equal(t, "value2", value)
  332. vals, err := client.Lrange("key", 0, 10)
  333. assert.Nil(t, err)
  334. assert.EqualValues(t, []string{"value2", "value1", "value3", "value4"}, vals)
  335. _, err = New(client.Addr, badType()).Lpop("key")
  336. assert.NotNil(t, err)
  337. v, err := client.Lpop("key")
  338. assert.Nil(t, err)
  339. assert.Equal(t, "value2", v)
  340. val, err = client.Lpush("key", "value1", "value2")
  341. assert.Nil(t, err)
  342. assert.Equal(t, 5, val)
  343. _, err = New(client.Addr, badType()).Rpop("key")
  344. assert.NotNil(t, err)
  345. v, err = client.Rpop("key")
  346. assert.Nil(t, err)
  347. assert.Equal(t, "value4", v)
  348. val, err = client.Rpush("key", "value4", "value3", "value3")
  349. assert.Nil(t, err)
  350. assert.Equal(t, 7, val)
  351. _, err = New(client.Addr, badType()).Lrem("key", 2, "value1")
  352. assert.NotNil(t, err)
  353. n, err := client.Lrem("key", 2, "value1")
  354. assert.Nil(t, err)
  355. assert.Equal(t, 2, n)
  356. _, err = New(client.Addr, badType()).Lrange("key", 0, 10)
  357. assert.NotNil(t, err)
  358. vals, err = client.Lrange("key", 0, 10)
  359. assert.Nil(t, err)
  360. assert.EqualValues(t, []string{"value2", "value3", "value4", "value3", "value3"}, vals)
  361. n, err = client.Lrem("key", -2, "value3")
  362. assert.Nil(t, err)
  363. assert.Equal(t, 2, n)
  364. vals, err = client.Lrange("key", 0, 10)
  365. assert.Nil(t, err)
  366. assert.EqualValues(t, []string{"value2", "value3", "value4"}, vals)
  367. err = New(client.Addr, badType()).Ltrim("key", 0, 1)
  368. assert.Error(t, err)
  369. err = client.Ltrim("key", 0, 1)
  370. assert.Nil(t, err)
  371. vals, err = client.Lrange("key", 0, 10)
  372. assert.Nil(t, err)
  373. assert.EqualValues(t, []string{"value2", "value3"}, vals)
  374. })
  375. })
  376. t.Run("list error", func(t *testing.T) {
  377. runOnRedisWithError(t, func(client *Redis) {
  378. _, err := client.Llen("key")
  379. assert.Error(t, err)
  380. _, err = client.Lpush("key", "value1", "value2")
  381. assert.Error(t, err)
  382. _, err = client.Lrem("key", 2, "value1")
  383. assert.Error(t, err)
  384. _, err = client.Rpush("key", "value3", "value4")
  385. assert.Error(t, err)
  386. })
  387. })
  388. }
  389. func TestRedis_Mget(t *testing.T) {
  390. t.Run("mget", func(t *testing.T) {
  391. runOnRedis(t, func(client *Redis) {
  392. err := client.Set("key1", "value1")
  393. assert.Nil(t, err)
  394. err = client.Set("key2", "value2")
  395. assert.Nil(t, err)
  396. _, err = New(client.Addr, badType()).Mget("key1", "key0", "key2", "key3")
  397. assert.NotNil(t, err)
  398. vals, err := client.Mget("key1", "key0", "key2", "key3")
  399. assert.Nil(t, err)
  400. assert.EqualValues(t, []string{"value1", "", "value2", ""}, vals)
  401. })
  402. })
  403. t.Run("mget error", func(t *testing.T) {
  404. runOnRedisWithError(t, func(client *Redis) {
  405. _, err := client.Mget("key1", "key0")
  406. assert.Error(t, err)
  407. })
  408. })
  409. }
  410. func TestRedis_SetBit(t *testing.T) {
  411. t.Run("setbit", func(t *testing.T) {
  412. runOnRedis(t, func(client *Redis) {
  413. _, err := New(client.Addr, badType()).SetBit("key", 1, 1)
  414. assert.Error(t, err)
  415. val, err := client.SetBit("key", 1, 1)
  416. if assert.NoError(t, err) {
  417. assert.Equal(t, 0, val)
  418. }
  419. })
  420. })
  421. t.Run("setbit error", func(t *testing.T) {
  422. runOnRedisWithError(t, func(client *Redis) {
  423. _, err := client.SetBit("key", 1, 1)
  424. assert.Error(t, err)
  425. })
  426. })
  427. }
  428. func TestRedis_GetBit(t *testing.T) {
  429. t.Run("getbit", func(t *testing.T) {
  430. runOnRedis(t, func(client *Redis) {
  431. val, err := client.SetBit("key", 2, 1)
  432. assert.Nil(t, err)
  433. assert.Equal(t, 0, val)
  434. _, err = New(client.Addr, badType()).GetBit("key", 2)
  435. assert.NotNil(t, err)
  436. v, err := client.GetBit("key", 2)
  437. assert.Nil(t, err)
  438. assert.Equal(t, 1, v)
  439. })
  440. })
  441. t.Run("getbit error", func(t *testing.T) {
  442. runOnRedisWithError(t, func(client *Redis) {
  443. _, err := client.GetBit("key", 2)
  444. assert.Error(t, err)
  445. })
  446. })
  447. }
  448. func TestRedis_BitCount(t *testing.T) {
  449. runOnRedis(t, func(client *Redis) {
  450. for i := 0; i < 11; i++ {
  451. val, err := client.SetBit("key", int64(i), 1)
  452. assert.Nil(t, err)
  453. assert.Equal(t, 0, val)
  454. }
  455. _, err := New(client.Addr, badType()).BitCount("key", 0, -1)
  456. assert.NotNil(t, err)
  457. val, err := client.BitCount("key", 0, -1)
  458. assert.Nil(t, err)
  459. assert.Equal(t, int64(11), val)
  460. val, err = client.BitCount("key", 0, 0)
  461. assert.Nil(t, err)
  462. assert.Equal(t, int64(8), val)
  463. val, err = client.BitCount("key", 1, 1)
  464. assert.Nil(t, err)
  465. assert.Equal(t, int64(3), val)
  466. val, err = client.BitCount("key", 0, 1)
  467. assert.Nil(t, err)
  468. assert.Equal(t, int64(11), val)
  469. val, err = client.BitCount("key", 2, 2)
  470. assert.Nil(t, err)
  471. assert.Equal(t, int64(0), val)
  472. })
  473. }
  474. func TestRedis_BitOpAnd(t *testing.T) {
  475. runOnRedis(t, func(client *Redis) {
  476. err := client.Set("key1", "0")
  477. assert.Nil(t, err)
  478. err = client.Set("key2", "1")
  479. assert.Nil(t, err)
  480. _, err = New(client.Addr, badType()).BitOpAnd("destKey", "key1", "key2")
  481. assert.NotNil(t, err)
  482. val, err := client.BitOpAnd("destKey", "key1", "key2")
  483. assert.Nil(t, err)
  484. assert.Equal(t, int64(1), val)
  485. valStr, err := client.Get("destKey")
  486. assert.Nil(t, err)
  487. // destKey binary 110000 ascii 0
  488. assert.Equal(t, "0", valStr)
  489. })
  490. }
  491. func TestRedis_BitOpNot(t *testing.T) {
  492. runOnRedis(t, func(client *Redis) {
  493. err := client.Set("key1", "\u0000")
  494. assert.Nil(t, err)
  495. _, err = New(client.Addr, badType()).BitOpNot("destKey", "key1")
  496. assert.NotNil(t, err)
  497. val, err := client.BitOpNot("destKey", "key1")
  498. assert.Nil(t, err)
  499. assert.Equal(t, int64(1), val)
  500. valStr, err := client.Get("destKey")
  501. assert.Nil(t, err)
  502. assert.Equal(t, "\xff", valStr)
  503. })
  504. }
  505. func TestRedis_BitOpOr(t *testing.T) {
  506. runOnRedis(t, func(client *Redis) {
  507. err := client.Set("key1", "1")
  508. assert.Nil(t, err)
  509. err = client.Set("key2", "0")
  510. assert.Nil(t, err)
  511. _, err = New(client.Addr, badType()).BitOpOr("destKey", "key1", "key2")
  512. assert.NotNil(t, err)
  513. val, err := client.BitOpOr("destKey", "key1", "key2")
  514. assert.Nil(t, err)
  515. assert.Equal(t, int64(1), val)
  516. valStr, err := client.Get("destKey")
  517. assert.Nil(t, err)
  518. assert.Equal(t, "1", valStr)
  519. })
  520. }
  521. func TestRedis_BitOpXor(t *testing.T) {
  522. runOnRedis(t, func(client *Redis) {
  523. err := client.Set("key1", "\xff")
  524. assert.Nil(t, err)
  525. err = client.Set("key2", "\x0f")
  526. assert.Nil(t, err)
  527. _, err = New(client.Addr, badType()).BitOpXor("destKey", "key1", "key2")
  528. assert.NotNil(t, err)
  529. val, err := client.BitOpXor("destKey", "key1", "key2")
  530. assert.Nil(t, err)
  531. assert.Equal(t, int64(1), val)
  532. valStr, err := client.Get("destKey")
  533. assert.Nil(t, err)
  534. assert.Equal(t, "\xf0", valStr)
  535. })
  536. }
  537. func TestRedis_BitPos(t *testing.T) {
  538. runOnRedis(t, func(client *Redis) {
  539. // 11111111 11110000 00000000
  540. err := client.Set("key", "\xff\xf0\x00")
  541. assert.Nil(t, err)
  542. _, err = New(client.Addr, badType()).BitPos("key", 0, 0, -1)
  543. assert.NotNil(t, err)
  544. val, err := client.BitPos("key", 0, 0, 2)
  545. assert.Nil(t, err)
  546. assert.Equal(t, int64(12), val)
  547. val, err = client.BitPos("key", 1, 0, 2)
  548. assert.Nil(t, err)
  549. assert.Equal(t, int64(0), val)
  550. val, err = client.BitPos("key", 0, 1, 2)
  551. assert.Nil(t, err)
  552. assert.Equal(t, int64(12), val)
  553. val, err = client.BitPos("key", 1, 1, 2)
  554. assert.Nil(t, err)
  555. assert.Equal(t, int64(8), val)
  556. val, err = client.BitPos("key", 1, 2, 2)
  557. assert.Nil(t, err)
  558. assert.Equal(t, int64(-1), val)
  559. })
  560. }
  561. func TestRedis_Persist(t *testing.T) {
  562. runOnRedis(t, func(client *Redis) {
  563. _, err := New(client.Addr, badType()).Persist("key")
  564. assert.NotNil(t, err)
  565. ok, err := client.Persist("key")
  566. assert.Nil(t, err)
  567. assert.False(t, ok)
  568. err = client.Set("key", "value")
  569. assert.Nil(t, err)
  570. ok, err = client.Persist("key")
  571. assert.Nil(t, err)
  572. assert.False(t, ok)
  573. err = New(client.Addr, badType()).Expire("key", 5)
  574. assert.NotNil(t, err)
  575. err = client.Expire("key", 5)
  576. assert.Nil(t, err)
  577. ok, err = client.Persist("key")
  578. assert.Nil(t, err)
  579. assert.True(t, ok)
  580. err = New(client.Addr, badType()).Expireat("key", time.Now().Unix()+5)
  581. assert.NotNil(t, err)
  582. err = client.Expireat("key", time.Now().Unix()+5)
  583. assert.Nil(t, err)
  584. ok, err = client.Persist("key")
  585. assert.Nil(t, err)
  586. assert.True(t, ok)
  587. })
  588. }
  589. func TestRedis_Ping(t *testing.T) {
  590. runOnRedis(t, func(client *Redis) {
  591. ok := client.Ping()
  592. assert.True(t, ok)
  593. })
  594. }
  595. func TestRedis_Scan(t *testing.T) {
  596. runOnRedis(t, func(client *Redis) {
  597. err := client.Set("key1", "value1")
  598. assert.Nil(t, err)
  599. err = client.Set("key2", "value2")
  600. assert.Nil(t, err)
  601. _, _, err = New(client.Addr, badType()).Scan(0, "*", 100)
  602. assert.NotNil(t, err)
  603. keys, _, err := client.Scan(0, "*", 100)
  604. assert.Nil(t, err)
  605. assert.ElementsMatch(t, []string{"key1", "key2"}, keys)
  606. })
  607. }
  608. func TestRedis_Sscan(t *testing.T) {
  609. runOnRedis(t, func(client *Redis) {
  610. key := "list"
  611. var list []string
  612. for i := 0; i < 1550; i++ {
  613. list = append(list, stringx.Randn(i))
  614. }
  615. lens, err := client.Sadd(key, list)
  616. assert.Nil(t, err)
  617. assert.Equal(t, lens, 1550)
  618. var cursor uint64 = 0
  619. sum := 0
  620. for {
  621. _, _, err := New(client.Addr, badType()).Sscan(key, cursor, "", 100)
  622. assert.NotNil(t, err)
  623. keys, next, err := client.Sscan(key, cursor, "", 100)
  624. assert.Nil(t, err)
  625. sum += len(keys)
  626. if next == 0 {
  627. break
  628. }
  629. cursor = next
  630. }
  631. assert.Equal(t, sum, 1550)
  632. _, err = New(client.Addr, badType()).Del(key)
  633. assert.NotNil(t, err)
  634. _, err = client.Del(key)
  635. assert.Nil(t, err)
  636. })
  637. }
  638. func TestRedis_Set(t *testing.T) {
  639. t.Run("set", func(t *testing.T) {
  640. runOnRedis(t, func(client *Redis) {
  641. _, err := New(client.Addr, badType()).Sadd("key", 1, 2, 3, 4)
  642. assert.NotNil(t, err)
  643. num, err := client.Sadd("key", 1, 2, 3, 4)
  644. assert.Nil(t, err)
  645. assert.Equal(t, 4, num)
  646. _, err = New(client.Addr, badType()).Scard("key")
  647. assert.NotNil(t, err)
  648. val, err := client.Scard("key")
  649. assert.Nil(t, err)
  650. assert.Equal(t, int64(4), val)
  651. _, err = New(client.Addr, badType()).Sismember("key", 2)
  652. assert.NotNil(t, err)
  653. ok, err := client.Sismember("key", 2)
  654. assert.Nil(t, err)
  655. assert.True(t, ok)
  656. _, err = New(client.Addr, badType()).Srem("key", 3, 4)
  657. assert.NotNil(t, err)
  658. num, err = client.Srem("key", 3, 4)
  659. assert.Nil(t, err)
  660. assert.Equal(t, 2, num)
  661. _, err = New(client.Addr, badType()).Smembers("key")
  662. assert.NotNil(t, err)
  663. vals, err := client.Smembers("key")
  664. assert.Nil(t, err)
  665. assert.ElementsMatch(t, []string{"1", "2"}, vals)
  666. _, err = New(client.Addr, badType()).Srandmember("key", 1)
  667. assert.NotNil(t, err)
  668. members, err := client.Srandmember("key", 1)
  669. assert.Nil(t, err)
  670. assert.Len(t, members, 1)
  671. assert.Contains(t, []string{"1", "2"}, members[0])
  672. _, err = New(client.Addr, badType()).Spop("key")
  673. assert.NotNil(t, err)
  674. member, err := client.Spop("key")
  675. assert.Nil(t, err)
  676. assert.Contains(t, []string{"1", "2"}, member)
  677. _, err = New(client.Addr, badType()).Smembers("key")
  678. assert.NotNil(t, err)
  679. vals, err = client.Smembers("key")
  680. assert.Nil(t, err)
  681. assert.NotContains(t, vals, member)
  682. _, err = New(client.Addr, badType()).Sadd("key1", 1, 2, 3, 4)
  683. assert.NotNil(t, err)
  684. num, err = client.Sadd("key1", 1, 2, 3, 4)
  685. assert.Nil(t, err)
  686. assert.Equal(t, 4, num)
  687. num, err = client.Sadd("key2", 2, 3, 4, 5)
  688. assert.Nil(t, err)
  689. assert.Equal(t, 4, num)
  690. _, err = New(client.Addr, badType()).Sunion("key1", "key2")
  691. assert.NotNil(t, err)
  692. vals, err = client.Sunion("key1", "key2")
  693. assert.Nil(t, err)
  694. assert.ElementsMatch(t, []string{"1", "2", "3", "4", "5"}, vals)
  695. _, err = New(client.Addr, badType()).Sunionstore("key3", "key1", "key2")
  696. assert.NotNil(t, err)
  697. num, err = client.Sunionstore("key3", "key1", "key2")
  698. assert.Nil(t, err)
  699. assert.Equal(t, 5, num)
  700. _, err = New(client.Addr, badType()).Sdiff("key1", "key2")
  701. assert.NotNil(t, err)
  702. vals, err = client.Sdiff("key1", "key2")
  703. assert.Nil(t, err)
  704. assert.EqualValues(t, []string{"1"}, vals)
  705. _, err = New(client.Addr, badType()).Sdiffstore("key4", "key1", "key2")
  706. assert.NotNil(t, err)
  707. num, err = client.Sdiffstore("key4", "key1", "key2")
  708. assert.Nil(t, err)
  709. assert.Equal(t, 1, num)
  710. _, err = New(client.Addr, badType()).Sinter("key1", "key2")
  711. assert.NotNil(t, err)
  712. vals, err = client.Sinter("key1", "key2")
  713. assert.Nil(t, err)
  714. assert.ElementsMatch(t, []string{"2", "3", "4"}, vals)
  715. _, err = New(client.Addr, badType()).Sinterstore("key4", "key1", "key2")
  716. assert.NotNil(t, err)
  717. num, err = client.Sinterstore("key4", "key1", "key2")
  718. assert.Nil(t, err)
  719. assert.Equal(t, 3, num)
  720. })
  721. })
  722. t.Run("set with error", func(t *testing.T) {
  723. runOnRedisWithError(t, func(client *Redis) {
  724. _, err := client.Sadd("key", 1, 2, 3, 4)
  725. assert.Error(t, err)
  726. _, err = client.Srem("key", 3, 4)
  727. assert.Error(t, err)
  728. _, err = client.Sunionstore("key3", "key1", "key2")
  729. assert.Error(t, err)
  730. _, err = client.Sdiffstore("key4", "key1", "key2")
  731. assert.Error(t, err)
  732. _, err = client.Sinterstore("key4", "key1", "key2")
  733. assert.Error(t, err)
  734. })
  735. })
  736. }
  737. func TestRedis_GetSet(t *testing.T) {
  738. runOnRedis(t, func(client *Redis) {
  739. _, err := New(client.Addr, badType()).GetSet("hello", "world")
  740. assert.NotNil(t, err)
  741. val, err := client.GetSet("hello", "world")
  742. assert.Nil(t, err)
  743. assert.Equal(t, "", val)
  744. val, err = client.Get("hello")
  745. assert.Nil(t, err)
  746. assert.Equal(t, "world", val)
  747. val, err = client.GetSet("hello", "newworld")
  748. assert.Nil(t, err)
  749. assert.Equal(t, "world", val)
  750. val, err = client.Get("hello")
  751. assert.Nil(t, err)
  752. assert.Equal(t, "newworld", val)
  753. ret, err := client.Del("hello")
  754. assert.Nil(t, err)
  755. assert.Equal(t, 1, ret)
  756. })
  757. }
  758. func TestRedis_SetGetDel(t *testing.T) {
  759. runOnRedis(t, func(client *Redis) {
  760. err := New(client.Addr, badType()).Set("hello", "world")
  761. assert.NotNil(t, err)
  762. err = client.Set("hello", "world")
  763. assert.Nil(t, err)
  764. _, err = New(client.Addr, badType()).Get("hello")
  765. assert.NotNil(t, err)
  766. val, err := client.Get("hello")
  767. assert.Nil(t, err)
  768. assert.Equal(t, "world", val)
  769. ret, err := client.Del("hello")
  770. assert.Nil(t, err)
  771. assert.Equal(t, 1, ret)
  772. })
  773. }
  774. func TestRedis_SetExNx(t *testing.T) {
  775. runOnRedis(t, func(client *Redis) {
  776. err := New(client.Addr, badType()).Setex("hello", "world", 5)
  777. assert.NotNil(t, err)
  778. err = client.Setex("hello", "world", 5)
  779. assert.Nil(t, err)
  780. _, err = New(client.Addr, badType()).Setnx("hello", "newworld")
  781. assert.NotNil(t, err)
  782. ok, err := client.Setnx("hello", "newworld")
  783. assert.Nil(t, err)
  784. assert.False(t, ok)
  785. ok, err = client.Setnx("newhello", "newworld")
  786. assert.Nil(t, err)
  787. assert.True(t, ok)
  788. val, err := client.Get("hello")
  789. assert.Nil(t, err)
  790. assert.Equal(t, "world", val)
  791. val, err = client.Get("newhello")
  792. assert.Nil(t, err)
  793. assert.Equal(t, "newworld", val)
  794. ttl, err := client.Ttl("hello")
  795. assert.Nil(t, err)
  796. assert.True(t, ttl > 0)
  797. _, err = New(client.Addr, badType()).SetnxEx("newhello", "newworld", 5)
  798. assert.NotNil(t, err)
  799. ok, err = client.SetnxEx("newhello", "newworld", 5)
  800. assert.Nil(t, err)
  801. assert.False(t, ok)
  802. num, err := client.Del("newhello")
  803. assert.Nil(t, err)
  804. assert.Equal(t, 1, num)
  805. ok, err = client.SetnxEx("newhello", "newworld", 5)
  806. assert.Nil(t, err)
  807. assert.True(t, ok)
  808. val, err = client.Get("newhello")
  809. assert.Nil(t, err)
  810. assert.Equal(t, "newworld", val)
  811. })
  812. }
  813. func TestRedis_SetGetDelHashField(t *testing.T) {
  814. t.Run("hash", func(t *testing.T) {
  815. runOnRedis(t, func(client *Redis) {
  816. err := client.Hset("key", "field", "value")
  817. assert.Nil(t, err)
  818. _, err = New(client.Addr, badType()).Hget("key", "field")
  819. assert.NotNil(t, err)
  820. val, err := client.Hget("key", "field")
  821. assert.Nil(t, err)
  822. assert.Equal(t, "value", val)
  823. _, err = New(client.Addr, badType()).Hexists("key", "field")
  824. assert.NotNil(t, err)
  825. ok, err := client.Hexists("key", "field")
  826. assert.Nil(t, err)
  827. assert.True(t, ok)
  828. _, err = New(client.Addr, badType()).Hdel("key", "field")
  829. assert.NotNil(t, err)
  830. ret, err := client.Hdel("key", "field")
  831. assert.Nil(t, err)
  832. assert.True(t, ret)
  833. ok, err = client.Hexists("key", "field")
  834. assert.Nil(t, err)
  835. assert.False(t, ok)
  836. })
  837. })
  838. t.Run("hash error", func(t *testing.T) {
  839. runOnRedisWithError(t, func(client *Redis) {
  840. _, err := client.Hdel("key", "field")
  841. assert.Error(t, err)
  842. _, err = client.Hincrby("key", "field", 1)
  843. assert.Error(t, err)
  844. _, err = client.HincrbyFloat("key", "field", 1)
  845. assert.Error(t, err)
  846. _, err = client.Hlen("key")
  847. assert.Error(t, err)
  848. _, err = client.Hmget("key", "field")
  849. assert.Error(t, err)
  850. })
  851. })
  852. }
  853. func TestRedis_SortedSet(t *testing.T) {
  854. t.Run("sorted set", func(t *testing.T) {
  855. runOnRedis(t, func(client *Redis) {
  856. ok, err := client.ZaddFloat("key", 1, "value1")
  857. assert.Nil(t, err)
  858. assert.True(t, ok)
  859. ok, err = client.Zadd("key", 2, "value1")
  860. assert.Nil(t, err)
  861. assert.False(t, ok)
  862. val, err := client.Zscore("key", "value1")
  863. assert.Nil(t, err)
  864. assert.Equal(t, int64(2), val)
  865. _, err = New(client.Addr, badType()).Zincrby("key", 3, "value1")
  866. assert.NotNil(t, err)
  867. val, err = client.Zincrby("key", 3, "value1")
  868. assert.Nil(t, err)
  869. assert.Equal(t, int64(5), val)
  870. _, err = New(client.Addr, badType()).Zscore("key", "value1")
  871. assert.NotNil(t, err)
  872. val, err = client.Zscore("key", "value1")
  873. assert.Nil(t, err)
  874. assert.Equal(t, int64(5), val)
  875. _, err = New(client.Addr, badType()).Zadds("key")
  876. assert.NotNil(t, err)
  877. val, err = client.Zadds("key", Pair{
  878. Key: "value2",
  879. Score: 6,
  880. }, Pair{
  881. Key: "value3",
  882. Score: 7,
  883. })
  884. assert.Nil(t, err)
  885. assert.Equal(t, int64(2), val)
  886. _, err = New(client.Addr, badType()).ZRevRangeWithScores("key", 1, 3)
  887. assert.NotNil(t, err)
  888. _, err = client.ZRevRangeWithScores("key", 1, 3)
  889. assert.Nil(t, err)
  890. _, err = client.ZRevRangeWithScoresCtx(context.Background(), "key", 1, 3)
  891. assert.Nil(t, err)
  892. pairs, err := client.ZrevrangeWithScores("key", 1, 3)
  893. assert.Nil(t, err)
  894. assert.EqualValues(t, []Pair{
  895. {
  896. Key: "value2",
  897. Score: 6,
  898. },
  899. {
  900. Key: "value1",
  901. Score: 5,
  902. },
  903. }, pairs)
  904. rank, err := client.Zrank("key", "value2")
  905. assert.Nil(t, err)
  906. assert.Equal(t, int64(1), rank)
  907. rank, err = client.Zrevrank("key", "value1")
  908. assert.Nil(t, err)
  909. assert.Equal(t, int64(2), rank)
  910. _, err = New(client.Addr, badType()).Zrank("key", "value4")
  911. assert.NotNil(t, err)
  912. _, err = client.Zrank("key", "value4")
  913. assert.Equal(t, Nil, err)
  914. _, err = New(client.Addr, badType()).Zrem("key", "value2", "value3")
  915. assert.NotNil(t, err)
  916. num, err := client.Zrem("key", "value2", "value3")
  917. assert.Nil(t, err)
  918. assert.Equal(t, 2, num)
  919. ok, err = client.Zadd("key", 6, "value2")
  920. assert.Nil(t, err)
  921. assert.True(t, ok)
  922. ok, err = client.Zadd("key", 7, "value3")
  923. assert.Nil(t, err)
  924. assert.True(t, ok)
  925. ok, err = client.Zadd("key", 8, "value4")
  926. assert.Nil(t, err)
  927. assert.True(t, ok)
  928. _, err = New(client.Addr, badType()).Zremrangebyscore("key", 6, 7)
  929. assert.NotNil(t, err)
  930. num, err = client.Zremrangebyscore("key", 6, 7)
  931. assert.Nil(t, err)
  932. assert.Equal(t, 2, num)
  933. ok, err = client.Zadd("key", 6, "value2")
  934. assert.Nil(t, err)
  935. assert.True(t, ok)
  936. _, err = New(client.Addr, badType()).Zadd("key", 7, "value3")
  937. assert.NotNil(t, err)
  938. ok, err = client.Zadd("key", 7, "value3")
  939. assert.Nil(t, err)
  940. assert.True(t, ok)
  941. _, err = New(client.Addr, badType()).Zcount("key", 6, 7)
  942. assert.NotNil(t, err)
  943. num, err = client.Zcount("key", 6, 7)
  944. assert.Nil(t, err)
  945. assert.Equal(t, 2, num)
  946. _, err = New(client.Addr, badType()).Zremrangebyrank("key", 1, 2)
  947. assert.NotNil(t, err)
  948. num, err = client.Zremrangebyrank("key", 1, 2)
  949. assert.Nil(t, err)
  950. assert.Equal(t, 2, num)
  951. _, err = New(client.Addr, badType()).Zcard("key")
  952. assert.NotNil(t, err)
  953. card, err := client.Zcard("key")
  954. assert.Nil(t, err)
  955. assert.Equal(t, 2, card)
  956. _, err = New(client.Addr, badType()).Zrange("key", 0, -1)
  957. assert.NotNil(t, err)
  958. vals, err := client.Zrange("key", 0, -1)
  959. assert.Nil(t, err)
  960. assert.EqualValues(t, []string{"value1", "value4"}, vals)
  961. _, err = New(client.Addr, badType()).Zrevrange("key", 0, -1)
  962. assert.NotNil(t, err)
  963. vals, err = client.Zrevrange("key", 0, -1)
  964. assert.Nil(t, err)
  965. assert.EqualValues(t, []string{"value4", "value1"}, vals)
  966. _, err = New(client.Addr, badType()).ZrangeWithScores("key", 0, -1)
  967. assert.NotNil(t, err)
  968. pairs, err = client.ZrangeWithScores("key", 0, -1)
  969. assert.Nil(t, err)
  970. assert.EqualValues(t, []Pair{
  971. {
  972. Key: "value1",
  973. Score: 5,
  974. },
  975. {
  976. Key: "value4",
  977. Score: 8,
  978. },
  979. }, pairs)
  980. _, err = New(client.Addr, badType()).ZrangebyscoreWithScores("key", 5, 8)
  981. assert.NotNil(t, err)
  982. pairs, err = client.ZrangebyscoreWithScores("key", 5, 8)
  983. assert.Nil(t, err)
  984. assert.EqualValues(t, []Pair{
  985. {
  986. Key: "value1",
  987. Score: 5,
  988. },
  989. {
  990. Key: "value4",
  991. Score: 8,
  992. },
  993. }, pairs)
  994. _, err = New(client.Addr, badType()).ZrangebyscoreWithScoresAndLimit(
  995. "key", 5, 8, 1, 1)
  996. assert.NotNil(t, err)
  997. pairs, err = client.ZrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 1)
  998. assert.Nil(t, err)
  999. assert.EqualValues(t, []Pair{
  1000. {
  1001. Key: "value4",
  1002. Score: 8,
  1003. },
  1004. }, pairs)
  1005. pairs, err = client.ZrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 0)
  1006. assert.Nil(t, err)
  1007. assert.Equal(t, 0, len(pairs))
  1008. _, err = New(client.Addr, badType()).ZrevrangebyscoreWithScores("key", 5, 8)
  1009. assert.NotNil(t, err)
  1010. pairs, err = client.ZrevrangebyscoreWithScores("key", 5, 8)
  1011. assert.Nil(t, err)
  1012. assert.EqualValues(t, []Pair{
  1013. {
  1014. Key: "value4",
  1015. Score: 8,
  1016. },
  1017. {
  1018. Key: "value1",
  1019. Score: 5,
  1020. },
  1021. }, pairs)
  1022. _, err = New(client.Addr, badType()).ZrevrangebyscoreWithScoresAndLimit(
  1023. "key", 5, 8, 1, 1)
  1024. assert.NotNil(t, err)
  1025. pairs, err = client.ZrevrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 1)
  1026. assert.Nil(t, err)
  1027. assert.EqualValues(t, []Pair{
  1028. {
  1029. Key: "value1",
  1030. Score: 5,
  1031. },
  1032. }, pairs)
  1033. pairs, err = client.ZrevrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 0)
  1034. assert.Nil(t, err)
  1035. assert.Equal(t, 0, len(pairs))
  1036. _, err = New(client.Addr, badType()).Zrevrank("key", "value")
  1037. assert.NotNil(t, err)
  1038. _, _ = client.Zadd("second", 2, "aa")
  1039. _, _ = client.Zadd("third", 3, "bbb")
  1040. val, err = client.Zunionstore("union", &ZStore{
  1041. Keys: []string{"second", "third"},
  1042. Weights: []float64{1, 2},
  1043. Aggregate: "SUM",
  1044. })
  1045. assert.Nil(t, err)
  1046. assert.Equal(t, int64(2), val)
  1047. _, err = New(client.Addr, badType()).Zunionstore("union", &ZStore{})
  1048. assert.NotNil(t, err)
  1049. vals, err = client.Zrange("union", 0, 10000)
  1050. assert.Nil(t, err)
  1051. assert.EqualValues(t, []string{"aa", "bbb"}, vals)
  1052. ival, err := client.Zcard("union")
  1053. assert.Nil(t, err)
  1054. assert.Equal(t, 2, ival)
  1055. })
  1056. })
  1057. t.Run("sorted set error", func(t *testing.T) {
  1058. runOnRedisWithError(t, func(client *Redis) {
  1059. _, err := client.ZaddFloat("key", 1, "value")
  1060. assert.Error(t, err)
  1061. })
  1062. runOnRedisWithError(t, func(client *Redis) {
  1063. _, err := client.Zadds("key")
  1064. assert.Error(t, err)
  1065. })
  1066. runOnRedisWithError(t, func(client *Redis) {
  1067. _, err := client.Zcard("key")
  1068. assert.Error(t, err)
  1069. })
  1070. runOnRedisWithError(t, func(client *Redis) {
  1071. _, err := client.Zcount("key", 1, 2)
  1072. assert.Error(t, err)
  1073. })
  1074. runOnRedisWithError(t, func(client *Redis) {
  1075. _, err := client.Zincrby("key", 1, "value")
  1076. assert.Error(t, err)
  1077. })
  1078. runOnRedisWithError(t, func(client *Redis) {
  1079. _, err := client.Zscore("key", "value")
  1080. assert.Error(t, err)
  1081. })
  1082. runOnRedisWithError(t, func(client *Redis) {
  1083. _, err := client.Zrem("key", "value")
  1084. assert.Error(t, err)
  1085. })
  1086. runOnRedisWithError(t, func(client *Redis) {
  1087. _, err := client.Zremrangebyscore("key", 1, 2)
  1088. assert.Error(t, err)
  1089. })
  1090. runOnRedisWithError(t, func(client *Redis) {
  1091. _, err := client.Zremrangebyrank("key", 1, 2)
  1092. assert.Error(t, err)
  1093. })
  1094. runOnRedisWithError(t, func(client *Redis) {
  1095. _, err := client.ZrangeWithScores("key", 1, 2)
  1096. assert.Error(t, err)
  1097. })
  1098. runOnRedisWithError(t, func(client *Redis) {
  1099. _, err := client.ZrangeWithScoresByFloat("key", 1, 2)
  1100. assert.Error(t, err)
  1101. })
  1102. runOnRedisWithError(t, func(client *Redis) {
  1103. _, err := client.ZRevRangeWithScores("key", 1, 2)
  1104. assert.Error(t, err)
  1105. })
  1106. runOnRedisWithError(t, func(client *Redis) {
  1107. _, err := client.ZRevRangeWithScoresByFloat("key", 1, 2)
  1108. assert.Error(t, err)
  1109. })
  1110. runOnRedisWithError(t, func(client *Redis) {
  1111. _, err := client.ZrangebyscoreWithScores("key", 1, 2)
  1112. assert.Error(t, err)
  1113. })
  1114. runOnRedisWithError(t, func(client *Redis) {
  1115. _, err := client.ZrangebyscoreWithScoresByFloat("key", 1, 2)
  1116. assert.Error(t, err)
  1117. })
  1118. runOnRedisWithError(t, func(client *Redis) {
  1119. _, err := client.ZrangebyscoreWithScoresAndLimit("key", 1, 2, 1, 1)
  1120. assert.Error(t, err)
  1121. })
  1122. runOnRedisWithError(t, func(client *Redis) {
  1123. _, err := client.ZrangebyscoreWithScoresByFloatAndLimit("key", 1, 2, 1, 1)
  1124. assert.Error(t, err)
  1125. })
  1126. runOnRedisWithError(t, func(client *Redis) {
  1127. _, err := client.ZrevrangebyscoreWithScores("key", 1, 2)
  1128. assert.Error(t, err)
  1129. })
  1130. runOnRedisWithError(t, func(client *Redis) {
  1131. _, err := client.ZrevrangebyscoreWithScoresByFloat("key", 1, 2)
  1132. assert.Error(t, err)
  1133. })
  1134. runOnRedisWithError(t, func(client *Redis) {
  1135. _, err := client.ZrevrangebyscoreWithScoresAndLimit("key", 1, 2, 1, 1)
  1136. assert.Error(t, err)
  1137. })
  1138. runOnRedisWithError(t, func(client *Redis) {
  1139. _, err := client.ZrevrangebyscoreWithScoresByFloatAndLimit("key", 1, 2, 1, 1)
  1140. assert.Error(t, err)
  1141. })
  1142. })
  1143. }
  1144. func TestRedis_SortedSetByFloat64(t *testing.T) {
  1145. runOnRedis(t, func(client *Redis) {
  1146. ok, err := client.ZaddFloat("key", 10.345, "value1")
  1147. assert.Nil(t, err)
  1148. assert.True(t, ok)
  1149. val, err := client.ZscoreByFloat("key", "value1")
  1150. assert.Nil(t, err)
  1151. assert.Equal(t, 10.345, val)
  1152. _, err = New(client.Addr, badType()).ZscoreByFloat("key", "value1")
  1153. assert.Error(t, err)
  1154. _, _ = client.ZaddFloat("key", 10.346, "value2")
  1155. _, err = New(client.Addr, badType()).ZRevRangeWithScoresByFloat("key", 0, -1)
  1156. assert.NotNil(t, err)
  1157. _, err = client.ZRevRangeWithScoresByFloat("key", 0, -1)
  1158. assert.Nil(t, err)
  1159. _, err = client.ZRevRangeWithScoresByFloatCtx(context.Background(), "key", 0, -1)
  1160. assert.Nil(t, err)
  1161. pairs, err := client.ZrevrangeWithScoresByFloat("key", 0, -1)
  1162. assert.Nil(t, err)
  1163. assert.EqualValues(t, []FloatPair{
  1164. {
  1165. Key: "value2",
  1166. Score: 10.346,
  1167. },
  1168. {
  1169. Key: "value1",
  1170. Score: 10.345,
  1171. },
  1172. }, pairs)
  1173. _, err = New(client.Addr, badType()).ZrangeWithScoresByFloat("key", 0, -1)
  1174. assert.Error(t, err)
  1175. pairs, err = client.ZrangeWithScoresByFloat("key", 0, -1)
  1176. if assert.NoError(t, err) {
  1177. assert.EqualValues(t, []FloatPair{
  1178. {
  1179. Key: "value1",
  1180. Score: 10.345,
  1181. },
  1182. {
  1183. Key: "value2",
  1184. Score: 10.346,
  1185. },
  1186. }, pairs)
  1187. }
  1188. _, err = New(client.Addr, badType()).ZrangebyscoreWithScoresByFloat("key", 0, 20)
  1189. assert.NotNil(t, err)
  1190. pairs, err = client.ZrangebyscoreWithScoresByFloat("key", 0, 20)
  1191. assert.Nil(t, err)
  1192. assert.EqualValues(t, []FloatPair{
  1193. {
  1194. Key: "value1",
  1195. Score: 10.345,
  1196. },
  1197. {
  1198. Key: "value2",
  1199. Score: 10.346,
  1200. },
  1201. }, pairs)
  1202. _, err = client.ZrangebyscoreWithScoresByFloatAndLimit("key", 10.1, 12.2, 1, 0)
  1203. assert.NoError(t, err)
  1204. _, err = New(client.Addr, badType()).ZrangebyscoreWithScoresByFloatAndLimit(
  1205. "key", 10.1, 12.2, 1, 1)
  1206. assert.NotNil(t, err)
  1207. pairs, err = client.ZrangebyscoreWithScoresByFloatAndLimit("key", 10.1, 12.2, 1, 1)
  1208. assert.Nil(t, err)
  1209. assert.EqualValues(t, []FloatPair{
  1210. {
  1211. Key: "value2",
  1212. Score: 10.346,
  1213. },
  1214. }, pairs)
  1215. _, err = New(client.Addr, badType()).ZrevrangebyscoreWithScoresByFloat("key", 10, 12)
  1216. assert.NotNil(t, err)
  1217. pairs, err = client.ZrevrangebyscoreWithScoresByFloat("key", 10, 12)
  1218. assert.Nil(t, err)
  1219. assert.EqualValues(t, []FloatPair{
  1220. {
  1221. Key: "value2",
  1222. Score: 10.346,
  1223. },
  1224. {
  1225. Key: "value1",
  1226. Score: 10.345,
  1227. },
  1228. }, pairs)
  1229. _, err = client.ZrevrangebyscoreWithScoresByFloatAndLimit("key", 10, 12, 1, 0)
  1230. assert.NoError(t, err)
  1231. _, err = New(client.Addr, badType()).ZrevrangebyscoreWithScoresByFloatAndLimit(
  1232. "key", 10, 12, 1, 1)
  1233. assert.NotNil(t, err)
  1234. pairs, err = client.ZrevrangebyscoreWithScoresByFloatAndLimit("key", 10, 12, 1, 1)
  1235. assert.Nil(t, err)
  1236. assert.EqualValues(t, []FloatPair{
  1237. {
  1238. Key: "value1",
  1239. Score: 10.345,
  1240. },
  1241. }, pairs)
  1242. })
  1243. }
  1244. func TestRedis_IncrbyFloat(t *testing.T) {
  1245. runOnRedis(t, func(client *Redis) {
  1246. incrVal, err := client.IncrbyFloat("key", 0.002)
  1247. assert.Nil(t, err)
  1248. assert.Equal(t, 0.002, incrVal)
  1249. _, err = New(client.Addr, badType()).IncrbyFloat("key", -0.001)
  1250. assert.Error(t, err)
  1251. incrVal2, err := client.IncrbyFloat("key", -0.001)
  1252. assert.Nil(t, err)
  1253. assert.Equal(t, 0.001, incrVal2)
  1254. _, err = New(client.Addr, badType()).HincrbyFloat("hkey", "i", 0.002)
  1255. assert.Error(t, err)
  1256. hincrVal, err := client.HincrbyFloat("hkey", "i", 0.002)
  1257. assert.Nil(t, err)
  1258. assert.Equal(t, 0.002, hincrVal)
  1259. hincrVal2, err := client.HincrbyFloat("hkey", "i", -0.001)
  1260. assert.Nil(t, err)
  1261. assert.Equal(t, 0.001, hincrVal2)
  1262. })
  1263. }
  1264. func TestRedis_Pipelined(t *testing.T) {
  1265. runOnRedis(t, func(client *Redis) {
  1266. assert.NotNil(t, New(client.Addr, badType()).Pipelined(func(pipeliner Pipeliner) error {
  1267. return nil
  1268. }))
  1269. err := client.Pipelined(
  1270. func(pipe Pipeliner) error {
  1271. pipe.Incr(context.Background(), "pipelined_counter")
  1272. pipe.Expire(context.Background(), "pipelined_counter", time.Hour)
  1273. pipe.ZAdd(context.Background(), "zadd", &Z{Score: 12, Member: "zadd"})
  1274. return nil
  1275. },
  1276. )
  1277. assert.Nil(t, err)
  1278. _, err = New(client.Addr, badType()).Ttl("pipelined_counter")
  1279. assert.NotNil(t, err)
  1280. ttl, err := client.Ttl("pipelined_counter")
  1281. assert.Nil(t, err)
  1282. assert.Equal(t, 3600, ttl)
  1283. value, err := client.Get("pipelined_counter")
  1284. assert.Nil(t, err)
  1285. assert.Equal(t, "1", value)
  1286. score, err := client.Zscore("zadd", "zadd")
  1287. assert.Nil(t, err)
  1288. assert.Equal(t, int64(12), score)
  1289. })
  1290. }
  1291. func TestRedisString(t *testing.T) {
  1292. runOnRedis(t, func(client *Redis) {
  1293. client.Ping()
  1294. _, err := getRedis(New(client.Addr, Cluster()))
  1295. assert.Nil(t, err)
  1296. assert.Equal(t, client.Addr, client.String())
  1297. assert.NotNil(t, New(client.Addr, badType()).Ping())
  1298. })
  1299. }
  1300. func TestRedisScriptLoad(t *testing.T) {
  1301. runOnRedis(t, func(client *Redis) {
  1302. client.Ping()
  1303. _, err := New(client.Addr, badType()).ScriptLoad("foo")
  1304. assert.NotNil(t, err)
  1305. _, err = client.ScriptLoad("foo")
  1306. assert.NotNil(t, err)
  1307. })
  1308. }
  1309. func TestRedisEvalSha(t *testing.T) {
  1310. runOnRedis(t, func(client *Redis) {
  1311. client.Ping()
  1312. scriptHash, err := client.ScriptLoad(`return redis.call("EXISTS", KEYS[1])`)
  1313. assert.Nil(t, err)
  1314. _, err = New(client.Addr, badType()).EvalSha(scriptHash, []string{"key1"})
  1315. assert.Error(t, err)
  1316. result, err := client.EvalSha(scriptHash, []string{"key1"})
  1317. assert.Nil(t, err)
  1318. assert.Equal(t, int64(0), result)
  1319. })
  1320. }
  1321. func TestRedis_Ttl(t *testing.T) {
  1322. t.Run("TTL", func(t *testing.T) {
  1323. runOnRedis(t, func(client *Redis) {
  1324. if assert.NoError(t, client.Set("key", "value")) {
  1325. _, err := client.Ttl("key")
  1326. assert.NoError(t, err)
  1327. }
  1328. })
  1329. })
  1330. t.Run("TTL error", func(t *testing.T) {
  1331. runOnRedisWithError(t, func(client *Redis) {
  1332. _, err := client.Ttl("key")
  1333. assert.Error(t, err)
  1334. })
  1335. })
  1336. }
  1337. func TestRedisToPairs(t *testing.T) {
  1338. pairs := toPairs([]red.Z{
  1339. {
  1340. Member: 1,
  1341. Score: 1,
  1342. },
  1343. {
  1344. Member: 2,
  1345. Score: 2,
  1346. },
  1347. })
  1348. assert.EqualValues(t, []Pair{
  1349. {
  1350. Key: "1",
  1351. Score: 1,
  1352. },
  1353. {
  1354. Key: "2",
  1355. Score: 2,
  1356. },
  1357. }, pairs)
  1358. }
  1359. func TestRedisToFloatPairs(t *testing.T) {
  1360. pairs := toFloatPairs([]red.Z{
  1361. {
  1362. Member: 1,
  1363. Score: 1,
  1364. },
  1365. {
  1366. Member: 2,
  1367. Score: 2,
  1368. },
  1369. })
  1370. assert.EqualValues(t, []FloatPair{
  1371. {
  1372. Key: "1",
  1373. Score: 1,
  1374. },
  1375. {
  1376. Key: "2",
  1377. Score: 2,
  1378. },
  1379. }, pairs)
  1380. }
  1381. func TestRedis_Zscan(t *testing.T) {
  1382. runOnRedis(t, func(client *Redis) {
  1383. key := "key"
  1384. for i := 0; i < 1550; i++ {
  1385. ok, err := client.Zadd(key, int64(i), "value_"+strconv.Itoa(i))
  1386. assert.Nil(t, err)
  1387. assert.True(t, ok)
  1388. }
  1389. var cursor uint64 = 0
  1390. sum := 0
  1391. for {
  1392. _, _, err := New(client.Addr, badType()).Zscan(key, cursor, "value_*", 100)
  1393. assert.NotNil(t, err)
  1394. keys, next, err := client.Zscan(key, cursor, "value_*", 100)
  1395. assert.Nil(t, err)
  1396. sum += len(keys)
  1397. if next == 0 {
  1398. break
  1399. }
  1400. cursor = next
  1401. }
  1402. assert.Equal(t, sum, 3100)
  1403. _, err := New(client.Addr, badType()).Del(key)
  1404. assert.NotNil(t, err)
  1405. _, err = client.Del(key)
  1406. assert.Nil(t, err)
  1407. })
  1408. }
  1409. func TestRedisToStrings(t *testing.T) {
  1410. vals := toStrings([]interface{}{1, 2})
  1411. assert.EqualValues(t, []string{"1", "2"}, vals)
  1412. }
  1413. func TestRedisBlpop(t *testing.T) {
  1414. runOnRedis(t, func(client *Redis) {
  1415. client.Ping()
  1416. var node mockedNode
  1417. _, err := client.Blpop(nil, "foo")
  1418. assert.NotNil(t, err)
  1419. _, err = client.Blpop(node, "foo")
  1420. assert.NotNil(t, err)
  1421. })
  1422. }
  1423. func TestRedisBlpopEx(t *testing.T) {
  1424. t.Run("blpopex", func(t *testing.T) {
  1425. runOnRedis(t, func(client *Redis) {
  1426. client.Ping()
  1427. var node mockedNode
  1428. _, _, err := client.BlpopEx(nil, "foo")
  1429. assert.Error(t, err)
  1430. _, _, err = client.BlpopEx(node, "foo")
  1431. assert.Error(t, err)
  1432. })
  1433. })
  1434. }
  1435. func TestRedisBlpopWithTimeout(t *testing.T) {
  1436. runOnRedis(t, func(client *Redis) {
  1437. client.Ping()
  1438. var node mockedNode
  1439. _, err := client.BlpopWithTimeout(nil, 10*time.Second, "foo")
  1440. assert.NotNil(t, err)
  1441. _, err = client.BlpopWithTimeout(node, 10*time.Second, "foo")
  1442. assert.NotNil(t, err)
  1443. })
  1444. }
  1445. func TestRedisGeo(t *testing.T) {
  1446. t.Run("geo", func(t *testing.T) {
  1447. runOnRedis(t, func(client *Redis) {
  1448. client.Ping()
  1449. geoLocation := []*GeoLocation{{Longitude: 13.361389, Latitude: 38.115556, Name: "Palermo"},
  1450. {Longitude: 15.087269, Latitude: 37.502669, Name: "Catania"}}
  1451. v, err := client.GeoAdd("sicily", geoLocation...)
  1452. assert.Nil(t, err)
  1453. assert.Equal(t, int64(2), v)
  1454. _, err = New(client.Addr, badType()).GeoDist("sicily", "Palermo", "Catania", "m")
  1455. assert.Error(t, err)
  1456. v2, err := client.GeoDist("sicily", "Palermo", "Catania", "m")
  1457. assert.Nil(t, err)
  1458. assert.Equal(t, 166274, int(v2))
  1459. // GeoHash not support
  1460. _, err = New(client.Addr, badType()).GeoPos("sicily", "Palermo", "Catania")
  1461. assert.Error(t, err)
  1462. v3, err := client.GeoPos("sicily", "Palermo", "Catania")
  1463. assert.Nil(t, err)
  1464. assert.Equal(t, int64(v3[0].Longitude), int64(13))
  1465. assert.Equal(t, int64(v3[0].Latitude), int64(38))
  1466. assert.Equal(t, int64(v3[1].Longitude), int64(15))
  1467. assert.Equal(t, int64(v3[1].Latitude), int64(37))
  1468. _, err = New(client.Addr, badType()).GeoRadius("sicily", 15, 37,
  1469. &red.GeoRadiusQuery{WithDist: true, Unit: "km", Radius: 200})
  1470. assert.Error(t, err)
  1471. v4, err := client.GeoRadius("sicily", 15, 37, &red.GeoRadiusQuery{
  1472. WithDist: true,
  1473. Unit: "km", Radius: 200,
  1474. })
  1475. assert.Nil(t, err)
  1476. assert.Equal(t, int64(v4[0].Dist), int64(190))
  1477. assert.Equal(t, int64(v4[1].Dist), int64(56))
  1478. geoLocation2 := []*GeoLocation{{Longitude: 13.583333, Latitude: 37.316667, Name: "Agrigento"}}
  1479. _, err = New(client.Addr, badType()).GeoAdd("sicily", geoLocation2...)
  1480. assert.Error(t, err)
  1481. v5, err := client.GeoAdd("sicily", geoLocation2...)
  1482. assert.Nil(t, err)
  1483. assert.Equal(t, int64(1), v5)
  1484. _, err = New(client.Addr, badType()).GeoRadiusByMember("sicily", "Agrigento",
  1485. &red.GeoRadiusQuery{Unit: "km", Radius: 100})
  1486. assert.Error(t, err)
  1487. v6, err := client.GeoRadiusByMember("sicily", "Agrigento",
  1488. &red.GeoRadiusQuery{Unit: "km", Radius: 100})
  1489. assert.Nil(t, err)
  1490. assert.Equal(t, v6[0].Name, "Agrigento")
  1491. assert.Equal(t, v6[1].Name, "Palermo")
  1492. })
  1493. })
  1494. t.Run("geo error", func(t *testing.T) {
  1495. runOnRedisWithError(t, func(client *Redis) {
  1496. _, err := client.GeoAdd("sicily", &GeoLocation{
  1497. Longitude: 13.3,
  1498. Latitude: 38.1,
  1499. Name: "Palermo",
  1500. })
  1501. assert.Error(t, err)
  1502. _, err = client.GeoDist("sicily", "Palermo", "Catania", "m")
  1503. assert.Error(t, err)
  1504. _, err = client.GeoRadius("sicily", 15, 37, &red.GeoRadiusQuery{
  1505. WithDist: true,
  1506. })
  1507. assert.Error(t, err)
  1508. _, err = client.GeoRadiusByMember("sicily", "Agrigento", &red.GeoRadiusQuery{
  1509. Unit: "km",
  1510. })
  1511. assert.Error(t, err)
  1512. _, err = client.GeoPos("sicily", "Palermo", "Catania")
  1513. assert.Error(t, err)
  1514. })
  1515. })
  1516. }
  1517. func TestSetSlowThreshold(t *testing.T) {
  1518. assert.Equal(t, defaultSlowThreshold, slowThreshold.Load())
  1519. SetSlowThreshold(time.Second)
  1520. assert.Equal(t, time.Second, slowThreshold.Load())
  1521. }
  1522. func TestRedis_WithPass(t *testing.T) {
  1523. runOnRedis(t, func(client *Redis) {
  1524. err := New(client.Addr, WithPass("any")).Ping()
  1525. assert.NotNil(t, err)
  1526. })
  1527. }
  1528. func runOnRedis(t *testing.T, fn func(client *Redis)) {
  1529. logx.Disable()
  1530. s, err := miniredis.Run()
  1531. assert.Nil(t, err)
  1532. defer func() {
  1533. client, err := clientManager.GetResource(s.Addr(), func() (io.Closer, error) {
  1534. return nil, errors.New("should already exist")
  1535. })
  1536. if err != nil {
  1537. t.Error(err)
  1538. }
  1539. if client != nil {
  1540. _ = client.Close()
  1541. }
  1542. }()
  1543. fn(New(s.Addr()))
  1544. }
  1545. func runOnRedisWithError(t *testing.T, fn func(client *Redis)) {
  1546. logx.Disable()
  1547. s, err := miniredis.Run()
  1548. assert.NoError(t, err)
  1549. defer func() {
  1550. client, err := clientManager.GetResource(s.Addr(), func() (io.Closer, error) {
  1551. return nil, errors.New("should already exist")
  1552. })
  1553. if err != nil {
  1554. t.Error(err)
  1555. }
  1556. if client != nil {
  1557. _ = client.Close()
  1558. }
  1559. }()
  1560. s.SetError("mock error")
  1561. fn(New(s.Addr()))
  1562. }
  1563. func runOnRedisTLS(t *testing.T, fn func(client *Redis)) {
  1564. logx.Disable()
  1565. s, err := miniredis.RunTLS(&tls.Config{
  1566. Certificates: make([]tls.Certificate, 1),
  1567. InsecureSkipVerify: true,
  1568. })
  1569. assert.Nil(t, err)
  1570. defer func() {
  1571. client, err := clientManager.GetResource(s.Addr(), func() (io.Closer, error) {
  1572. return nil, errors.New("should already exist")
  1573. })
  1574. if err != nil {
  1575. t.Error(err)
  1576. }
  1577. if client != nil {
  1578. _ = client.Close()
  1579. }
  1580. }()
  1581. fn(New(s.Addr(), WithTLS()))
  1582. }
  1583. func badType() Option {
  1584. return func(r *Redis) {
  1585. r.Type = "bad"
  1586. }
  1587. }
  1588. type mockedNode struct {
  1589. RedisNode
  1590. }
  1591. func (n mockedNode) BLPop(_ context.Context, _ time.Duration, _ ...string) *red.StringSliceCmd {
  1592. return red.NewStringSliceCmd(context.Background(), "foo", "bar")
  1593. }