Added test for out of order receive and fix bug creating task

[osm/N2VC.git] / juju / client / connection.py

diff --git a/juju/client/connection.py b/juju/client/connection.py
index 625c609..82ec4fa 100644 (file)
--- a/juju/client/connection.py
+++ b/juju/client/connection.py
@@ -76,6 +76,7 @@ class Connection:
         kw['loop'] = self.loop
         self.addr = url
         self.ws = await websockets.connect(url, **kw)
+        self.loop.create_task(self.receiver())
         log.info("Driver connected to juju %s", url)
         return self
 
@@ -216,7 +217,6 @@ class Connection:
         client = cls(endpoint, uuid, username, password, cacert, macaroons,
                      loop)
         await client.open()
-        client.loop.create_task(client.receiver)
 
         redirect_info = await client.redirect_info()
         if not redirect_info: