redis_test.go 38 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365
  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.NotNil(t, err)
  85. })
  86. }
  87. func TestRedis_Hgetall(t *testing.T) {
  88. runOnRedis(t, func(client *Redis) {
  89. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  90. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  91. _, err := New(client.Addr, badType()).Hgetall("a")
  92. assert.NotNil(t, err)
  93. vals, err := client.Hgetall("a")
  94. assert.Nil(t, err)
  95. assert.EqualValues(t, map[string]string{
  96. "aa": "aaa",
  97. "bb": "bbb",
  98. }, vals)
  99. })
  100. }
  101. func TestRedis_Hvals(t *testing.T) {
  102. runOnRedis(t, func(client *Redis) {
  103. assert.NotNil(t, New(client.Addr, badType()).Hset("a", "aa", "aaa"))
  104. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  105. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  106. _, err := New(client.Addr, badType()).Hvals("a")
  107. assert.NotNil(t, err)
  108. vals, err := client.Hvals("a")
  109. assert.Nil(t, err)
  110. assert.ElementsMatch(t, []string{"aaa", "bbb"}, vals)
  111. })
  112. }
  113. func TestRedis_Hsetnx(t *testing.T) {
  114. runOnRedis(t, func(client *Redis) {
  115. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  116. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  117. _, err := New(client.Addr, badType()).Hsetnx("a", "bb", "ccc")
  118. assert.NotNil(t, err)
  119. ok, err := client.Hsetnx("a", "bb", "ccc")
  120. assert.Nil(t, err)
  121. assert.False(t, ok)
  122. ok, err = client.Hsetnx("a", "dd", "ddd")
  123. assert.Nil(t, err)
  124. assert.True(t, ok)
  125. vals, err := client.Hvals("a")
  126. assert.Nil(t, err)
  127. assert.ElementsMatch(t, []string{"aaa", "bbb", "ddd"}, vals)
  128. })
  129. }
  130. func TestRedis_HdelHlen(t *testing.T) {
  131. runOnRedis(t, func(client *Redis) {
  132. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  133. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  134. _, err := New(client.Addr, badType()).Hlen("a")
  135. assert.NotNil(t, err)
  136. num, err := client.Hlen("a")
  137. assert.Nil(t, err)
  138. assert.Equal(t, 2, num)
  139. val, err := client.Hdel("a", "aa")
  140. assert.Nil(t, err)
  141. assert.True(t, val)
  142. vals, err := client.Hvals("a")
  143. assert.Nil(t, err)
  144. assert.ElementsMatch(t, []string{"bbb"}, vals)
  145. })
  146. }
  147. func TestRedis_HIncrBy(t *testing.T) {
  148. runOnRedis(t, func(client *Redis) {
  149. _, err := New(client.Addr, badType()).Hincrby("key", "field", 2)
  150. assert.NotNil(t, err)
  151. val, err := client.Hincrby("key", "field", 2)
  152. assert.Nil(t, err)
  153. assert.Equal(t, 2, val)
  154. val, err = client.Hincrby("key", "field", 3)
  155. assert.Nil(t, err)
  156. assert.Equal(t, 5, val)
  157. })
  158. }
  159. func TestRedis_Hkeys(t *testing.T) {
  160. runOnRedis(t, func(client *Redis) {
  161. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  162. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  163. _, err := New(client.Addr, badType()).Hkeys("a")
  164. assert.NotNil(t, err)
  165. vals, err := client.Hkeys("a")
  166. assert.Nil(t, err)
  167. assert.ElementsMatch(t, []string{"aa", "bb"}, vals)
  168. })
  169. }
  170. func TestRedis_Hmget(t *testing.T) {
  171. runOnRedis(t, func(client *Redis) {
  172. assert.Nil(t, client.Hset("a", "aa", "aaa"))
  173. assert.Nil(t, client.Hset("a", "bb", "bbb"))
  174. _, err := New(client.Addr, badType()).Hmget("a", "aa", "bb")
  175. assert.NotNil(t, err)
  176. vals, err := client.Hmget("a", "aa", "bb")
  177. assert.Nil(t, err)
  178. assert.EqualValues(t, []string{"aaa", "bbb"}, vals)
  179. vals, err = client.Hmget("a", "aa", "no", "bb")
  180. assert.Nil(t, err)
  181. assert.EqualValues(t, []string{"aaa", "", "bbb"}, vals)
  182. })
  183. }
  184. func TestRedis_Hmset(t *testing.T) {
  185. runOnRedis(t, func(client *Redis) {
  186. assert.NotNil(t, New(client.Addr, badType()).Hmset("a", nil))
  187. assert.Nil(t, client.Hmset("a", map[string]string{
  188. "aa": "aaa",
  189. "bb": "bbb",
  190. }))
  191. vals, err := client.Hmget("a", "aa", "bb")
  192. assert.Nil(t, err)
  193. assert.EqualValues(t, []string{"aaa", "bbb"}, vals)
  194. })
  195. }
  196. func TestRedis_Hscan(t *testing.T) {
  197. runOnRedis(t, func(client *Redis) {
  198. key := "hash:test"
  199. fieldsAndValues := make(map[string]string)
  200. for i := 0; i < 1550; i++ {
  201. fieldsAndValues["filed_"+strconv.Itoa(i)] = stringx.Randn(i)
  202. }
  203. err := client.Hmset(key, fieldsAndValues)
  204. assert.Nil(t, err)
  205. var cursor uint64 = 0
  206. sum := 0
  207. for {
  208. _, _, err := New(client.Addr, badType()).Hscan(key, cursor, "*", 100)
  209. assert.NotNil(t, err)
  210. reMap, next, err := client.Hscan(key, cursor, "*", 100)
  211. assert.Nil(t, err)
  212. sum += len(reMap)
  213. if next == 0 {
  214. break
  215. }
  216. cursor = next
  217. }
  218. assert.Equal(t, sum, 3100)
  219. _, err = New(client.Addr, badType()).Del(key)
  220. assert.NotNil(t, err)
  221. _, err = client.Del(key)
  222. assert.Nil(t, err)
  223. })
  224. }
  225. func TestRedis_Incr(t *testing.T) {
  226. runOnRedis(t, func(client *Redis) {
  227. _, err := New(client.Addr, badType()).Incr("a")
  228. assert.NotNil(t, err)
  229. val, err := client.Incr("a")
  230. assert.Nil(t, err)
  231. assert.Equal(t, int64(1), val)
  232. val, err = client.Incr("a")
  233. assert.Nil(t, err)
  234. assert.Equal(t, int64(2), val)
  235. })
  236. }
  237. func TestRedis_IncrBy(t *testing.T) {
  238. runOnRedis(t, func(client *Redis) {
  239. _, err := New(client.Addr, badType()).Incrby("a", 2)
  240. assert.NotNil(t, err)
  241. val, err := client.Incrby("a", 2)
  242. assert.Nil(t, err)
  243. assert.Equal(t, int64(2), val)
  244. val, err = client.Incrby("a", 3)
  245. assert.Nil(t, err)
  246. assert.Equal(t, int64(5), val)
  247. })
  248. }
  249. func TestRedis_Keys(t *testing.T) {
  250. runOnRedis(t, func(client *Redis) {
  251. err := client.Set("key1", "value1")
  252. assert.Nil(t, err)
  253. err = client.Set("key2", "value2")
  254. assert.Nil(t, err)
  255. _, err = New(client.Addr, badType()).Keys("*")
  256. assert.NotNil(t, err)
  257. keys, err := client.Keys("*")
  258. assert.Nil(t, err)
  259. assert.ElementsMatch(t, []string{"key1", "key2"}, keys)
  260. })
  261. }
  262. func TestRedis_HyperLogLog(t *testing.T) {
  263. runOnRedis(t, func(client *Redis) {
  264. client.Ping()
  265. r := New(client.Addr)
  266. ok, err := r.Pfadd("key1", "val1")
  267. assert.Nil(t, err)
  268. assert.True(t, ok)
  269. val, err := r.Pfcount("key1")
  270. assert.Nil(t, err)
  271. assert.Equal(t, int64(1), val)
  272. ok, err = r.Pfadd("key2", "val2")
  273. assert.Nil(t, err)
  274. assert.True(t, ok)
  275. val, err = r.Pfcount("key2")
  276. assert.Nil(t, err)
  277. assert.Equal(t, int64(1), val)
  278. err = r.Pfmerge("key1", "key2")
  279. assert.Nil(t, err)
  280. val, err = r.Pfcount("key1")
  281. assert.Nil(t, err)
  282. assert.Equal(t, int64(2), val)
  283. })
  284. }
  285. func TestRedis_List(t *testing.T) {
  286. runOnRedis(t, func(client *Redis) {
  287. _, err := New(client.Addr, badType()).Lpush("key", "value1", "value2")
  288. assert.NotNil(t, err)
  289. val, err := client.Lpush("key", "value1", "value2")
  290. assert.Nil(t, err)
  291. assert.Equal(t, 2, val)
  292. _, err = New(client.Addr, badType()).Rpush("key", "value3", "value4")
  293. assert.NotNil(t, err)
  294. val, err = client.Rpush("key", "value3", "value4")
  295. assert.Nil(t, err)
  296. assert.Equal(t, 4, val)
  297. _, err = New(client.Addr, badType()).Llen("key")
  298. assert.NotNil(t, err)
  299. val, err = client.Llen("key")
  300. assert.Nil(t, err)
  301. assert.Equal(t, 4, val)
  302. _, err = New(client.Addr, badType()).Lindex("key", 1)
  303. assert.NotNil(t, err)
  304. value, err := client.Lindex("key", 0)
  305. assert.Nil(t, err)
  306. assert.Equal(t, "value2", value)
  307. vals, err := client.Lrange("key", 0, 10)
  308. assert.Nil(t, err)
  309. assert.EqualValues(t, []string{"value2", "value1", "value3", "value4"}, vals)
  310. _, err = New(client.Addr, badType()).Lpop("key")
  311. assert.NotNil(t, err)
  312. v, err := client.Lpop("key")
  313. assert.Nil(t, err)
  314. assert.Equal(t, "value2", v)
  315. val, err = client.Lpush("key", "value1", "value2")
  316. assert.Nil(t, err)
  317. assert.Equal(t, 5, val)
  318. _, err = New(client.Addr, badType()).Rpop("key")
  319. assert.NotNil(t, err)
  320. v, err = client.Rpop("key")
  321. assert.Nil(t, err)
  322. assert.Equal(t, "value4", v)
  323. val, err = client.Rpush("key", "value4", "value3", "value3")
  324. assert.Nil(t, err)
  325. assert.Equal(t, 7, val)
  326. _, err = New(client.Addr, badType()).Lrem("key", 2, "value1")
  327. assert.NotNil(t, err)
  328. n, err := client.Lrem("key", 2, "value1")
  329. assert.Nil(t, err)
  330. assert.Equal(t, 2, n)
  331. _, err = New(client.Addr, badType()).Lrange("key", 0, 10)
  332. assert.NotNil(t, err)
  333. vals, err = client.Lrange("key", 0, 10)
  334. assert.Nil(t, err)
  335. assert.EqualValues(t, []string{"value2", "value3", "value4", "value3", "value3"}, vals)
  336. n, err = client.Lrem("key", -2, "value3")
  337. assert.Nil(t, err)
  338. assert.Equal(t, 2, n)
  339. vals, err = client.Lrange("key", 0, 10)
  340. assert.Nil(t, err)
  341. assert.EqualValues(t, []string{"value2", "value3", "value4"}, vals)
  342. err = client.Ltrim("key", 0, 1)
  343. assert.Nil(t, err)
  344. vals, err = client.Lrange("key", 0, 10)
  345. assert.Nil(t, err)
  346. assert.EqualValues(t, []string{"value2", "value3"}, vals)
  347. })
  348. }
  349. func TestRedis_Mget(t *testing.T) {
  350. runOnRedis(t, func(client *Redis) {
  351. err := client.Set("key1", "value1")
  352. assert.Nil(t, err)
  353. err = client.Set("key2", "value2")
  354. assert.Nil(t, err)
  355. _, err = New(client.Addr, badType()).Mget("key1", "key0", "key2", "key3")
  356. assert.NotNil(t, err)
  357. vals, err := client.Mget("key1", "key0", "key2", "key3")
  358. assert.Nil(t, err)
  359. assert.EqualValues(t, []string{"value1", "", "value2", ""}, vals)
  360. })
  361. }
  362. func TestRedis_SetBit(t *testing.T) {
  363. runOnRedis(t, func(client *Redis) {
  364. _, err := New(client.Addr, badType()).SetBit("key", 1, 1)
  365. assert.NotNil(t, err)
  366. val, err := client.SetBit("key", 1, 1)
  367. assert.Nil(t, err)
  368. assert.Equal(t, 0, val)
  369. })
  370. }
  371. func TestRedis_GetBit(t *testing.T) {
  372. runOnRedis(t, func(client *Redis) {
  373. val, err := client.SetBit("key", 2, 1)
  374. assert.Nil(t, err)
  375. assert.Equal(t, 0, val)
  376. _, err = New(client.Addr, badType()).GetBit("key", 2)
  377. assert.NotNil(t, err)
  378. v, err := client.GetBit("key", 2)
  379. assert.Nil(t, err)
  380. assert.Equal(t, 1, v)
  381. })
  382. }
  383. func TestRedis_BitCount(t *testing.T) {
  384. runOnRedis(t, func(client *Redis) {
  385. for i := 0; i < 11; i++ {
  386. val, err := client.SetBit("key", int64(i), 1)
  387. assert.Nil(t, err)
  388. assert.Equal(t, 0, val)
  389. }
  390. _, err := New(client.Addr, badType()).BitCount("key", 0, -1)
  391. assert.NotNil(t, err)
  392. val, err := client.BitCount("key", 0, -1)
  393. assert.Nil(t, err)
  394. assert.Equal(t, int64(11), val)
  395. val, err = client.BitCount("key", 0, 0)
  396. assert.Nil(t, err)
  397. assert.Equal(t, int64(8), val)
  398. val, err = client.BitCount("key", 1, 1)
  399. assert.Nil(t, err)
  400. assert.Equal(t, int64(3), val)
  401. val, err = client.BitCount("key", 0, 1)
  402. assert.Nil(t, err)
  403. assert.Equal(t, int64(11), val)
  404. val, err = client.BitCount("key", 2, 2)
  405. assert.Nil(t, err)
  406. assert.Equal(t, int64(0), val)
  407. })
  408. }
  409. func TestRedis_BitOpAnd(t *testing.T) {
  410. runOnRedis(t, func(client *Redis) {
  411. err := client.Set("key1", "0")
  412. assert.Nil(t, err)
  413. err = client.Set("key2", "1")
  414. assert.Nil(t, err)
  415. _, err = New(client.Addr, badType()).BitOpAnd("destKey", "key1", "key2")
  416. assert.NotNil(t, err)
  417. val, err := client.BitOpAnd("destKey", "key1", "key2")
  418. assert.Nil(t, err)
  419. assert.Equal(t, int64(1), val)
  420. valStr, err := client.Get("destKey")
  421. assert.Nil(t, err)
  422. // destKey binary 110000 ascii 0
  423. assert.Equal(t, "0", valStr)
  424. })
  425. }
  426. func TestRedis_BitOpNot(t *testing.T) {
  427. runOnRedis(t, func(client *Redis) {
  428. err := client.Set("key1", "\u0000")
  429. assert.Nil(t, err)
  430. _, err = New(client.Addr, badType()).BitOpNot("destKey", "key1")
  431. assert.NotNil(t, err)
  432. val, err := client.BitOpNot("destKey", "key1")
  433. assert.Nil(t, err)
  434. assert.Equal(t, int64(1), val)
  435. valStr, err := client.Get("destKey")
  436. assert.Nil(t, err)
  437. assert.Equal(t, "\xff", valStr)
  438. })
  439. }
  440. func TestRedis_BitOpOr(t *testing.T) {
  441. runOnRedis(t, func(client *Redis) {
  442. err := client.Set("key1", "1")
  443. assert.Nil(t, err)
  444. err = client.Set("key2", "0")
  445. assert.Nil(t, err)
  446. _, err = New(client.Addr, badType()).BitOpOr("destKey", "key1", "key2")
  447. assert.NotNil(t, err)
  448. val, err := client.BitOpOr("destKey", "key1", "key2")
  449. assert.Nil(t, err)
  450. assert.Equal(t, int64(1), val)
  451. valStr, err := client.Get("destKey")
  452. assert.Nil(t, err)
  453. assert.Equal(t, "1", valStr)
  454. })
  455. }
  456. func TestRedis_BitOpXor(t *testing.T) {
  457. runOnRedis(t, func(client *Redis) {
  458. err := client.Set("key1", "\xff")
  459. assert.Nil(t, err)
  460. err = client.Set("key2", "\x0f")
  461. assert.Nil(t, err)
  462. _, err = New(client.Addr, badType()).BitOpXor("destKey", "key1", "key2")
  463. assert.NotNil(t, err)
  464. val, err := client.BitOpXor("destKey", "key1", "key2")
  465. assert.Nil(t, err)
  466. assert.Equal(t, int64(1), val)
  467. valStr, err := client.Get("destKey")
  468. assert.Nil(t, err)
  469. assert.Equal(t, "\xf0", valStr)
  470. })
  471. }
  472. func TestRedis_BitPos(t *testing.T) {
  473. runOnRedis(t, func(client *Redis) {
  474. // 11111111 11110000 00000000
  475. err := client.Set("key", "\xff\xf0\x00")
  476. assert.Nil(t, err)
  477. _, err = New(client.Addr, badType()).BitPos("key", 0, 0, -1)
  478. assert.NotNil(t, err)
  479. val, err := client.BitPos("key", 0, 0, 2)
  480. assert.Nil(t, err)
  481. assert.Equal(t, int64(12), val)
  482. val, err = client.BitPos("key", 1, 0, 2)
  483. assert.Nil(t, err)
  484. assert.Equal(t, int64(0), val)
  485. val, err = client.BitPos("key", 0, 1, 2)
  486. assert.Nil(t, err)
  487. assert.Equal(t, int64(12), val)
  488. val, err = client.BitPos("key", 1, 1, 2)
  489. assert.Nil(t, err)
  490. assert.Equal(t, int64(8), val)
  491. val, err = client.BitPos("key", 1, 2, 2)
  492. assert.Nil(t, err)
  493. assert.Equal(t, int64(-1), val)
  494. })
  495. }
  496. func TestRedis_Persist(t *testing.T) {
  497. runOnRedis(t, func(client *Redis) {
  498. _, err := New(client.Addr, badType()).Persist("key")
  499. assert.NotNil(t, err)
  500. ok, err := client.Persist("key")
  501. assert.Nil(t, err)
  502. assert.False(t, ok)
  503. err = client.Set("key", "value")
  504. assert.Nil(t, err)
  505. ok, err = client.Persist("key")
  506. assert.Nil(t, err)
  507. assert.False(t, ok)
  508. err = New(client.Addr, badType()).Expire("key", 5)
  509. assert.NotNil(t, err)
  510. err = client.Expire("key", 5)
  511. assert.Nil(t, err)
  512. ok, err = client.Persist("key")
  513. assert.Nil(t, err)
  514. assert.True(t, ok)
  515. err = New(client.Addr, badType()).Expireat("key", time.Now().Unix()+5)
  516. assert.NotNil(t, err)
  517. err = client.Expireat("key", time.Now().Unix()+5)
  518. assert.Nil(t, err)
  519. ok, err = client.Persist("key")
  520. assert.Nil(t, err)
  521. assert.True(t, ok)
  522. })
  523. }
  524. func TestRedis_Ping(t *testing.T) {
  525. runOnRedis(t, func(client *Redis) {
  526. ok := client.Ping()
  527. assert.True(t, ok)
  528. })
  529. }
  530. func TestRedis_Scan(t *testing.T) {
  531. runOnRedis(t, func(client *Redis) {
  532. err := client.Set("key1", "value1")
  533. assert.Nil(t, err)
  534. err = client.Set("key2", "value2")
  535. assert.Nil(t, err)
  536. _, _, err = New(client.Addr, badType()).Scan(0, "*", 100)
  537. assert.NotNil(t, err)
  538. keys, _, err := client.Scan(0, "*", 100)
  539. assert.Nil(t, err)
  540. assert.ElementsMatch(t, []string{"key1", "key2"}, keys)
  541. })
  542. }
  543. func TestRedis_Sscan(t *testing.T) {
  544. runOnRedis(t, func(client *Redis) {
  545. key := "list"
  546. var list []string
  547. for i := 0; i < 1550; i++ {
  548. list = append(list, stringx.Randn(i))
  549. }
  550. lens, err := client.Sadd(key, list)
  551. assert.Nil(t, err)
  552. assert.Equal(t, lens, 1550)
  553. var cursor uint64 = 0
  554. sum := 0
  555. for {
  556. _, _, err := New(client.Addr, badType()).Sscan(key, cursor, "", 100)
  557. assert.NotNil(t, err)
  558. keys, next, err := client.Sscan(key, cursor, "", 100)
  559. assert.Nil(t, err)
  560. sum += len(keys)
  561. if next == 0 {
  562. break
  563. }
  564. cursor = next
  565. }
  566. assert.Equal(t, sum, 1550)
  567. _, err = New(client.Addr, badType()).Del(key)
  568. assert.NotNil(t, err)
  569. _, err = client.Del(key)
  570. assert.Nil(t, err)
  571. })
  572. }
  573. func TestRedis_Set(t *testing.T) {
  574. runOnRedis(t, func(client *Redis) {
  575. _, err := New(client.Addr, badType()).Sadd("key", 1, 2, 3, 4)
  576. assert.NotNil(t, err)
  577. num, err := client.Sadd("key", 1, 2, 3, 4)
  578. assert.Nil(t, err)
  579. assert.Equal(t, 4, num)
  580. _, err = New(client.Addr, badType()).Scard("key")
  581. assert.NotNil(t, err)
  582. val, err := client.Scard("key")
  583. assert.Nil(t, err)
  584. assert.Equal(t, int64(4), val)
  585. _, err = New(client.Addr, badType()).Sismember("key", 2)
  586. assert.NotNil(t, err)
  587. ok, err := client.Sismember("key", 2)
  588. assert.Nil(t, err)
  589. assert.True(t, ok)
  590. _, err = New(client.Addr, badType()).Srem("key", 3, 4)
  591. assert.NotNil(t, err)
  592. num, err = client.Srem("key", 3, 4)
  593. assert.Nil(t, err)
  594. assert.Equal(t, 2, num)
  595. _, err = New(client.Addr, badType()).Smembers("key")
  596. assert.NotNil(t, err)
  597. vals, err := client.Smembers("key")
  598. assert.Nil(t, err)
  599. assert.ElementsMatch(t, []string{"1", "2"}, vals)
  600. _, err = New(client.Addr, badType()).Srandmember("key", 1)
  601. assert.NotNil(t, err)
  602. members, err := client.Srandmember("key", 1)
  603. assert.Nil(t, err)
  604. assert.Len(t, members, 1)
  605. assert.Contains(t, []string{"1", "2"}, members[0])
  606. _, err = New(client.Addr, badType()).Spop("key")
  607. assert.NotNil(t, err)
  608. member, err := client.Spop("key")
  609. assert.Nil(t, err)
  610. assert.Contains(t, []string{"1", "2"}, member)
  611. _, err = New(client.Addr, badType()).Smembers("key")
  612. assert.NotNil(t, err)
  613. vals, err = client.Smembers("key")
  614. assert.Nil(t, err)
  615. assert.NotContains(t, vals, member)
  616. _, err = New(client.Addr, badType()).Sadd("key1", 1, 2, 3, 4)
  617. assert.NotNil(t, err)
  618. num, err = client.Sadd("key1", 1, 2, 3, 4)
  619. assert.Nil(t, err)
  620. assert.Equal(t, 4, num)
  621. num, err = client.Sadd("key2", 2, 3, 4, 5)
  622. assert.Nil(t, err)
  623. assert.Equal(t, 4, num)
  624. _, err = New(client.Addr, badType()).Sunion("key1", "key2")
  625. assert.NotNil(t, err)
  626. vals, err = client.Sunion("key1", "key2")
  627. assert.Nil(t, err)
  628. assert.ElementsMatch(t, []string{"1", "2", "3", "4", "5"}, vals)
  629. _, err = New(client.Addr, badType()).Sunionstore("key3", "key1", "key2")
  630. assert.NotNil(t, err)
  631. num, err = client.Sunionstore("key3", "key1", "key2")
  632. assert.Nil(t, err)
  633. assert.Equal(t, 5, num)
  634. _, err = New(client.Addr, badType()).Sdiff("key1", "key2")
  635. assert.NotNil(t, err)
  636. vals, err = client.Sdiff("key1", "key2")
  637. assert.Nil(t, err)
  638. assert.EqualValues(t, []string{"1"}, vals)
  639. _, err = New(client.Addr, badType()).Sdiffstore("key4", "key1", "key2")
  640. assert.NotNil(t, err)
  641. num, err = client.Sdiffstore("key4", "key1", "key2")
  642. assert.Nil(t, err)
  643. assert.Equal(t, 1, num)
  644. _, err = New(client.Addr, badType()).Sinter("key1", "key2")
  645. assert.NotNil(t, err)
  646. vals, err = client.Sinter("key1", "key2")
  647. assert.Nil(t, err)
  648. assert.ElementsMatch(t, []string{"2", "3", "4"}, vals)
  649. _, err = New(client.Addr, badType()).Sinterstore("key4", "key1", "key2")
  650. assert.NotNil(t, err)
  651. num, err = client.Sinterstore("key4", "key1", "key2")
  652. assert.Nil(t, err)
  653. assert.Equal(t, 3, num)
  654. })
  655. }
  656. func TestRedis_GetSet(t *testing.T) {
  657. runOnRedis(t, func(client *Redis) {
  658. _, err := New(client.Addr, badType()).GetSet("hello", "world")
  659. assert.NotNil(t, err)
  660. val, err := client.GetSet("hello", "world")
  661. assert.Nil(t, err)
  662. assert.Equal(t, "", val)
  663. val, err = client.Get("hello")
  664. assert.Nil(t, err)
  665. assert.Equal(t, "world", val)
  666. val, err = client.GetSet("hello", "newworld")
  667. assert.Nil(t, err)
  668. assert.Equal(t, "world", val)
  669. val, err = client.Get("hello")
  670. assert.Nil(t, err)
  671. assert.Equal(t, "newworld", val)
  672. ret, err := client.Del("hello")
  673. assert.Nil(t, err)
  674. assert.Equal(t, 1, ret)
  675. })
  676. }
  677. func TestRedis_SetGetDel(t *testing.T) {
  678. runOnRedis(t, func(client *Redis) {
  679. err := New(client.Addr, badType()).Set("hello", "world")
  680. assert.NotNil(t, err)
  681. err = client.Set("hello", "world")
  682. assert.Nil(t, err)
  683. _, err = New(client.Addr, badType()).Get("hello")
  684. assert.NotNil(t, err)
  685. val, err := client.Get("hello")
  686. assert.Nil(t, err)
  687. assert.Equal(t, "world", val)
  688. ret, err := client.Del("hello")
  689. assert.Nil(t, err)
  690. assert.Equal(t, 1, ret)
  691. })
  692. }
  693. func TestRedis_SetExNx(t *testing.T) {
  694. runOnRedis(t, func(client *Redis) {
  695. err := New(client.Addr, badType()).Setex("hello", "world", 5)
  696. assert.NotNil(t, err)
  697. err = client.Setex("hello", "world", 5)
  698. assert.Nil(t, err)
  699. _, err = New(client.Addr, badType()).Setnx("hello", "newworld")
  700. assert.NotNil(t, err)
  701. ok, err := client.Setnx("hello", "newworld")
  702. assert.Nil(t, err)
  703. assert.False(t, ok)
  704. ok, err = client.Setnx("newhello", "newworld")
  705. assert.Nil(t, err)
  706. assert.True(t, ok)
  707. val, err := client.Get("hello")
  708. assert.Nil(t, err)
  709. assert.Equal(t, "world", val)
  710. val, err = client.Get("newhello")
  711. assert.Nil(t, err)
  712. assert.Equal(t, "newworld", val)
  713. ttl, err := client.Ttl("hello")
  714. assert.Nil(t, err)
  715. assert.True(t, ttl > 0)
  716. _, err = New(client.Addr, badType()).SetnxEx("newhello", "newworld", 5)
  717. assert.NotNil(t, err)
  718. ok, err = client.SetnxEx("newhello", "newworld", 5)
  719. assert.Nil(t, err)
  720. assert.False(t, ok)
  721. num, err := client.Del("newhello")
  722. assert.Nil(t, err)
  723. assert.Equal(t, 1, num)
  724. ok, err = client.SetnxEx("newhello", "newworld", 5)
  725. assert.Nil(t, err)
  726. assert.True(t, ok)
  727. val, err = client.Get("newhello")
  728. assert.Nil(t, err)
  729. assert.Equal(t, "newworld", val)
  730. })
  731. }
  732. func TestRedis_SetGetDelHashField(t *testing.T) {
  733. runOnRedis(t, func(client *Redis) {
  734. err := client.Hset("key", "field", "value")
  735. assert.Nil(t, err)
  736. _, err = New(client.Addr, badType()).Hget("key", "field")
  737. assert.NotNil(t, err)
  738. val, err := client.Hget("key", "field")
  739. assert.Nil(t, err)
  740. assert.Equal(t, "value", val)
  741. _, err = New(client.Addr, badType()).Hexists("key", "field")
  742. assert.NotNil(t, err)
  743. ok, err := client.Hexists("key", "field")
  744. assert.Nil(t, err)
  745. assert.True(t, ok)
  746. _, err = New(client.Addr, badType()).Hdel("key", "field")
  747. assert.NotNil(t, err)
  748. ret, err := client.Hdel("key", "field")
  749. assert.Nil(t, err)
  750. assert.True(t, ret)
  751. ok, err = client.Hexists("key", "field")
  752. assert.Nil(t, err)
  753. assert.False(t, ok)
  754. })
  755. }
  756. func TestRedis_SortedSet(t *testing.T) {
  757. runOnRedis(t, func(client *Redis) {
  758. ok, err := client.ZaddFloat("key", 1, "value1")
  759. assert.Nil(t, err)
  760. assert.True(t, ok)
  761. ok, err = client.Zadd("key", 2, "value1")
  762. assert.Nil(t, err)
  763. assert.False(t, ok)
  764. val, err := client.Zscore("key", "value1")
  765. assert.Nil(t, err)
  766. assert.Equal(t, int64(2), val)
  767. _, err = New(client.Addr, badType()).Zincrby("key", 3, "value1")
  768. assert.NotNil(t, err)
  769. val, err = client.Zincrby("key", 3, "value1")
  770. assert.Nil(t, err)
  771. assert.Equal(t, int64(5), val)
  772. _, err = New(client.Addr, badType()).Zscore("key", "value1")
  773. assert.NotNil(t, err)
  774. val, err = client.Zscore("key", "value1")
  775. assert.Nil(t, err)
  776. assert.Equal(t, int64(5), val)
  777. _, err = New(client.Addr, badType()).Zadds("key")
  778. assert.NotNil(t, err)
  779. val, err = client.Zadds("key", Pair{
  780. Key: "value2",
  781. Score: 6,
  782. }, Pair{
  783. Key: "value3",
  784. Score: 7,
  785. })
  786. assert.Nil(t, err)
  787. assert.Equal(t, int64(2), val)
  788. _, err = New(client.Addr, badType()).ZRevRangeWithScores("key", 1, 3)
  789. assert.NotNil(t, err)
  790. pairs, err := client.ZRevRangeWithScores("key", 1, 3)
  791. assert.Nil(t, err)
  792. assert.EqualValues(t, []Pair{
  793. {
  794. Key: "value2",
  795. Score: 6,
  796. },
  797. {
  798. Key: "value1",
  799. Score: 5,
  800. },
  801. }, pairs)
  802. rank, err := client.Zrank("key", "value2")
  803. assert.Nil(t, err)
  804. assert.Equal(t, int64(1), rank)
  805. rank, err = client.Zrevrank("key", "value1")
  806. assert.Nil(t, err)
  807. assert.Equal(t, int64(2), rank)
  808. _, err = New(client.Addr, badType()).Zrank("key", "value4")
  809. assert.NotNil(t, err)
  810. _, err = client.Zrank("key", "value4")
  811. assert.Equal(t, Nil, err)
  812. _, err = New(client.Addr, badType()).Zrem("key", "value2", "value3")
  813. assert.NotNil(t, err)
  814. num, err := client.Zrem("key", "value2", "value3")
  815. assert.Nil(t, err)
  816. assert.Equal(t, 2, num)
  817. ok, err = client.Zadd("key", 6, "value2")
  818. assert.Nil(t, err)
  819. assert.True(t, ok)
  820. ok, err = client.Zadd("key", 7, "value3")
  821. assert.Nil(t, err)
  822. assert.True(t, ok)
  823. ok, err = client.Zadd("key", 8, "value4")
  824. assert.Nil(t, err)
  825. assert.True(t, ok)
  826. _, err = New(client.Addr, badType()).Zremrangebyscore("key", 6, 7)
  827. assert.NotNil(t, err)
  828. num, err = client.Zremrangebyscore("key", 6, 7)
  829. assert.Nil(t, err)
  830. assert.Equal(t, 2, num)
  831. ok, err = client.Zadd("key", 6, "value2")
  832. assert.Nil(t, err)
  833. assert.True(t, ok)
  834. _, err = New(client.Addr, badType()).Zadd("key", 7, "value3")
  835. assert.NotNil(t, err)
  836. ok, err = client.Zadd("key", 7, "value3")
  837. assert.Nil(t, err)
  838. assert.True(t, ok)
  839. _, err = New(client.Addr, badType()).Zcount("key", 6, 7)
  840. assert.NotNil(t, err)
  841. num, err = client.Zcount("key", 6, 7)
  842. assert.Nil(t, err)
  843. assert.Equal(t, 2, num)
  844. _, err = New(client.Addr, badType()).Zremrangebyrank("key", 1, 2)
  845. assert.NotNil(t, err)
  846. num, err = client.Zremrangebyrank("key", 1, 2)
  847. assert.Nil(t, err)
  848. assert.Equal(t, 2, num)
  849. _, err = New(client.Addr, badType()).Zcard("key")
  850. assert.NotNil(t, err)
  851. card, err := client.Zcard("key")
  852. assert.Nil(t, err)
  853. assert.Equal(t, 2, card)
  854. _, err = New(client.Addr, badType()).Zrange("key", 0, -1)
  855. assert.NotNil(t, err)
  856. vals, err := client.Zrange("key", 0, -1)
  857. assert.Nil(t, err)
  858. assert.EqualValues(t, []string{"value1", "value4"}, vals)
  859. _, err = New(client.Addr, badType()).Zrevrange("key", 0, -1)
  860. assert.NotNil(t, err)
  861. vals, err = client.Zrevrange("key", 0, -1)
  862. assert.Nil(t, err)
  863. assert.EqualValues(t, []string{"value4", "value1"}, vals)
  864. _, err = New(client.Addr, badType()).ZrangeWithScores("key", 0, -1)
  865. assert.NotNil(t, err)
  866. pairs, err = client.ZrangeWithScores("key", 0, -1)
  867. assert.Nil(t, err)
  868. assert.EqualValues(t, []Pair{
  869. {
  870. Key: "value1",
  871. Score: 5,
  872. },
  873. {
  874. Key: "value4",
  875. Score: 8,
  876. },
  877. }, pairs)
  878. _, err = New(client.Addr, badType()).ZrangebyscoreWithScores("key", 5, 8)
  879. assert.NotNil(t, err)
  880. pairs, err = client.ZrangebyscoreWithScores("key", 5, 8)
  881. assert.Nil(t, err)
  882. assert.EqualValues(t, []Pair{
  883. {
  884. Key: "value1",
  885. Score: 5,
  886. },
  887. {
  888. Key: "value4",
  889. Score: 8,
  890. },
  891. }, pairs)
  892. _, err = New(client.Addr, badType()).ZrangebyscoreWithScoresAndLimit(
  893. "key", 5, 8, 1, 1)
  894. assert.NotNil(t, err)
  895. pairs, err = client.ZrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 1)
  896. assert.Nil(t, err)
  897. assert.EqualValues(t, []Pair{
  898. {
  899. Key: "value4",
  900. Score: 8,
  901. },
  902. }, pairs)
  903. pairs, err = client.ZrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 0)
  904. assert.Nil(t, err)
  905. assert.Equal(t, 0, len(pairs))
  906. _, err = New(client.Addr, badType()).ZrevrangebyscoreWithScores("key", 5, 8)
  907. assert.NotNil(t, err)
  908. pairs, err = client.ZrevrangebyscoreWithScores("key", 5, 8)
  909. assert.Nil(t, err)
  910. assert.EqualValues(t, []Pair{
  911. {
  912. Key: "value4",
  913. Score: 8,
  914. },
  915. {
  916. Key: "value1",
  917. Score: 5,
  918. },
  919. }, pairs)
  920. _, err = New(client.Addr, badType()).ZrevrangebyscoreWithScoresAndLimit(
  921. "key", 5, 8, 1, 1)
  922. assert.NotNil(t, err)
  923. pairs, err = client.ZrevrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 1)
  924. assert.Nil(t, err)
  925. assert.EqualValues(t, []Pair{
  926. {
  927. Key: "value1",
  928. Score: 5,
  929. },
  930. }, pairs)
  931. pairs, err = client.ZrevrangebyscoreWithScoresAndLimit("key", 5, 8, 1, 0)
  932. assert.Nil(t, err)
  933. assert.Equal(t, 0, len(pairs))
  934. _, err = New(client.Addr, badType()).Zrevrank("key", "value")
  935. assert.NotNil(t, err)
  936. _, _ = client.Zadd("second", 2, "aa")
  937. _, _ = client.Zadd("third", 3, "bbb")
  938. val, err = client.Zunionstore("union", &ZStore{
  939. Keys: []string{"second", "third"},
  940. Weights: []float64{1, 2},
  941. Aggregate: "SUM",
  942. })
  943. assert.Nil(t, err)
  944. assert.Equal(t, int64(2), val)
  945. _, err = New(client.Addr, badType()).Zunionstore("union", &ZStore{})
  946. assert.NotNil(t, err)
  947. vals, err = client.Zrange("union", 0, 10000)
  948. assert.Nil(t, err)
  949. assert.EqualValues(t, []string{"aa", "bbb"}, vals)
  950. ival, err := client.Zcard("union")
  951. assert.Nil(t, err)
  952. assert.Equal(t, 2, ival)
  953. })
  954. }
  955. func TestRedis_SortedSetByFloat64(t *testing.T) {
  956. runOnRedis(t, func(client *Redis) {
  957. ok, err := client.ZaddFloat("key", 10.345, "value1")
  958. assert.Nil(t, err)
  959. assert.True(t, ok)
  960. val, err := client.ZscoreByFloat("key", "value1")
  961. assert.Nil(t, err)
  962. assert.Equal(t, 10.345, val)
  963. _, _ = client.ZaddFloat("key", 10.346, "value2")
  964. _, err = New(client.Addr, badType()).ZRevRangeWithScoresByFloat("key", 0, -1)
  965. assert.NotNil(t, err)
  966. pairs, err := client.ZRevRangeWithScoresByFloat("key", 0, -1)
  967. assert.Nil(t, err)
  968. assert.EqualValues(t, []FloatPair{
  969. {
  970. Key: "value2",
  971. Score: 10.346,
  972. },
  973. {
  974. Key: "value1",
  975. Score: 10.345,
  976. },
  977. }, pairs)
  978. pairs, err = client.ZrangeWithScoresByFloat("key", 0, -1)
  979. assert.Nil(t, err)
  980. assert.EqualValues(t, []FloatPair{
  981. {
  982. Key: "value1",
  983. Score: 10.345,
  984. },
  985. {
  986. Key: "value2",
  987. Score: 10.346,
  988. },
  989. }, pairs)
  990. _, err = New(client.Addr, badType()).ZrangebyscoreWithScoresByFloat("key", 0, 20)
  991. assert.NotNil(t, err)
  992. pairs, err = client.ZrangebyscoreWithScoresByFloat("key", 0, 20)
  993. assert.Nil(t, err)
  994. assert.EqualValues(t, []FloatPair{
  995. {
  996. Key: "value1",
  997. Score: 10.345,
  998. },
  999. {
  1000. Key: "value2",
  1001. Score: 10.346,
  1002. },
  1003. }, pairs)
  1004. _, err = New(client.Addr, badType()).ZrangebyscoreWithScoresByFloatAndLimit(
  1005. "key", 10.1, 12.2, 1, 1)
  1006. assert.NotNil(t, err)
  1007. pairs, err = client.ZrangebyscoreWithScoresByFloatAndLimit("key", 10.1, 12.2, 1, 1)
  1008. assert.Nil(t, err)
  1009. assert.EqualValues(t, []FloatPair{
  1010. {
  1011. Key: "value2",
  1012. Score: 10.346,
  1013. },
  1014. }, pairs)
  1015. _, err = New(client.Addr, badType()).ZrevrangebyscoreWithScoresByFloat("key", 10, 12)
  1016. assert.NotNil(t, err)
  1017. pairs, err = client.ZrevrangebyscoreWithScoresByFloat("key", 10, 12)
  1018. assert.Nil(t, err)
  1019. assert.EqualValues(t, []FloatPair{
  1020. {
  1021. Key: "value2",
  1022. Score: 10.346,
  1023. },
  1024. {
  1025. Key: "value1",
  1026. Score: 10.345,
  1027. },
  1028. }, pairs)
  1029. _, err = New(client.Addr, badType()).ZrevrangebyscoreWithScoresByFloatAndLimit(
  1030. "key", 10, 12, 1, 1)
  1031. assert.NotNil(t, err)
  1032. pairs, err = client.ZrevrangebyscoreWithScoresByFloatAndLimit("key", 10, 12, 1, 1)
  1033. assert.Nil(t, err)
  1034. assert.EqualValues(t, []FloatPair{
  1035. {
  1036. Key: "value1",
  1037. Score: 10.345,
  1038. },
  1039. }, pairs)
  1040. })
  1041. }
  1042. func TestRedis_IncrbyFloat(t *testing.T) {
  1043. runOnRedis(t, func(client *Redis) {
  1044. incrVal, err := client.IncrbyFloat("key", 0.002)
  1045. assert.Nil(t, err)
  1046. assert.Equal(t, 0.002, incrVal)
  1047. incrVal2, err := client.IncrbyFloat("key", -0.001)
  1048. assert.Nil(t, err)
  1049. assert.Equal(t, 0.001, incrVal2)
  1050. hincrVal, err := client.HincrbyFloat("hkey", "i", 0.002)
  1051. assert.Nil(t, err)
  1052. assert.Equal(t, 0.002, hincrVal)
  1053. hincrVal2, err := client.HincrbyFloat("hkey", "i", -0.001)
  1054. assert.Nil(t, err)
  1055. assert.Equal(t, 0.001, hincrVal2)
  1056. })
  1057. }
  1058. func TestRedis_Pipelined(t *testing.T) {
  1059. runOnRedis(t, func(client *Redis) {
  1060. assert.NotNil(t, New(client.Addr, badType()).Pipelined(func(pipeliner Pipeliner) error {
  1061. return nil
  1062. }))
  1063. err := client.Pipelined(
  1064. func(pipe Pipeliner) error {
  1065. pipe.Incr(context.Background(), "pipelined_counter")
  1066. pipe.Expire(context.Background(), "pipelined_counter", time.Hour)
  1067. pipe.ZAdd(context.Background(), "zadd", &Z{Score: 12, Member: "zadd"})
  1068. return nil
  1069. },
  1070. )
  1071. assert.Nil(t, err)
  1072. _, err = New(client.Addr, badType()).Ttl("pipelined_counter")
  1073. assert.NotNil(t, err)
  1074. ttl, err := client.Ttl("pipelined_counter")
  1075. assert.Nil(t, err)
  1076. assert.Equal(t, 3600, ttl)
  1077. value, err := client.Get("pipelined_counter")
  1078. assert.Nil(t, err)
  1079. assert.Equal(t, "1", value)
  1080. score, err := client.Zscore("zadd", "zadd")
  1081. assert.Nil(t, err)
  1082. assert.Equal(t, int64(12), score)
  1083. })
  1084. }
  1085. func TestRedisString(t *testing.T) {
  1086. runOnRedis(t, func(client *Redis) {
  1087. client.Ping()
  1088. _, err := getRedis(New(client.Addr, Cluster()))
  1089. assert.Nil(t, err)
  1090. assert.Equal(t, client.Addr, client.String())
  1091. assert.NotNil(t, New(client.Addr, badType()).Ping())
  1092. })
  1093. }
  1094. func TestRedisScriptLoad(t *testing.T) {
  1095. runOnRedis(t, func(client *Redis) {
  1096. client.Ping()
  1097. _, err := New(client.Addr, badType()).ScriptLoad("foo")
  1098. assert.NotNil(t, err)
  1099. _, err = client.ScriptLoad("foo")
  1100. assert.NotNil(t, err)
  1101. })
  1102. }
  1103. func TestRedisEvalSha(t *testing.T) {
  1104. runOnRedis(t, func(client *Redis) {
  1105. client.Ping()
  1106. scriptHash, err := client.ScriptLoad(`return redis.call("EXISTS", KEYS[1])`)
  1107. assert.Nil(t, err)
  1108. result, err := client.EvalSha(scriptHash, []string{"key1"})
  1109. assert.Nil(t, err)
  1110. assert.Equal(t, int64(0), result)
  1111. })
  1112. }
  1113. func TestRedisToPairs(t *testing.T) {
  1114. pairs := toPairs([]red.Z{
  1115. {
  1116. Member: 1,
  1117. Score: 1,
  1118. },
  1119. {
  1120. Member: 2,
  1121. Score: 2,
  1122. },
  1123. })
  1124. assert.EqualValues(t, []Pair{
  1125. {
  1126. Key: "1",
  1127. Score: 1,
  1128. },
  1129. {
  1130. Key: "2",
  1131. Score: 2,
  1132. },
  1133. }, pairs)
  1134. }
  1135. func TestRedis_Zscan(t *testing.T) {
  1136. runOnRedis(t, func(client *Redis) {
  1137. key := "key"
  1138. for i := 0; i < 1550; i++ {
  1139. ok, err := client.Zadd(key, int64(i), "value_"+strconv.Itoa(i))
  1140. assert.Nil(t, err)
  1141. assert.True(t, ok)
  1142. }
  1143. var cursor uint64 = 0
  1144. sum := 0
  1145. for {
  1146. _, _, err := New(client.Addr, badType()).Zscan(key, cursor, "value_*", 100)
  1147. assert.NotNil(t, err)
  1148. keys, next, err := client.Zscan(key, cursor, "value_*", 100)
  1149. assert.Nil(t, err)
  1150. sum += len(keys)
  1151. if next == 0 {
  1152. break
  1153. }
  1154. cursor = next
  1155. }
  1156. assert.Equal(t, sum, 3100)
  1157. _, err := New(client.Addr, badType()).Del(key)
  1158. assert.NotNil(t, err)
  1159. _, err = client.Del(key)
  1160. assert.Nil(t, err)
  1161. })
  1162. }
  1163. func TestRedisToStrings(t *testing.T) {
  1164. vals := toStrings([]interface{}{1, 2})
  1165. assert.EqualValues(t, []string{"1", "2"}, vals)
  1166. }
  1167. func TestRedisBlpop(t *testing.T) {
  1168. runOnRedis(t, func(client *Redis) {
  1169. client.Ping()
  1170. var node mockedNode
  1171. _, err := client.Blpop(nil, "foo")
  1172. assert.NotNil(t, err)
  1173. _, err = client.Blpop(node, "foo")
  1174. assert.NotNil(t, err)
  1175. })
  1176. }
  1177. func TestRedisBlpopEx(t *testing.T) {
  1178. runOnRedis(t, func(client *Redis) {
  1179. client.Ping()
  1180. var node mockedNode
  1181. _, _, err := client.BlpopEx(nil, "foo")
  1182. assert.NotNil(t, err)
  1183. _, _, err = client.BlpopEx(node, "foo")
  1184. assert.NotNil(t, err)
  1185. })
  1186. }
  1187. func TestRedisBlpopWithTimeout(t *testing.T) {
  1188. runOnRedis(t, func(client *Redis) {
  1189. client.Ping()
  1190. var node mockedNode
  1191. _, err := client.BlpopWithTimeout(nil, 10*time.Second, "foo")
  1192. assert.NotNil(t, err)
  1193. _, err = client.BlpopWithTimeout(node, 10*time.Second, "foo")
  1194. assert.NotNil(t, err)
  1195. })
  1196. }
  1197. func TestRedisGeo(t *testing.T) {
  1198. runOnRedis(t, func(client *Redis) {
  1199. client.Ping()
  1200. geoLocation := []*GeoLocation{{Longitude: 13.361389, Latitude: 38.115556, Name: "Palermo"}, {Longitude: 15.087269, Latitude: 37.502669, Name: "Catania"}}
  1201. v, err := client.GeoAdd("sicily", geoLocation...)
  1202. assert.Nil(t, err)
  1203. assert.Equal(t, int64(2), v)
  1204. v2, err := client.GeoDist("sicily", "Palermo", "Catania", "m")
  1205. assert.Nil(t, err)
  1206. assert.Equal(t, 166274, int(v2))
  1207. // GeoHash not support
  1208. v3, err := client.GeoPos("sicily", "Palermo", "Catania")
  1209. assert.Nil(t, err)
  1210. assert.Equal(t, int64(v3[0].Longitude), int64(13))
  1211. assert.Equal(t, int64(v3[0].Latitude), int64(38))
  1212. assert.Equal(t, int64(v3[1].Longitude), int64(15))
  1213. assert.Equal(t, int64(v3[1].Latitude), int64(37))
  1214. v4, err := client.GeoRadius("sicily", 15, 37, &red.GeoRadiusQuery{WithDist: true, Unit: "km", Radius: 200})
  1215. assert.Nil(t, err)
  1216. assert.Equal(t, int64(v4[0].Dist), int64(190))
  1217. assert.Equal(t, int64(v4[1].Dist), int64(56))
  1218. geoLocation2 := []*GeoLocation{{Longitude: 13.583333, Latitude: 37.316667, Name: "Agrigento"}}
  1219. v5, err := client.GeoAdd("sicily", geoLocation2...)
  1220. assert.Nil(t, err)
  1221. assert.Equal(t, int64(1), v5)
  1222. v6, err := client.GeoRadiusByMember("sicily", "Agrigento", &red.GeoRadiusQuery{Unit: "km", Radius: 100})
  1223. assert.Nil(t, err)
  1224. assert.Equal(t, v6[0].Name, "Agrigento")
  1225. assert.Equal(t, v6[1].Name, "Palermo")
  1226. })
  1227. }
  1228. func TestSetSlowThreshold(t *testing.T) {
  1229. assert.Equal(t, defaultSlowThreshold, slowThreshold.Load())
  1230. SetSlowThreshold(time.Second)
  1231. assert.Equal(t, time.Second, slowThreshold.Load())
  1232. }
  1233. func TestRedis_WithPass(t *testing.T) {
  1234. runOnRedis(t, func(client *Redis) {
  1235. err := New(client.Addr, WithPass("any")).Ping()
  1236. assert.NotNil(t, err)
  1237. })
  1238. }
  1239. func runOnRedis(t *testing.T, fn func(client *Redis)) {
  1240. logx.Disable()
  1241. s, err := miniredis.Run()
  1242. assert.Nil(t, err)
  1243. defer func() {
  1244. client, err := clientManager.GetResource(s.Addr(), func() (io.Closer, error) {
  1245. return nil, errors.New("should already exist")
  1246. })
  1247. if err != nil {
  1248. t.Error(err)
  1249. }
  1250. if client != nil {
  1251. _ = client.Close()
  1252. }
  1253. }()
  1254. fn(New(s.Addr()))
  1255. }
  1256. func runOnRedisTLS(t *testing.T, fn func(client *Redis)) {
  1257. logx.Disable()
  1258. s, err := miniredis.RunTLS(&tls.Config{
  1259. Certificates: make([]tls.Certificate, 1),
  1260. InsecureSkipVerify: true,
  1261. })
  1262. assert.Nil(t, err)
  1263. defer func() {
  1264. client, err := clientManager.GetResource(s.Addr(), func() (io.Closer, error) {
  1265. return nil, errors.New("should already exist")
  1266. })
  1267. if err != nil {
  1268. t.Error(err)
  1269. }
  1270. if client != nil {
  1271. _ = client.Close()
  1272. }
  1273. }()
  1274. fn(New(s.Addr(), WithTLS()))
  1275. }
  1276. func badType() Option {
  1277. return func(r *Redis) {
  1278. r.Type = "bad"
  1279. }
  1280. }
  1281. type mockedNode struct {
  1282. RedisNode
  1283. }
  1284. func (n mockedNode) BLPop(_ context.Context, _ time.Duration, _ ...string) *red.StringSliceCmd {
  1285. return red.NewStringSliceCmd(context.Background(), "foo", "bar")
  1286. }