def setUp(self): super().setUp() self.hosts = ['{}:{}'.format(self.server.host, self.server.port)] self.client = self.loop.run_until_complete( connect(self.hosts, loop=self.loop)) if not self.topic: topic = "%s-%s" % (self.id()[self.id().rindex(".") + 1:], random_string(10).decode('utf-8')) self.topic = topic.encode('utf-8') self.loop.run_until_complete( self.client.ensure_topic_exists(self.topic)) self._messages = {}
def test_connect_with_global_loop(self): asyncio.set_event_loop(self.loop) client = yield from connect(self.hosts) self.assertIs(client._loop, self.loop) client.close()