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