def createAcme(name=None, wallet=None, basedirpath=None, port=None):
return createAgent(AcmeAgent,
name or "Acme Corp",
wallet or buildAcmeWallet(),
basedirpath,
port,
clientClass=TestClient)
def createFaber(name=None, wallet=None, basedirpath=None, port=None):
return createAgent(FaberAgent,
name or "Faber College",
wallet or buildFaberWallet(),
basedirpath,
port,
clientClass=TestClient)
def createThrift(name=None, wallet=None, basedirpath=None, port=None):
return createAgent(ThriftAgent,
name or "Thrift Bank",
wallet or buildThriftWallet(),
basedirpath,
port,
clientClass=TestClient)
def createBulldog(name=None, wallet=None, basedirpath=None, port=None):
return createAgent(BulldogAgent,
name or "Bulldog",
wallet or buildBulldogWallet(),
basedirpath,
port,
clientClass=TestClient)
def testAgentCreatesWalletIfItDoesntHaveOne(tdirWithPoolTxns):
agent = createAgent(AcmeAgent,
"Acme Corp",
wallet=None,
basedirpath=tdirWithPoolTxns,
port=genHa()[1],
clientClass=TestClient)
assert agent._wallet is not None
def startAgent(name, seed, loop=None):
agentWallet = Wallet(name)
agentWallet.addIdentifier(signer=SimpleSigner(seed=bytes(seed, 'utf-8')))
agent = createAgent(WalletedAgent, name, wallet=agentWallet, loop=loop)
agentPort = agent.endpoint.stackParams['ha'].port
with Looper(debug=True) as looper:
looper.add(agent)
log.debug("Running {} now (port: {})".format(name, agentPort))
return agent
def createAndRunAgent(name: str,
agentClass=None,
wallet=None,
basedirpath=None,
port=None,
looper=None,
clientClass=Client,
bootstrap=True,
cliAgentCreator=None):
loop = looper.loop if looper else None
_, with_cli = TestWalletedAgent.getPassedArgs()
try:
if cliAgentCreator and with_cli:
runAgentCli(name=name, agentCreator=cliAgentCreator)
else:
assert agentClass
agent = createAgent(agentClass, name, wallet, basedirpath,
port, loop, clientClass)
runAgent(agent, looper, bootstrap)
except Exception as exc:
error = "Agent startup failed: [cause : {}]".format(str(exc))
logger.error(getFormattedErrorMsg(error))
def testAgentCreatesWalletIfItDoesntHaveOne(tdirWithPoolTxns):
agent = createAgent(AcmeAgent, "Acme Corp",
wallet=None, basedirpath=tdirWithPoolTxns,
port=genHa()[1], clientClass=TestClient)
assert agent._wallet is not None
