X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=examples%2Fadd_machine.py;h=8ae2d4029e6265a2e0bd4ad31d7ebd21bc9e3e77;hb=refs%2Ftags%2F0.4.0;hp=34f7869aa95c8b06a563bc1b7306560577627ae0;hpb=5fef7503b13f145adc7cd4ee31c2d684e09a6a85;p=osm%2FN2VC.git

diff --git a/examples/add_machine.py b/examples/add_machine.py
index 34f7869..8ae2d40 100755
--- a/examples/add_machine.py
+++ b/examples/add_machine.py
@@ -13,8 +13,8 @@ import logging
 from juju import loop
 from juju.model import Model
 
-MB = 1024 * 1024
-GB = MB * 1024
+MB = 1
+GB = 1024
 
 
 async def main():
@@ -54,15 +54,17 @@ async def main():
                         for unit in application.units))
 
         await application.remove()
-        await machine3.destroy()
-        await machine2.destroy()
-        await machine1.destroy()
+
+        await machine3.destroy(force=True)
+        await machine2.destroy(force=True)
+        await machine1.destroy(force=True)
     finally:
         await model.disconnect()
 
 
-logging.basicConfig(level=logging.DEBUG)
-ws_logger = logging.getLogger('websockets.protocol')
-ws_logger.setLevel(logging.INFO)
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.DEBUG)
+    ws_logger = logging.getLogger('websockets.protocol')
+    ws_logger.setLevel(logging.INFO)
 
-loop.run(main())
+    loop.run(main())