Revisit unitrun example

Author: cderici

examples/unitrun.py

@@ -15,10 +15,20 @@
 
 async def run_command(unit):
     logging.debug('Running command on unit %s', unit.name)
-
-    # unit.run() returns a juju.action.Action instance
+    # unit.run() returns a juju.action.Action instance,
+    # it needs to be wait()'ed to get the results
     action = await unit.run('unit-get public-address')
-    logging.debug("Action results: %s", action.results)
+    await action.wait()
+
+    out1 = f"Action status: {action.status}"
+    logging.debug(out1)
+    print(out1)
+    # 'completed'
+
+    out2 = f"Action results: {action.results}"
+    logging.debug(out2)
+    print(out2)
+    # {'return-code': 0, 'stdout': '<IP Addr>\n'}
 
 
 async def main():
@@ -36,11 +46,21 @@ async def main():
     for unit in app.units:
         await run_command(unit)
 
+    await app.remove()
+
     await model.disconnect()
 
 
 if __name__ == '__main__':
-    logging.basicConfig(level=logging.DEBUG)
-    ws_logger = logging.getLogger('websockets.protocol')
-    ws_logger.setLevel(logging.INFO)
+    # Uncomment below to get logging output
+
+    # logging.basicConfig(level=logging.DEBUG)
+    # ws_logger = logging.getLogger('websockets.protocol')
+    # ws_logger.setLevel(logging.INFO)
+    """
+    Should see something like the following in the output if using LOGGING
+
+    DEBUG:root:Action status: completed
+    DEBUG:root:Action results: {'return-code': 0, 'stdout': '10.42.51.101\n'}
+    """
     jasyncio.run(main())