Skip to content

Commit

Permalink
Reboot device after L 1 tests
Browse files Browse the repository at this point in the history
Signed-off-by: Stepan Vovk <[email protected]>
  • Loading branch information
StepanVovkPLV authored and yuriilisovskyi committed Apr 20, 2023
1 parent 5f9c6e9 commit 131ad5d
Showing 1 changed file with 29 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,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
Expand Down Expand Up @@ -76,6 +77,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:
Expand All @@ -87,6 +97,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)
Expand Down Expand Up @@ -125,6 +140,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


Expand All @@ -148,6 +169,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
Expand Down

0 comments on commit 131ad5d

Please sign in to comment.