mirror of https://github.com/mongodb/mongo
33 lines
973 B
Python
33 lines
973 B
Python
"""The unittest.TestCase for sleeping a given amount of time."""
|
|
|
|
import time
|
|
|
|
from buildscripts.resmokelib import logging
|
|
from buildscripts.resmokelib.testing.testcases import interface
|
|
|
|
|
|
class SleepTestCase(interface.TestCase):
|
|
"""SleepTestCase class."""
|
|
|
|
REGISTERED_NAME = "sleep_test"
|
|
|
|
def __init__(self, logger: logging.Logger, sleep_duration_secs: int):
|
|
"""Initialize the SleepTestCase with the duration to sleep for."""
|
|
|
|
sleep_duration_secs = int(sleep_duration_secs)
|
|
|
|
interface.TestCase.__init__(
|
|
self, logger, "Sleep", "{:d} seconds".format(sleep_duration_secs)
|
|
)
|
|
|
|
self.__sleep_duration_secs = sleep_duration_secs
|
|
|
|
def run_test(self):
|
|
"""Execute sleep."""
|
|
time.sleep(self.__sleep_duration_secs)
|
|
self.return_code = 0
|
|
|
|
def as_command(self):
|
|
"""Return the command invocation used to run the test."""
|
|
return "sleep {:d}".format(self.__sleep_duration_secs)
|