def on_connection_opened(self, event): if event.connection == self.conn_route: self.conn_normal = event.container.connect(self.normal_addr) elif event.connection == self.conn_normal: self.sender = event.container.create_sender(self.conn_normal, self.dest) def on_connection_closed(self, event): self.conn_route.close() self.timer.cancel() def on_link_opened(self, event): if event.receiver: self.receiver = event.receiver self.receiver.flow(1) def on_sendable(self, event): msg = Message(body="CloseWithUnsettled") event.sender.send(msg) def on_message(self, event): self.conn_normal.close() def run(self): Container(self).run() if __name__ == '__main__': unittest.main(main_module())
self.assertRaises( BadRequestStatus, self.node.create, dict(attrs, type=CONNECTOR, name="bad2", port=str(self.get_port()))) conf = Qdrouterd.Config([ ('router', { 'mode': 'standalone', 'id': 'all_by_myself1'}), ('listener', {'port':self.get_port(), 'role':'inter-router'}) ]) r = self.qdrouterd('routerX', conf, wait=False) r.expect = Process.EXIT_FAIL self.assertTrue(r.wait() != 0) conf = Qdrouterd.Config([ ('router', { 'mode': 'standalone', 'id': 'all_by_myself2'}), ('listener', {'port':self.get_port(), 'role':'normal'}), ('connector', {'port':self.get_port(), 'role':'inter-router'}) ]) r = self.qdrouterd('routerY', conf, wait=False) r.expect = Process.EXIT_FAIL self.assertTrue(r.wait() != 0) def test_get_schema(self): schema = dictify(QdSchema().dump()) got = self.node.call(self.node.request(operation="GET-JSON-SCHEMA", identity="self")).body self.assertEquals(schema, dictify(json.loads(got))) got = self.node.call(self.node.request(operation="GET-SCHEMA", identity="self")).body self.assertEquals(schema, got) if __name__ == '__main__': unittest.main(system_test.main_module())
def record_trace ( self, observed_trace ): for trace_record in self.expected_trace_counts : trace = trace_record [ 0 ] if observed_trace == trace : trace_record [ 1 ] += 1 return # If we get here, the trace is one we were not expecting. That's bad. self.bail ( "Unexpected trace: %s" % observed_trace ) # Shut down everything and make sure that all of the extected traces # have been seen. def finish_test ( self ) : self.test_timer.cancel() self.send_timer.cancel() for trace_record in self.expected_trace_counts : count = trace_record[1] # Deliberate flaw for debugging. # count = 0 if count <= 0 : self.bail ( "Trace %s was not seen." % trace_record[0] ) return # success self.bail ( None ) if __name__ == '__main__': unittest.main(main_module())
def __init__(self, conn_address, address): super(SimpleSndRecv, self).__init__() self.conn_address = conn_address self.address = address self.sender = None self.receiver = None self.conn = None self.message_received = False def on_start(self, event): self.conn = event.container.connect(self.conn_address) self.receiver = event.container.create_receiver( self.conn, self.address) self.sender = event.container.create_sender(self.conn, self.address) def on_sendable(self, event): msg = Message(body="Hello World") event.sender.send(msg) def on_message(self, event): if "Hello World" == event.message.body: self.message_received = True self.conn.close() def run(self): Container(self).run() if __name__ == '__main__': unittest.main(system_test.main_module())