From 3f1ec453daa74da7b8c231daa92395eef1b5a4c6 Mon Sep 17 00:00:00 2001
From: Andrew McLeod <andrew.mcleod@canonical.com>
Date: Fri, 6 Mar 2020 09:40:32 +0100
Subject: [PATCH] add lxd container support to get_machine_status

---
 zaza/openstack/utilities/juju.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/zaza/openstack/utilities/juju.py b/zaza/openstack/utilities/juju.py
index fd7abc1..27107fa 100644
--- a/zaza/openstack/utilities/juju.py
+++ b/zaza/openstack/utilities/juju.py
@@ -137,7 +137,11 @@ def get_machine_status(machine, key=None, model_name=None):
     :rtype: dict
     """
     status = get_full_juju_status(model_name=model_name)
-    status = status.machines.get(machine)
+    if "lxd" in machine:
+        host = machine.split('/')[0]
+        status = status.machines.get(host)['containers'][machine]
+    else:
+        status = status.machines.get(machine)
     if key:
         status = status.get(key)
     return status