def cli(nodeRegsForCLI, cliLooper, tdir):
mockOutput = MockOutput()
Cli = TestCli(looper=cliLooper,
basedirpath=tdir,
nodeReg=nodeRegsForCLI.nodeReg,
cliNodeReg=nodeRegsForCLI.cliNodeReg,
output=mockOutput,
debug=True)
Cli.NodeClass = TestNode
Cli.ClientClass = TestClient
Cli.basedirpath = tdir
return Cli
def newCLI(looper, tdir, cliClass=TestCli,
nodeClass=TestNode,
clientClass=TestClient,
config=None):
mockOutput = MockOutput()
newcli = cliClass(looper=looper,
basedirpath=tdir,
nodeReg=None,
cliNodeReg=None,
output=mockOutput,
debug=True,
config=config)
newcli.NodeClass = nodeClass
newcli.ClientClass = clientClass
newcli.basedirpath = tdir
return newcli
def newCLI(looper,
basedir,
ledger_base_dir,
cliClass=TestCli,
nodeClass=TestNode,
clientClass=TestClient,
config=None,
partition: str = None,
unique_name=None,
logFileName=None,
name=None,
agentCreator=None,
nodes_chroot: str = None):
if partition:
recorder = Recorder(partition)
else:
recorder = CombinedRecorder()
mockOutput = MockOutput(recorder=recorder)
recorder.write("~ be {}\n".format(unique_name))
otags = config.log_override_tags['cli'] if config else None
cliClassParams = {
'looper': looper,
'basedirpath': basedir,
'ledger_base_dir': ledger_base_dir,
'nodeReg': None,
'cliNodeReg': None,
'output': mockOutput,
'debug': True,
'config': config,
'unique_name': unique_name,
'override_tags': otags,
'logFileName': logFileName,
'nodes_chroot': nodes_chroot
}
if name is not None and agentCreator is not None:
cliClassParams['name'] = name
cliClassParams['agentCreator'] = agentCreator
newcli = cliClass(**cliClassParams)
newcli.recorder = recorder
newcli.NodeClass = nodeClass
newcli.ClientClass = clientClass
newcli.basedirpath = basedir
return newcli