def test_application(self): manager = NetWorkManager() manager.start() @gen.coroutine def version_callback(msg_id, msg, param): self.is_callback = True self.assertEqual(msg_id, 0) self.assertEqual(msg, {'params': {}, 'method': 'server.version'}) manager.add_message(Version({}), version_callback) manager.quit()
class HowToUseChain(AsyncTestCase): def setUp(self): super(HowToUseChain, self).setUp() self.manager = NetWorkManager() self.manager.start() def tearDown(self): self.manager.quit() self.wait_quit() super(HowToUseChain, self).tearDown() @gen_test() def wait_quit(self): yield gen.sleep(MAX_WAIT_SECONDS_BEFORE_SHUTDOWN + 0.01) def test_application(self): bc = BlockChain() future = Future() future.set_result(open('../files/testnet_headers').read()) bc.init_header_callback(future)