forked from mirror/ledisdb
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
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()
|
||
|
|
assert self.l.get("a") == "tx"
|
||
|
|
|
||
|
|
@unittest.skipIf(check, reason="db not support transaction")
|
||
|
|
def test_rollback(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 tx.get("a") == "tx"
|
||
|
|
assert self.l.get("a") == "no-tx"
|
||
|
|
|
||
|
|
tx.rollback()
|
||
|
|
assert self.l.get("a") == "no-tx"
|