| 137 | driver.acknowledgeStatusUpdate(update) |
| 138 | |
| 139 | def frameworkMessage(self, driver, executorId, slaveId, message): |
| 140 | self.messagesReceived += 1 |
| 141 | |
| 142 | # The message bounced back as expected. |
| 143 | if message != "data with a \0 byte": |
| 144 | print "The returned message data did not match!" |
| 145 | print " Expected: 'data with a \\x00 byte'" |
| 146 | print " Actual: ", repr(str(message)) |
| 147 | sys.exit(1) |
| 148 | print "Received message:", repr(str(message)) |
| 149 | |
| 150 | if self.messagesReceived == TOTAL_TASKS: |
| 151 | if self.messagesReceived != self.messagesSent: |
| 152 | print "Sent", self.messagesSent, |
| 153 | print "but received", self.messagesReceived |
| 154 | sys.exit(1) |
| 155 | print "All tasks done, and all messages received, exiting" |
| 156 | driver.stop() |
| 157 | |
| 158 | if __name__ == "__main__": |
| 159 | if len(sys.argv) != 2: |