X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=examples%2Frelate.py;h=a8e39a050656e58719dfebe23f4e1ed2a8fada79;hb=096fdda8130b0b0262194643745d694b6156a2e8;hp=967a785909b88e42cb49067a33d09342c962405c;hpb=a56869a418a68b795817d4f14d4065d11784f09f;p=osm%2FN2VC.git

diff --git a/examples/relate.py b/examples/relate.py
index 967a785..a8e39a0 100644
--- a/examples/relate.py
+++ b/examples/relate.py
@@ -9,11 +9,11 @@ from juju.model import Model, ModelObserver
 
 
 class MyModelObserver(ModelObserver):
-    def on_change(self, delta, old, new, model):
+    async def on_change(self, delta, old, new, model):
         if model.all_units_idle():
             logging.debug('All units idle, disconnecting')
-            task = model.loop.create_task(model.disconnect())
-            task.add_done_callback(lambda fut: model.loop.stop())
+            await model.disconnect()
+            model.loop.stop()
 
 
 async def run():
@@ -21,9 +21,6 @@ async def run():
     await model.connect_current()
 
     await model.reset(force=True)
-    await model.block_until(
-        lambda: len(model.machines) == 0
-    )
     model.add_observer(MyModelObserver())
 
     await model.deploy(
@@ -44,10 +41,10 @@ async def run():
         'nrpe',
     )
 
-
 logging.basicConfig(level=logging.DEBUG)
 ws_logger = logging.getLogger('websockets.protocol')
 ws_logger.setLevel(logging.INFO)
 loop = asyncio.get_event_loop()
+loop.set_debug(False)
 loop.create_task(run())
 loop.run_forever()