| 321 | @pytest.fixture() |
| 322 | def ui_websocket(site, user): |
| 323 | class WsMock: |
| 324 | def __init__(self): |
| 325 | self.result = gevent.event.AsyncResult() |
| 326 | |
| 327 | def send(self, data): |
| 328 | self.result.set(json.loads(data)["result"]) |
| 329 | |
| 330 | def getResult(self): |
| 331 | back = self.result.get() |
| 332 | self.result = gevent.event.AsyncResult() |
| 333 | return back |
| 334 | |
| 335 | ws_mock = WsMock() |
| 336 | ui_websocket = UiWebsocket(ws_mock, site, None, user, None) |