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.
32 lines
999 B
32 lines
999 B
1 year ago
|
import pybbt
|
||
|
|
||
|
from helpers.commands.checker import Checker
|
||
|
from helpers.redis import Redis
|
||
|
|
||
|
|
||
|
class StringChecker(Checker):
|
||
|
PREFIX = "string"
|
||
|
|
||
|
def __init__(self):
|
||
|
self.cnt = 0
|
||
|
|
||
|
def add_data(self, r: Redis, cross_slots_cmd: bool):
|
||
|
p = r.pipeline()
|
||
|
p.set(f"{self.PREFIX}_{self.cnt}_str", "string")
|
||
|
p.set(f"{self.PREFIX}_{self.cnt}_int", 0)
|
||
|
p.set(f"{self.PREFIX}_{self.cnt}_int0", -1)
|
||
|
p.set(f"{self.PREFIX}_{self.cnt}_int1", 123456789)
|
||
|
ret = p.execute()
|
||
|
pybbt.ASSERT_EQ(ret, [True, True, True, True])
|
||
|
self.cnt += 1
|
||
|
|
||
|
def check_data(self, r: Redis, cross_slots_cmd: bool):
|
||
|
for i in range(self.cnt):
|
||
|
p = r.pipeline()
|
||
|
p.get(f"{self.PREFIX}_{i}_str")
|
||
|
p.get(f"{self.PREFIX}_{i}_int")
|
||
|
p.get(f"{self.PREFIX}_{i}_int0")
|
||
|
p.get(f"{self.PREFIX}_{i}_int1")
|
||
|
ret = p.execute()
|
||
|
pybbt.ASSERT_EQ(ret, [b"string", b"0", b"-1", b"123456789"])
|