Reboot device after L 1 tests

Signed-off-by: Stepan Vovk <[email protected]>

DentOS_Framework/DentOsTestbed/src/dent_os_testbed/test/conftest.py

@@ -25,6 +25,7 @@
     tgen_utils_get_dent_devices_with_tgen,
     tgen_utils_stop_protocols,
 )
+reboot_after_test = None
 
 # Add python files for defining per folder fixtures here
 # And depending on the scope of fixtures, they can be used
@@ -75,6 +76,15 @@ def pytest_unconfigure(config):
     # Save and parameters for each suite run
 
 
+def pytest_collection_finish(session):
+    global reboot_after_test
+    for item in session.items:
+        # Save the name of the last L1 test in session
+        if "test_l1" in item.name:
+            reboot_after_test = session.items[-1]
+            break
+
+
 def pytest_runtest_setup(item):
     logger = AppLogger(DEFAULT_LOGGER)
     if logger:
@@ -86,6 +96,11 @@ def pytest_runtest_setup(item):
         )
         logger.info("=================================================================")
 
+    if reboot_after_test:
+        if reboot_after_test.name == item.name:
+            # Add `reboot_device` fixture to last L1 test
+            item.fixturenames.append("reboot_device")
+
 
 def pytest_runtest_teardown(item, nextitem):
     logger = AppLogger(DEFAULT_LOGGER)
@@ -124,6 +139,12 @@ def pytest_collection_modifyitems(session, config, items):
             if mark and mark.name.startswith("feature"):
                 if logger:
                     logger.info("pytest %s has feature markers:%s" % (item.name, item.own_markers))
+
+    for item in items:
+        # Move l1 tests to the end of the list
+        if "test_l1" in item.name:
+            items.sort(key=lambda item: "test_l1" in item.name)
+            break
     return False
 
 
@@ -147,6 +168,14 @@ async def _get_dent_devs_from_testbed(testbed):
     return devs
 
 
+@pytest_asyncio.fixture()
+async def reboot_device(testbed):
+    yield
+    devices = await _get_dent_devs_from_testbed(testbed)
+    to_reboot = [dev.reboot() for dev in devices]
+    await asyncio.gather(*to_reboot)
+
+
 @pytest_asyncio.fixture()
 async def cleanup_qdiscs(testbed):
     yield