projects
/
osm
/
N2VC.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Improve error handling
[osm/N2VC.git]
/
tests
/
unit
/
test_connection.py
diff --git
a/tests/unit/test_connection.py
b/tests/unit/test_connection.py
index
354b363
..
5371fdb
100644
(file)
--- a/
tests/unit/test_connection.py
+++ b/
tests/unit/test_connection.py
@@
-3,6
+3,8
@@
import mock
import pytest
from collections import deque
import pytest
from collections import deque
+from websockets.exceptions import ConnectionClosed
+
from .. import base
from juju.client.connection import Connection
from .. import base
from juju.client.connection import Connection
@@
-17,6
+19,8
@@
class WebsocketMock:
pass
async def recv(self):
pass
async def recv(self):
+ if not self.responses:
+ raise ConnectionClosed(0, 'no reason')
return json.dumps(self.responses.popleft())
async def close(self):
return json.dumps(self.responses.popleft())
async def close(self):