def ensureConnectedToTestEnv(cli):
    if not cli.activeEnv:
        cli.enterCmd("connect test")
        timeout = waits.expectedClientConnectionTimeout(
            util.getMaxFailures(len(cli.nodeReg)))
        cli.looper.run(
            eventually(checkConnectedToEnv, cli, retryWait=1, timeout=timeout))
Exemple #2
0
def ensureConnectedToTestEnv(cli):
    if not cli.activeEnv:
        cli.enterCmd("connect test")
        timeout = waits.expectedClientConnectionTimeout(
            util.getMaxFailures(len(cli.nodeReg))
        )
        cli.looper.run(
            eventually(checkConnectedToEnv, cli, retryWait=1,
                       timeout=timeout))
def newNodeAdded(be, do, poolNodesStarted, philCli, newStewardCli,
                 connectedToTest):
    be(philCli)

    if not philCli._isConnectedToAnyEnv():
        do('connect test', within=3, expect=connectedToTest)

    be(newStewardCli)
    doNodeCmd(do)
    newNodeData = vals["newNodeData"]

    def checkClientConnected(client):
        name = newNodeData[ALIAS] + CLIENT_STACK_SUFFIX
        assert name in client.nodeReg

    def checkNodeConnected(nodes):
        for node in nodes:
            name = newNodeData[ALIAS]
            assert name in node.nodeReg

    timeout = waits.expectedClientConnectionTimeout(
        util.getMaxFailures(len(philCli.nodeReg)))
    newStewardCli.looper.run(
        eventually(checkClientConnected,
                   newStewardCli.activeClient,
                   timeout=timeout))
    timeout = waits.expectedClientConnectionTimeout(
        util.getMaxFailures(len(philCli.nodeReg)))
    philCli.looper.run(
        eventually(checkClientConnected, philCli.activeClient,
                   timeout=timeout))

    timeout = waits.expectedClientConnectionTimeout(
        util.getMaxFailures(len(philCli.nodeReg)))
    poolNodesStarted.looper.run(
        eventually(checkNodeConnected,
                   list(poolNodesStarted.nodes.values()),
                   timeout=timeout))
    return vals