def connectToMaster(self):
logging.info("Connecting to master at %s:%s",
self.clusterMaster,
MultiMachineWorkerRunner.TestControlPort)
while True:
socketError = None
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
try:
sock.connect((self.clusterMaster, MultiMachineWorkerRunner.TestControlPort))
common.writeString(sock, MultiMachineTestRunner.ProtoclVersion)
common.writeString(sock, self.testId)
response = common.readString(sock)
if response != 'ok:start_worker':
logging.error("Error response from cluster master: %s", response)
sock.close()
sock = None
logging.info("Connected to cluster master. Response: %s", response)
return sock
except common.SocketException:
socketError = traceback.format_exc()
except socket.error:
socketError = traceback.format_exc()
if socketError:
sock.close()
logging.info("Waiting for master. %s", socketError)
time.sleep(0.5)
def waitForStopCommand(sock):
message = common.readString(sock)
if message != 'ok:stop_worker':
logging.error("Expected stop command from server. Received: %s",
message)
else:
logging.info("Received 'stop' command from master")
def connectToMaster(self):
logging.info("Connecting to master at %s:%s", self.clusterMaster,
MultiMachineWorkerRunner.TestControlPort)
while True:
socketError = None
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
try:
sock.connect((self.clusterMaster,
MultiMachineWorkerRunner.TestControlPort))
common.writeString(sock, MultiMachineTestRunner.ProtoclVersion)
common.writeString(sock, self.testId)
response = common.readString(sock)
if response != 'ok:start_worker':
logging.error("Error response from cluster master: %s",
response)
sock.close()
sock = None
logging.info("Connected to cluster master. Response: %s",
response)
return sock
except common.SocketException:
socketError = traceback.format_exc()
except socket.error:
socketError = traceback.format_exc()
if socketError:
sock.close()
logging.info("Waiting for master. %s", socketError)
time.sleep(0.5)
def verifyTestId(self, workerSocket):
workerTestId = common.readString(workerSocket)
return workerTestId == self.testId
def protocolHandshake(self, workerSocket):
workerVersion = common.readString(workerSocket)
return workerVersion == MultiMachineTestRunner.ProtoclVersion
def waitForStopCommand(sock):
message = common.readString(sock)
if message != 'ok:stop_worker':
logging.error("Expected stop command from server. Received: %s", message)
else:
logging.info("Received 'stop' command from master")
def verifyTestId(self, workerSocket):
workerTestId = common.readString(workerSocket)
return workerTestId == self.testId
def protocolHandshake(self, workerSocket):
workerVersion = common.readString(workerSocket)
return workerVersion == MultiMachineTestRunner.ProtoclVersion
