def main(): if len(sys.argv) < 6: usage() sys.exit() convert = False if len(sys.argv) >= 6: (redis_host, redis_port, redis_db, ledis_host, ledis_port) = sys.argv[1:6] if int(redis_db) >= 16: print redis_db sys.exit("LedisDB only support 16 databases([0-15]") choice = raw_input("[y/N]").lower() if not get_prompt(choice): sys.exit("No proceed") redis_c = redis.Redis(host=redis_host, port=int(redis_port), db=int(redis_db)) ledis_c = ledis.Ledis(host=ledis_host, port=int(ledis_port), db=int(redis_db)) try: redis_c.ping() except redis.ConnectionError: print "Could not connect to Redis Server" sys.exit() try: ledis_c.ping() except redis.ConnectionError: print "Could not connect to LedisDB Server" sys.exit() copy(redis_c, ledis_c, convert=convert) print "Done\n"
def setUp(self): self.l = ledis.Ledis(port=6380)
# coding: utf-8 # Test Cases for k/v commands import unittest import sys import datetime, time sys.path.append('..') import ledis from ledis._compat import b, iteritems l = ledis.Ledis(port=6380) def current_time(): return datetime.datetime.now() class TestCmdKv(unittest.TestCase): def setUp(self): pass def tearDown(self): l.delete('a', 'b', 'c', 'non_exist_key') def test_decr(self): assert l.delete('a') == 1 assert l.decr('a') == -1 assert l['a'] == b('-1') assert l.decr('a') == -2 assert l['a'] == b('-2')
import unittest import sys sys.path.append("..") import ledis global_l = ledis.Ledis() #db that do not support transaction dbs = ["leveldb", "rocksdb", "hyperleveldb", "goleveldb"] check = global_l.info().get("db_name") in dbs class TestTx(unittest.TestCase): def setUp(self): self.l = ledis.Ledis(port=6380) def tearDown(self): self.l.flushdb() @unittest.skipIf(check, reason="db not support transaction") def test_commit(self): tx = self.l.tx() self.l.set("a", "no-tx") assert self.l.get("a") == "no-tx" tx.begin() tx.set("a", "tx") assert self.l.get("a") == "no-tx" assert tx.get("a") == "tx" tx.commit()