def test_raw_websocket(self, ws): ws.return_value.process.return_value = asyncio.Future(loop=self.loop) ws.return_value.process.return_value.set_result(web.HTTPOk()) route = self.make_route() request = self.make_request( 'GET', '/sm/', headers=CIMultiDict({})) res = self.loop.run_until_complete(route.websocket(request)) self.assertIsInstance(res, web.HTTPOk) self.assertTrue(ws.called) self.assertTrue(ws.return_value.process.called)
def test_raw_websocket_fail(self): route = self.make_route() request = self.make_request('GET', '/sm/') res = self.loop.run_until_complete(route.websocket(request)) self.assertIsInstance(res, web.HTTPNotFound)