Skip to content

Commit 75287ee

Browse files
committed
Add environment config to disable RDB checksum
1 parent 1abac3f commit 75287ee

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

RLTest/env.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@ class Defaults:
128128
randomize_ports = False
129129
oss_password = None
130130
cluster_node_timeout = None
131+
disable_rdb_checksum = False
131132

132133
def getKwargs(self):
133134
kwargs = {
@@ -166,7 +167,7 @@ def __init__(self, testName=None, testDescription=None, module=None,
166167
moduleArgs=None, env=None, useSlaves=None, shardsCount=None, decodeResponses=None,
167168
useAof=None, useRdbPreamble=None, forceTcp=False, useTLS=False, tlsCertFile=None, tlsKeyFile=None,
168169
tlsCaCertFile=None, logDir=None, redisBinaryPath=None, dmcBinaryPath=None,
169-
redisEnterpriseBinaryPath=None, noDefaultModuleArgs=False, clusterNodeTimeout = None):
170+
redisEnterpriseBinaryPath=None, noDefaultModuleArgs=False, clusterNodeTimeout=None, disableRdbChecksum=False):
170171

171172
self.testName = testName if testName else '%s.%s' % (inspect.getmodule(inspect.currentframe().f_back).__name__, inspect.currentframe().f_back.f_code.co_name)
172173
self.testName = self.testName.replace(' ', '_')
@@ -185,6 +186,7 @@ def __init__(self, testName=None, testDescription=None, module=None,
185186
self.decodeResponses = decodeResponses if decodeResponses else Defaults.decode_responses
186187
self.useAof = useAof if useAof else Defaults.use_aof
187188
self.useRdbPreamble = useRdbPreamble if useRdbPreamble is not None else Defaults.use_rdb_preamble
189+
self.disableRdbChecksum = disableRdbChecksum if disableRdbChecksum else Defaults.disable_rdb_checksum
188190
self.verbose = Defaults.verbose
189191
self.logDir = logDir if logDir else Defaults.logdir
190192
self.forceTcp = forceTcp

RLTest/redis_std.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ class StandardEnv(object):
2121
def __init__(self, redisBinaryPath, port=6379, modulePath=None, moduleArgs=None, outputFilesFormat=None,
2222
dbDirPath=None, useSlaves=False, serverId=1, password=None, libPath=None, clusterEnabled=False, decodeResponses=False,
2323
useAof=False, useRdbPreamble=True, debugger=None, noCatch=False, unix=False, verbose=False, useTLS=False, tlsCertFile=None,
24-
tlsKeyFile=None, tlsCaCertFile=None, clusterNodeTimeout = None):
24+
tlsKeyFile=None, tlsCaCertFile=None, clusterNodeTimeout=None, disableRdbChecksum=False):
2525
self.uuid = uuid.uuid4().hex
2626
self.redisBinaryPath = os.path.expanduser(redisBinaryPath) if redisBinaryPath.startswith(
2727
'~/') else redisBinaryPath
@@ -53,6 +53,7 @@ def __init__(self, redisBinaryPath, port=6379, modulePath=None, moduleArgs=None,
5353
self.tlsKeyFile = tlsKeyFile
5454
self.tlsCaCertFile = tlsCaCertFile
5555
self.clusterNodeTimeout = clusterNodeTimeout
56+
self.disableRdbChecksum = disableRdbChecksum
5657

5758
if port > 0:
5859
self.port = port
@@ -197,6 +198,8 @@ def createCmdArgs(self, role):
197198
cmdArgs += ['--tls-cert-file', self.getTLSCertFile()]
198199
cmdArgs += ['--tls-key-file', self.getTLSKeyFile()]
199200
cmdArgs += ['--tls-ca-cert-file', self.getTLSCACertFile()]
201+
if self.disableRdbChecksum:
202+
cmdArgs += ['--rdbchecksum', 'no']
200203

201204
return cmdArgs
202205

0 commit comments

Comments
 (0)