aboutsummaryrefslogtreecommitdiff
path: root/cros_utils/command_executer_unittest.py
blob: 7cd46a71276383550c820e691b5a8994b4780bdd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2019 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unittest for command_executer.py."""


import time
import unittest

from cros_utils import command_executer


class CommandExecuterTest(unittest.TestCase):
    """Test for CommandExecuter class."""

    def testTimeout(self):
        timeout = 1
        logging_level = "average"
        ce = command_executer.CommandExecuter(logging_level)
        start = time.time()
        command = "sleep 20"
        ce.RunCommand(
            command, command_timeout=timeout, terminated_timeout=timeout
        )
        end = time.time()
        self.assertTrue(round(end - start) == timeout)


if __name__ == "__main__":
    unittest.main()