refactor: separate bind mount setup/clean

Move bind mount setup/clean up into separate functions.

Signed-off-by: Egor Lazarchuk <yegorlz@amazon.co.uk>

Author: ShadowCurse

tests/integration_tests/performance/test_jailer.py

@@ -14,6 +14,26 @@
 from framework.properties import global_props
 
 
+def setup_bind_mounts(tmp_path, n):
+    """
+    Create bind mount points. The exact location of them
+    does not matter, they just need to exist.
+    """
+    mounts_paths = tmp_path / "mounts"
+    os.makedirs(mounts_paths)
+    for m in range(n):
+        mount_path = f"{mounts_paths}/mount{m}"
+        os.makedirs(mount_path)
+        utils.check_output(f"mount --bind {mount_path} {mount_path}")
+
+
+def clean_up_mounts(tmp_path):
+    """Cleanup mounts and jailer dirs"""
+    mounts_paths = tmp_path / "mounts"
+    for d in os.listdir(mounts_paths):
+        utils.check_output(f"umount {mounts_paths}/{d}")
+
+
 @pytest.mark.nonci
 @pytest.mark.parametrize("jailers", [1, 100, 300, 500])
 @pytest.mark.parametrize("mounts", [0, 100, 300, 500])
@@ -26,14 +46,7 @@ def test_jailer_startup(
 
     jailer_binary = microvm_factory.jailer_binary_path
 
-    # Create bind mount points. The exact location of them
-    # does not matter, they just need to exist.
-    mounts_paths = tmp_path / "mounts"
-    os.makedirs(mounts_paths)
-    for m in range(mounts):
-        mount_path = f"{mounts_paths}/mount{m}"
-        os.makedirs(mount_path)
-        utils.check_output(f"mount --bind {mount_path} {mount_path}")
+    setup_bind_mounts(tmp_path, mounts)
 
     metrics.set_dimensions(
         {
@@ -79,9 +92,7 @@ def test_jailer_startup(
                 unit="Microseconds",
             )
 
-    # Cleanup mounts and jailer dirs
-    for d in os.listdir(mounts_paths):
-        utils.check_output(f"umount {mounts_paths}/{d}")
+    clean_up_mounts(tmp_path)
     shutil.rmtree(DEFAULT_CHROOT_PATH)
 
 
@@ -98,14 +109,7 @@ def test_jailer_startup_parallel(
 
     jailer_binary = microvm_factory.jailer_binary_path
 
-    # Create bind mount points. The exact location of them
-    # does not matter, they just need to exist.
-    mounts_paths = tmp_path / "mounts"
-    os.makedirs(mounts_paths)
-    for m in range(mounts):
-        mount_path = f"{mounts_paths}/mount{m}"
-        os.makedirs(mount_path)
-        utils.check_output(f"mount --bind {mount_path} {mount_path}")
+    setup_bind_mounts(tmp_path, mounts)
 
     metrics.set_dimensions(
         {
@@ -143,7 +147,5 @@ def test_jailer_startup_parallel(
                 unit="Microseconds",
             )
 
-    # Cleanup mounts and jailer dirs
-    for d in os.listdir(mounts_paths):
-        utils.check_output(f"umount {mounts_paths}/{d}")
+    clean_up_mounts(tmp_path)
     shutil.rmtree(DEFAULT_CHROOT_PATH)