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