cephadm: add cephadm ContextManager

to mock the CephadmContext during unit tests

Signed-off-by: Michael Fritch <mfritch@suse.com>
(cherry picked from commit 3e714ef1dcd4d30816932fdb35e8abcde773dca8)

diff --git a/src/cephadm/tests/fixtures.py b/src/cephadm/tests/fixtures.py
index ab5dcab652d3acd15909c6703bf6bd09bc258f11..ec4244abede8e8af05e718c5ad645ce7c38347d1 100644 (file)
--- a/src/cephadm/tests/fixtures.py
+++ b/src/cephadm/tests/fixtures.py
@@ -3,7 +3,9 @@ import os
 import pytest
 import time
 
+from contextlib import contextmanager
 from pyfakefs import fake_filesystem
+from typing import Dict, List, Optional
 
 
 with mock.patch('builtins.open', create=True):
@@ -62,3 +64,26 @@ def cephadm_fs(
             fs.create_dir(cd.UNIT_DIR)
 
             yield fs
+
+
+@contextmanager
+def with_cephadm_ctx(
+    cmd: List[str],
+    list_networks: Optional[Dict[str,Dict[str,List[str]]]] = None
+):
+    """
+    :param cmd: cephadm command argv
+    :param list_networks: mock 'list-networks' return
+    """
+    if not list_networks:
+        list_networks = {}
+
+    with mock.patch('cephadm.get_parm'), \
+         mock.patch('cephadm.attempt_bind'), \
+         mock.patch('cephadm.call', return_value=('', '', 0)), \
+         mock.patch('cephadm.find_executable', return_value='foo'), \
+         mock.patch('cephadm.is_available', return_value=True), \
+         mock.patch('cephadm.json_loads_retry', return_value={'epoch' : 1}), \
+         mock.patch('cephadm.list_networks', return_value=list_networks):
+             ctx: cd.CephadmContext = cd.cephadm_init_ctx(cmd)
+             yield ctx