scripting.tcl 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743
  1. start_server {tags {"scripting"}} {
  2. test {EVAL - Does Lua interpreter replies to our requests?} {
  3. r eval {return 'hello'} 0
  4. } {hello}
  5. test {EVAL - Lua integer -> Redis protocol type conversion} {
  6. r eval {return 100.5} 0
  7. } {100}
  8. test {EVAL - Lua string -> Redis protocol type conversion} {
  9. r eval {return 'hello world'} 0
  10. } {hello world}
  11. test {EVAL - Lua true boolean -> Redis protocol type conversion} {
  12. r eval {return true} 0
  13. } {1}
  14. test {EVAL - Lua false boolean -> Redis protocol type conversion} {
  15. r eval {return false} 0
  16. } {}
  17. test {EVAL - Lua status code reply -> Redis protocol type conversion} {
  18. r eval {return {ok='fine'}} 0
  19. } {fine}
  20. test {EVAL - Lua error reply -> Redis protocol type conversion} {
  21. catch {
  22. r eval {return {err='this is an error'}} 0
  23. } e
  24. set _ $e
  25. } {this is an error}
  26. test {EVAL - Lua table -> Redis protocol type conversion} {
  27. r eval {return {1,2,3,'ciao',{1,2}}} 0
  28. } {1 2 3 ciao {1 2}}
  29. test {EVAL - Are the KEYS and ARGV arrays populated correctly?} {
  30. r eval {return {KEYS[1],KEYS[2],ARGV[1],ARGV[2]}} 2 a b c d
  31. } {a b c d}
  32. test {EVAL - is Lua able to call Redis API?} {
  33. r set mykey myval
  34. r eval {return redis.call('get',KEYS[1])} 1 mykey
  35. } {myval}
  36. test {EVALSHA - Can we call a SHA1 if already defined?} {
  37. r evalsha fd758d1589d044dd850a6f05d52f2eefd27f033f 1 mykey
  38. } {myval}
  39. test {EVALSHA - Can we call a SHA1 in uppercase?} {
  40. r evalsha FD758D1589D044DD850A6F05D52F2EEFD27F033F 1 mykey
  41. } {myval}
  42. test {EVALSHA - Do we get an error on invalid SHA1?} {
  43. catch {r evalsha NotValidShaSUM 0} e
  44. set _ $e
  45. } {NOSCRIPT*}
  46. test {EVALSHA - Do we get an error on non defined SHA1?} {
  47. catch {r evalsha ffd632c7d33e571e9f24556ebed26c3479a87130 0} e
  48. set _ $e
  49. } {NOSCRIPT*}
  50. test {EVAL - Redis integer -> Lua type conversion} {
  51. r set x 0
  52. r eval {
  53. local foo = redis.pcall('incr',KEYS[1])
  54. return {type(foo),foo}
  55. } 1 x
  56. } {number 1}
  57. test {EVAL - Redis bulk -> Lua type conversion} {
  58. r set mykey myval
  59. r eval {
  60. local foo = redis.pcall('get',KEYS[1])
  61. return {type(foo),foo}
  62. } 1 mykey
  63. } {string myval}
  64. test {EVAL - Redis multi bulk -> Lua type conversion} {
  65. r del mylist
  66. r rpush mylist a
  67. r rpush mylist b
  68. r rpush mylist c
  69. r eval {
  70. local foo = redis.pcall('lrange',KEYS[1],0,-1)
  71. return {type(foo),foo[1],foo[2],foo[3],# foo}
  72. } 1 mylist
  73. } {table a b c 3}
  74. test {EVAL - Redis status reply -> Lua type conversion} {
  75. r eval {
  76. local foo = redis.pcall('set',KEYS[1],'myval')
  77. return {type(foo),foo['ok']}
  78. } 1 mykey
  79. } {table OK}
  80. test {EVAL - Redis error reply -> Lua type conversion} {
  81. r set mykey myval
  82. r eval {
  83. local foo = redis.pcall('incr',KEYS[1])
  84. return {type(foo),foo['err']}
  85. } 1 mykey
  86. } {table {ERR value is not an integer or out of range}}
  87. test {EVAL - Redis nil bulk reply -> Lua type conversion} {
  88. r del mykey
  89. r eval {
  90. local foo = redis.pcall('get',KEYS[1])
  91. return {type(foo),foo == false}
  92. } 1 mykey
  93. } {boolean 1}
  94. test {EVAL - Is the Lua client using the currently selected DB?} {
  95. r set mykey "this is DB 9"
  96. r select 10
  97. r set mykey "this is DB 10"
  98. r eval {return redis.pcall('get',KEYS[1])} 1 mykey
  99. } {this is DB 10}
  100. test {EVAL - SELECT inside Lua should not affect the caller} {
  101. # here we DB 10 is selected
  102. r set mykey "original value"
  103. r eval {return redis.pcall('select','9')} 0
  104. set res [r get mykey]
  105. r select 9
  106. set res
  107. } {original value}
  108. if 0 {
  109. test {EVAL - Script can't run more than configured time limit} {
  110. r config set lua-time-limit 1
  111. catch {
  112. r eval {
  113. local i = 0
  114. while true do i=i+1 end
  115. } 0
  116. } e
  117. set _ $e
  118. } {*execution time*}
  119. }
  120. test {EVAL - Scripts can't run certain commands} {
  121. set e {}
  122. catch {r eval {return redis.pcall('blpop','x',0)} 0} e
  123. set e
  124. } {*not allowed*}
  125. test {EVAL - Scripts can't run certain commands} {
  126. set e {}
  127. r debug lua-always-replicate-commands 0
  128. catch {
  129. r eval "redis.pcall('randomkey'); return redis.pcall('set','x','ciao')" 0
  130. } e
  131. r debug lua-always-replicate-commands 1
  132. set e
  133. } {*not allowed after*}
  134. test {EVAL - No arguments to redis.call/pcall is considered an error} {
  135. set e {}
  136. catch {r eval {return redis.call()} 0} e
  137. set e
  138. } {*one argument*}
  139. test {EVAL - redis.call variant raises a Lua error on Redis cmd error (1)} {
  140. set e {}
  141. catch {
  142. r eval "redis.call('nosuchcommand')" 0
  143. } e
  144. set e
  145. } {*Unknown Redis*}
  146. test {EVAL - redis.call variant raises a Lua error on Redis cmd error (1)} {
  147. set e {}
  148. catch {
  149. r eval "redis.call('get','a','b','c')" 0
  150. } e
  151. set e
  152. } {*number of args*}
  153. test {EVAL - redis.call variant raises a Lua error on Redis cmd error (1)} {
  154. set e {}
  155. r set foo bar
  156. catch {
  157. r eval {redis.call('lpush',KEYS[1],'val')} 1 foo
  158. } e
  159. set e
  160. } {*against a key*}
  161. test {EVAL - JSON numeric decoding} {
  162. # We must return the table as a string because otherwise
  163. # Redis converts floats to ints and we get 0 and 1023 instead
  164. # of 0.0003 and 1023.2 as the parsed output.
  165. r eval {return
  166. table.concat(
  167. cjson.decode(
  168. "[0.0, -5e3, -1, 0.3e-3, 1023.2, 0e10]"), " ")
  169. } 0
  170. } {0 -5000 -1 0.0003 1023.2 0}
  171. test {EVAL - JSON string decoding} {
  172. r eval {local decoded = cjson.decode('{"keya": "a", "keyb": "b"}')
  173. return {decoded.keya, decoded.keyb}
  174. } 0
  175. } {a b}
  176. test {EVAL - cmsgpack can pack double?} {
  177. r eval {local encoded = cmsgpack.pack(0.1)
  178. local h = ""
  179. for i = 1, #encoded do
  180. h = h .. string.format("%02x",string.byte(encoded,i))
  181. end
  182. return h
  183. } 0
  184. } {cb3fb999999999999a}
  185. test {EVAL - cmsgpack can pack negative int64?} {
  186. r eval {local encoded = cmsgpack.pack(-1099511627776)
  187. local h = ""
  188. for i = 1, #encoded do
  189. h = h .. string.format("%02x",string.byte(encoded,i))
  190. end
  191. return h
  192. } 0
  193. } {d3ffffff0000000000}
  194. test {EVAL - cmsgpack can pack and unpack circular references?} {
  195. r eval {local a = {x=nil,y=5}
  196. local b = {x=a}
  197. a['x'] = b
  198. local encoded = cmsgpack.pack(a)
  199. local h = ""
  200. -- cmsgpack encodes to a depth of 16, but can't encode
  201. -- references, so the encoded object has a deep copy recusive
  202. -- depth of 16.
  203. for i = 1, #encoded do
  204. h = h .. string.format("%02x",string.byte(encoded,i))
  205. end
  206. -- when unpacked, re.x.x != re because the unpack creates
  207. -- individual tables down to a depth of 16.
  208. -- (that's why the encoded output is so large)
  209. local re = cmsgpack.unpack(encoded)
  210. assert(re)
  211. assert(re.x)
  212. assert(re.x.x.y == re.y)
  213. assert(re.x.x.x.x.y == re.y)
  214. assert(re.x.x.x.x.x.x.y == re.y)
  215. assert(re.x.x.x.x.x.x.x.x.x.x.y == re.y)
  216. -- maximum working depth:
  217. assert(re.x.x.x.x.x.x.x.x.x.x.x.x.x.x.y == re.y)
  218. -- now the last x would be b above and has no y
  219. assert(re.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x)
  220. -- so, the final x.x is at the depth limit and was assigned nil
  221. assert(re.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x.x == nil)
  222. return {h, re.x.x.x.x.x.x.x.x.y == re.y, re.y == 5}
  223. } 0
  224. } {82a17905a17881a17882a17905a17881a17882a17905a17881a17882a17905a17881a17882a17905a17881a17882a17905a17881a17882a17905a17881a17882a17905a17881a178c0 1 1}
  225. test {EVAL - Numerical sanity check from bitop} {
  226. r eval {assert(0x7fffffff == 2147483647, "broken hex literals");
  227. assert(0xffffffff == -1 or 0xffffffff == 2^32-1,
  228. "broken hex literals");
  229. assert(tostring(-1) == "-1", "broken tostring()");
  230. assert(tostring(0xffffffff) == "-1" or
  231. tostring(0xffffffff) == "4294967295",
  232. "broken tostring()")
  233. } 0
  234. } {}
  235. test {EVAL - Verify minimal bitop functionality} {
  236. r eval {assert(bit.tobit(1) == 1);
  237. assert(bit.band(1) == 1);
  238. assert(bit.bxor(1,2) == 3);
  239. assert(bit.bor(1,2,4,8,16,32,64,128) == 255)
  240. } 0
  241. } {}
  242. test {EVAL - Able to parse trailing comments} {
  243. r eval {return 'hello' --trailing comment} 0
  244. } {hello}
  245. test {SCRIPTING FLUSH - is able to clear the scripts cache?} {
  246. r set mykey myval
  247. set v [r evalsha fd758d1589d044dd850a6f05d52f2eefd27f033f 1 mykey]
  248. assert_equal $v myval
  249. set e ""
  250. r script flush
  251. catch {r evalsha fd758d1589d044dd850a6f05d52f2eefd27f033f 1 mykey} e
  252. set e
  253. } {NOSCRIPT*}
  254. test {SCRIPT EXISTS - can detect already defined scripts?} {
  255. r eval "return 1+1" 0
  256. r script exists a27e7e8a43702b7046d4f6a7ccf5b60cef6b9bd9 a27e7e8a43702b7046d4f6a7ccf5b60cef6b9bda
  257. } {1 0}
  258. test {SCRIPT LOAD - is able to register scripts in the scripting cache} {
  259. list \
  260. [r script load "return 'loaded'"] \
  261. [r evalsha b534286061d4b9e4026607613b95c06c06015ae8 0]
  262. } {b534286061d4b9e4026607613b95c06c06015ae8 loaded}
  263. test "In the context of Lua the output of random commands gets ordered" {
  264. r debug lua-always-replicate-commands 0
  265. r del myset
  266. r sadd myset a b c d e f g h i l m n o p q r s t u v z aa aaa azz
  267. set res [r eval {return redis.call('smembers',KEYS[1])} 1 myset]
  268. r debug lua-always-replicate-commands 1
  269. set res
  270. } {a aa aaa azz b c d e f g h i l m n o p q r s t u v z}
  271. test "SORT is normally not alpha re-ordered for the scripting engine" {
  272. r del myset
  273. r sadd myset 1 2 3 4 10
  274. r eval {return redis.call('sort',KEYS[1],'desc')} 1 myset
  275. } {10 4 3 2 1}
  276. test "SORT BY <constant> output gets ordered for scripting" {
  277. r del myset
  278. r sadd myset a b c d e f g h i l m n o p q r s t u v z aa aaa azz
  279. r eval {return redis.call('sort',KEYS[1],'by','_')} 1 myset
  280. } {a aa aaa azz b c d e f g h i l m n o p q r s t u v z}
  281. test "SORT BY <constant> with GET gets ordered for scripting" {
  282. r del myset
  283. r sadd myset a b c
  284. r eval {return redis.call('sort',KEYS[1],'by','_','get','#','get','_:*')} 1 myset
  285. } {a {} b {} c {}}
  286. test "redis.sha1hex() implementation" {
  287. list [r eval {return redis.sha1hex('')} 0] \
  288. [r eval {return redis.sha1hex('Pizza & Mandolino')} 0]
  289. } {da39a3ee5e6b4b0d3255bfef95601890afd80709 74822d82031af7493c20eefa13bd07ec4fada82f}
  290. test {Globals protection reading an undeclared global variable} {
  291. catch {r eval {return a} 0} e
  292. set e
  293. } {*ERR*attempted to access * global*}
  294. test {Globals protection setting an undeclared global*} {
  295. catch {r eval {a=10} 0} e
  296. set e
  297. } {*ERR*attempted to create global*}
  298. test {Test an example script DECR_IF_GT} {
  299. set decr_if_gt {
  300. local current
  301. current = redis.call('get',KEYS[1])
  302. if not current then return nil end
  303. if current > ARGV[1] then
  304. return redis.call('decr',KEYS[1])
  305. else
  306. return redis.call('get',KEYS[1])
  307. end
  308. }
  309. r set foo 5
  310. set res {}
  311. lappend res [r eval $decr_if_gt 1 foo 2]
  312. lappend res [r eval $decr_if_gt 1 foo 2]
  313. lappend res [r eval $decr_if_gt 1 foo 2]
  314. lappend res [r eval $decr_if_gt 1 foo 2]
  315. lappend res [r eval $decr_if_gt 1 foo 2]
  316. set res
  317. } {4 3 2 2 2}
  318. test {Scripting engine resets PRNG at every script execution} {
  319. set rand1 [r eval {return tostring(math.random())} 0]
  320. set rand2 [r eval {return tostring(math.random())} 0]
  321. assert_equal $rand1 $rand2
  322. }
  323. test {Scripting engine PRNG can be seeded correctly} {
  324. set rand1 [r eval {
  325. math.randomseed(ARGV[1]); return tostring(math.random())
  326. } 0 10]
  327. set rand2 [r eval {
  328. math.randomseed(ARGV[1]); return tostring(math.random())
  329. } 0 10]
  330. set rand3 [r eval {
  331. math.randomseed(ARGV[1]); return tostring(math.random())
  332. } 0 20]
  333. assert_equal $rand1 $rand2
  334. assert {$rand2 ne $rand3}
  335. }
  336. test {EVAL does not leak in the Lua stack} {
  337. r set x 0
  338. # Use a non blocking client to speedup the loop.
  339. set rd [redis_deferring_client]
  340. for {set j 0} {$j < 10000} {incr j} {
  341. $rd eval {return redis.call("incr",KEYS[1])} 1 x
  342. }
  343. for {set j 0} {$j < 10000} {incr j} {
  344. $rd read
  345. }
  346. assert {[s used_memory_lua] < 1024*100}
  347. $rd close
  348. r get x
  349. } {10000}
  350. test {EVAL processes writes from AOF in read-only slaves} {
  351. r flushall
  352. r config set appendonly yes
  353. r config set aof-use-rdb-preamble no
  354. r eval {redis.call("set",KEYS[1],"100")} 1 foo
  355. r eval {redis.call("incr",KEYS[1])} 1 foo
  356. r eval {redis.call("incr",KEYS[1])} 1 foo
  357. wait_for_condition 50 100 {
  358. [s aof_rewrite_in_progress] == 0
  359. } else {
  360. fail "AOF rewrite can't complete after CONFIG SET appendonly yes."
  361. }
  362. r config set slave-read-only yes
  363. r slaveof 127.0.0.1 0
  364. r debug loadaof
  365. set res [r get foo]
  366. r slaveof no one
  367. set res
  368. } {102}
  369. test {We can call scripts rewriting client->argv from Lua} {
  370. r del myset
  371. r sadd myset a b c
  372. r mset a 1 b 2 c 3 d 4
  373. assert {[r spop myset] ne {}}
  374. assert {[r spop myset 1] ne {}}
  375. assert {[r spop myset] ne {}}
  376. assert {[r mget a b c d] eq {1 2 3 4}}
  377. assert {[r spop myset] eq {}}
  378. }
  379. test {Call Redis command with many args from Lua (issue #1764)} {
  380. r eval {
  381. local i
  382. local x={}
  383. redis.call('del','mylist')
  384. for i=1,100 do
  385. table.insert(x,i)
  386. end
  387. redis.call('rpush','mylist',unpack(x))
  388. return redis.call('lrange','mylist',0,-1)
  389. } 0
  390. } {1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100}
  391. test {Number conversion precision test (issue #1118)} {
  392. r eval {
  393. local value = 9007199254740991
  394. redis.call("set","foo",value)
  395. return redis.call("get","foo")
  396. } 0
  397. } {9007199254740991}
  398. test {String containing number precision test (regression of issue #1118)} {
  399. r eval {
  400. redis.call("set", "key", "12039611435714932082")
  401. return redis.call("get", "key")
  402. } 0
  403. } {12039611435714932082}
  404. test {Verify negative arg count is error instead of crash (issue #1842)} {
  405. catch { r eval { return "hello" } -12 } e
  406. set e
  407. } {ERR Number of keys can't be negative}
  408. test {Correct handling of reused argv (issue #1939)} {
  409. r eval {
  410. for i = 0, 10 do
  411. redis.call('SET', 'a', '1')
  412. redis.call('MGET', 'a', 'b', 'c')
  413. redis.call('EXPIRE', 'a', 0)
  414. redis.call('GET', 'a')
  415. redis.call('MGET', 'a', 'b', 'c')
  416. end
  417. } 0
  418. }
  419. test {Functions in the Redis namespace are able to report errors} {
  420. catch {
  421. r eval {
  422. redis.sha1hex()
  423. } 0
  424. } e
  425. set e
  426. } {*wrong number*}
  427. }
  428. # Start a new server since the last test in this stanza will kill the
  429. # instance at all.
  430. start_server {tags {"scripting"}} {
  431. test {Timedout read-only scripts can be killed by SCRIPT KILL} {
  432. set rd [redis_deferring_client]
  433. r config set lua-time-limit 10
  434. $rd eval {while true do end} 0
  435. after 200
  436. catch {r ping} e
  437. assert_match {BUSY*} $e
  438. r script kill
  439. after 200 ; # Give some time to Lua to call the hook again...
  440. assert_equal [r ping] "PONG"
  441. }
  442. test {Timedout script link is still usable after Lua returns} {
  443. r config set lua-time-limit 10
  444. r eval {for i=1,100000 do redis.call('ping') end return 'ok'} 0
  445. r ping
  446. } {PONG}
  447. test {Timedout scripts that modified data can't be killed by SCRIPT KILL} {
  448. set rd [redis_deferring_client]
  449. r config set lua-time-limit 10
  450. $rd eval {redis.call('set',KEYS[1],'y'); while true do end} 1 x
  451. after 200
  452. catch {r ping} e
  453. assert_match {BUSY*} $e
  454. catch {r script kill} e
  455. assert_match {UNKILLABLE*} $e
  456. catch {r ping} e
  457. assert_match {BUSY*} $e
  458. }
  459. # Note: keep this test at the end of this server stanza because it
  460. # kills the server.
  461. test {SHUTDOWN NOSAVE can kill a timedout script anyway} {
  462. # The server could be still unresponding to normal commands.
  463. catch {r ping} e
  464. assert_match {BUSY*} $e
  465. catch {r shutdown nosave}
  466. # Make sure the server was killed
  467. catch {set rd [redis_deferring_client]} e
  468. assert_match {*connection refused*} $e
  469. }
  470. }
  471. foreach cmdrepl {0 1} {
  472. start_server {tags {"scripting repl"}} {
  473. start_server {} {
  474. if {$cmdrepl == 1} {
  475. set rt "(commmands replication)"
  476. } else {
  477. set rt "(scripts replication)"
  478. r debug lua-always-replicate-commands 1
  479. }
  480. test "Before the replica connects we issue two EVAL commands $rt" {
  481. # One with an error, but still executing a command.
  482. # SHA is: 67164fc43fa971f76fd1aaeeaf60c1c178d25876
  483. catch {
  484. r eval {redis.call('incr',KEYS[1]); redis.call('nonexisting')} 1 x
  485. }
  486. # One command is correct:
  487. # SHA is: 6f5ade10a69975e903c6d07b10ea44c6382381a5
  488. r eval {return redis.call('incr',KEYS[1])} 1 x
  489. } {2}
  490. test "Connect a replica to the master instance $rt" {
  491. r -1 slaveof [srv 0 host] [srv 0 port]
  492. wait_for_condition 50 100 {
  493. [s -1 role] eq {slave} &&
  494. [string match {*master_link_status:up*} [r -1 info replication]]
  495. } else {
  496. fail "Can't turn the instance into a replica"
  497. }
  498. }
  499. test "Now use EVALSHA against the master, with both SHAs $rt" {
  500. # The server should replicate successful and unsuccessful
  501. # commands as EVAL instead of EVALSHA.
  502. catch {
  503. r evalsha 67164fc43fa971f76fd1aaeeaf60c1c178d25876 1 x
  504. }
  505. r evalsha 6f5ade10a69975e903c6d07b10ea44c6382381a5 1 x
  506. } {4}
  507. test "If EVALSHA was replicated as EVAL, 'x' should be '4' $rt" {
  508. wait_for_condition 50 100 {
  509. [r -1 get x] eq {4}
  510. } else {
  511. fail "Expected 4 in x, but value is '[r -1 get x]'"
  512. }
  513. }
  514. test "Replication of script multiple pushes to list with BLPOP $rt" {
  515. set rd [redis_deferring_client]
  516. $rd brpop a 0
  517. r eval {
  518. redis.call("lpush",KEYS[1],"1");
  519. redis.call("lpush",KEYS[1],"2");
  520. } 1 a
  521. set res [$rd read]
  522. $rd close
  523. wait_for_condition 50 100 {
  524. [r -1 lrange a 0 -1] eq [r lrange a 0 -1]
  525. } else {
  526. fail "Expected list 'a' in replica and master to be the same, but they are respectively '[r -1 lrange a 0 -1]' and '[r lrange a 0 -1]'"
  527. }
  528. set res
  529. } {a 1}
  530. test "EVALSHA replication when first call is readonly $rt" {
  531. r del x
  532. r eval {if tonumber(ARGV[1]) > 0 then redis.call('incr', KEYS[1]) end} 1 x 0
  533. r evalsha 6e0e2745aa546d0b50b801a20983b70710aef3ce 1 x 0
  534. r evalsha 6e0e2745aa546d0b50b801a20983b70710aef3ce 1 x 1
  535. wait_for_condition 50 100 {
  536. [r -1 get x] eq {1}
  537. } else {
  538. fail "Expected 1 in x, but value is '[r -1 get x]'"
  539. }
  540. }
  541. test "Lua scripts using SELECT are replicated correctly $rt" {
  542. r eval {
  543. redis.call("set","foo1","bar1")
  544. redis.call("select","10")
  545. redis.call("incr","x")
  546. redis.call("select","11")
  547. redis.call("incr","z")
  548. } 0
  549. r eval {
  550. redis.call("set","foo1","bar1")
  551. redis.call("select","10")
  552. redis.call("incr","x")
  553. redis.call("select","11")
  554. redis.call("incr","z")
  555. } 0
  556. wait_for_condition 50 100 {
  557. [r -1 debug digest] eq [r debug digest]
  558. } else {
  559. fail "Master-Replica desync after Lua script using SELECT."
  560. }
  561. }
  562. }
  563. }
  564. }
  565. start_server {tags {"scripting repl"}} {
  566. start_server {overrides {appendonly yes aof-use-rdb-preamble no}} {
  567. test "Connect a replica to the master instance" {
  568. r -1 slaveof [srv 0 host] [srv 0 port]
  569. wait_for_condition 50 100 {
  570. [s -1 role] eq {slave} &&
  571. [string match {*master_link_status:up*} [r -1 info replication]]
  572. } else {
  573. fail "Can't turn the instance into a replica"
  574. }
  575. }
  576. test "Redis.replicate_commands() must be issued before any write" {
  577. r eval {
  578. redis.call('set','foo','bar');
  579. return redis.replicate_commands();
  580. } 0
  581. } {}
  582. test "Redis.replicate_commands() must be issued before any write (2)" {
  583. r eval {
  584. return redis.replicate_commands();
  585. } 0
  586. } {1}
  587. test "Redis.set_repl() must be issued after replicate_commands()" {
  588. r debug lua-always-replicate-commands 0
  589. catch {
  590. r eval {
  591. redis.set_repl(redis.REPL_ALL);
  592. } 0
  593. } e
  594. r debug lua-always-replicate-commands 1
  595. set e
  596. } {*only after turning on*}
  597. test "Redis.set_repl() don't accept invalid values" {
  598. catch {
  599. r eval {
  600. redis.replicate_commands();
  601. redis.set_repl(12345);
  602. } 0
  603. } e
  604. set e
  605. } {*Invalid*flags*}
  606. test "Test selective replication of certain Redis commands from Lua" {
  607. r del a b c d
  608. r eval {
  609. redis.replicate_commands();
  610. redis.call('set','a','1');
  611. redis.set_repl(redis.REPL_NONE);
  612. redis.call('set','b','2');
  613. redis.set_repl(redis.REPL_AOF);
  614. redis.call('set','c','3');
  615. redis.set_repl(redis.REPL_ALL);
  616. redis.call('set','d','4');
  617. } 0
  618. wait_for_condition 50 100 {
  619. [r -1 mget a b c d] eq {1 {} {} 4}
  620. } else {
  621. fail "Only a and c should be replicated to replica"
  622. }
  623. # Master should have everything right now
  624. assert {[r mget a b c d] eq {1 2 3 4}}
  625. # After an AOF reload only a, c and d should exist
  626. r debug loadaof
  627. assert {[r mget a b c d] eq {1 {} 3 4}}
  628. }
  629. test "PRNG is seeded randomly for command replication" {
  630. set a [
  631. r eval {
  632. redis.replicate_commands();
  633. return math.random()*100000;
  634. } 0
  635. ]
  636. set b [
  637. r eval {
  638. redis.replicate_commands();
  639. return math.random()*100000;
  640. } 0
  641. ]
  642. assert {$a ne $b}
  643. }
  644. test "Using side effects is not a problem with command replication" {
  645. r eval {
  646. redis.replicate_commands();
  647. redis.call('set','time',redis.call('time')[1])
  648. } 0
  649. assert {[r get time] ne {}}
  650. wait_for_condition 50 100 {
  651. [r get time] eq [r -1 get time]
  652. } else {
  653. fail "Time key does not match between master and replica"
  654. }
  655. }
  656. }
  657. }