X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=examples%2Fconfig.py;h=bacc840a04e56f7d040b5a909e80ea1c0acb97c7;hb=dbcac620b62e16f9e4b7a763e1400f2f3a79b285;hp=579308b55b929f601b868b54415684184c0a6936;hpb=b64327bb58139f1e1f770b16492968423f011eab;p=osm%2FN2VC.git

diff --git a/examples/config.py b/examples/config.py
index 579308b..bacc840 100644
--- a/examples/config.py
+++ b/examples/config.py
@@ -10,12 +10,14 @@ import asyncio
 import logging
 
 from juju.model import Model
+from juju import loop
 
 log = logging.getLogger(__name__)
 
-MB = 1024 * 1024
+MB = 1
 
-async def run():
+
+async def main():
     model = Model()
     await model.connect_current()
     await model.reset(force=True)
@@ -44,12 +46,10 @@ async def run():
     assert(constraints['mem'] == 512 * MB)
 
     await model.disconnect()
-    model.loop.stop()
-
-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()
+
+    
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.DEBUG)
+    ws_logger = logging.getLogger('websockets.protocol')
+    ws_logger.setLevel(logging.INFO)
+    loop.run(main())