redis_test.go 49 KB

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