From 93a9aff92796ce626ae969fc0554f3411dc3adf2 Mon Sep 17 00:00:00 2001
From: Liam Young <liam.young@canonical.com>
Date: Mon, 25 Jan 2021 11:47:42 +0000
Subject: [PATCH] Fix context manager mock

---
 unit_tests/utilities/test_zaza_utilities_openstack.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/unit_tests/utilities/test_zaza_utilities_openstack.py b/unit_tests/utilities/test_zaza_utilities_openstack.py
index e4abfb1..67cd417 100644
--- a/unit_tests/utilities/test_zaza_utilities_openstack.py
+++ b/unit_tests/utilities/test_zaza_utilities_openstack.py
@@ -1392,7 +1392,7 @@ class TestAsyncOpenstackUtils(ut_utils.AioTestCase):
 
         class AsyncContextManagerMock(test_mock):
             async def __aenter__(self):
-                yield model_mock
+                return self
 
             async def __aexit__(self, *args):
                 pass