X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=tests%2Funit%2Ftest_connection.py;h=5371fdb9fbc0c52bc46604143557c7b80d4cd14c;hb=a9180cca8165ab6396971a30e3063845a428dd5e;hp=354b363673b8d86385a678d0e1f61b9f9f38de07;hpb=afd9bf49afc5025f3b668fd3a09b0275d7edc382;p=osm%2FN2VC.git

diff --git a/tests/unit/test_connection.py b/tests/unit/test_connection.py
index 354b363..5371fdb 100644
--- a/tests/unit/test_connection.py
+++ b/tests/unit/test_connection.py
@@ -3,6 +3,8 @@ import mock
 import pytest
 from collections import deque
 
+from websockets.exceptions import ConnectionClosed
+
 from .. import base
 from juju.client.connection import Connection
 
@@ -17,6 +19,8 @@ class WebsocketMock:
         pass
 
     async def recv(self):
+        if not self.responses:
+            raise ConnectionClosed(0, 'no reason')
         return json.dumps(self.responses.popleft())
 
     async def close(self):