testhelper: add directmnt and get_share()

Add an additional entity directmnt which points to the mount to the
underlying filesystem.

Also add get_share() to get the share entity.

Signed-off-by: Sachin Prabhu <[email protected]>

Author: spuiuk

testhelper/testhelper.py

@@ -93,19 +93,58 @@ def get_shares(test_info: dict) -> typing.List[dict]:
     default_fs = test_info.get("test_backend", None)
     arr = []
     for sharename in test_info.get("exported_sharenames", []):
-        arr.append({"name": sharename, "mntdir": None, "fs": default_fs})
+        arr.append(
+            {
+                "name": sharename,
+                "mntdir": None,
+                "fs": default_fs,
+                "directmnt": None,
+            }
+        )
     for share in test_info.get("exported_shares", []):
         arr.append(
             {
                 "name": share.get("name", None),
                 "mntdir": share.get("mntdir", None),
                 "fs": share.get("fs", default_fs),
+                "directmnt": share.get("directmnt", None),
             }
         )
     test_info["shares"] = arr
     return arr
 
 
+def get_share(test_info: dict, sharename: str) -> dict:
+    """
+    Get share dict for a given sharename
+
+    Parameters:
+    test_info: Dict containing the parsed yaml file.
+    sharename: name of the share
+    Returns:
+    dict of the share
+    """
+    for share in get_shares(test_info):
+        if share["name"] == sharename:
+            return share
+    return {}
+
+def get_shares_with_directmnt(test_info: dict) -> typing.List[dict]:
+    """
+    Get list of shares with directmnt enabled
+
+    Parameters:
+    test_info: Dict containing the parsed yaml file.
+    Returns:
+    list of shares
+    """
+    arr = []
+    for share in get_shares(test_info):
+        if share["directmnt"] is not None:
+            arr.append(share)
+    return arr
+
+
 def is_premounted_share(share: dict) -> bool:
     """
     Is the share dict a premounted share