"""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)