redis-shake工具
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

68 lines
1.4 KiB

import pybbt as p
import helpers as h
def test(src, dst):
cross_slots_cmd = not (src.is_cluster() or dst.is_cluster())
inserter = h.DataInserter()
inserter.add_data(src, cross_slots_cmd=cross_slots_cmd)
p.ASSERT_TRUE(src.do("save"))
inserter.add_data(src, cross_slots_cmd=cross_slots_cmd) # add data again
opts = h.ShakeOpts.create_scan_opts(src, dst)
p.log(f"opts: {opts}")
# run shake
h.Shake.run_once(opts)
# check data
inserter.check_data(src, cross_slots_cmd=cross_slots_cmd)
inserter.check_data(dst, cross_slots_cmd=cross_slots_cmd)
p.ASSERT_EQ(src.dbsize(), dst.dbsize())
@p.subcase()
def standalone_to_standalone():
src = h.Redis()
dst = h.Redis()
test(src, dst)
@p.subcase()
def standalone_to_cluster():
if h.REDIS_SERVER_VERSION < 3.0:
return
src = h.Redis()
dst = h.Cluster()
test(src, dst)
@p.subcase()
def cluster_to_standalone():
if h.REDIS_SERVER_VERSION < 3.0:
return
src = h.Cluster()
dst = h.Redis()
test(src, dst)
@p.subcase()
def cluster_to_cluster():
if h.REDIS_SERVER_VERSION < 3.0:
return
src = h.Cluster()
dst = h.Cluster()
test(src, dst)
@p.case(tags=["scan"])
def main():
standalone_to_standalone()
standalone_to_cluster()
cluster_to_standalone()
cluster_to_cluster()
if __name__ == '__main__':
main()