1ed69676d435b7b6983271ed8fab200627a0b966eGeorge Burgess IV#!/usr/bin/env python2 23b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian"""Unittest for command_executer.py.""" 3cbc5a2617f8d63e3319e159eab2c407952df1df9Rahul Chaudhry 4cbc5a2617f8d63e3319e159eab2c407952df1df9Rahul Chaudhryfrom __future__ import print_function 5cbc5a2617f8d63e3319e159eab2c407952df1df9Rahul Chaudhry 63b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlianimport time 73b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlianimport unittest 83b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian 93b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlianimport command_executer 103b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian 113b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian 123b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlianclass CommandExecuterTest(unittest.TestCase): 13e7c6fbf9cfd5f45dda21f2c4742bf5a62589d2edRahul Chaudhry """Test for CommandExecuter class.""" 14f2a3ef46f75d2196a93d3ed27f4d1fcf22b54fbeLuis Lozano 153b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian def testTimeout(self): 163b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian timeout = 1 17f2a3ef46f75d2196a93d3ed27f4d1fcf22b54fbeLuis Lozano logging_level = 'average' 18c454cee542ca459ef9bd87c9f72e81c822caf1e5cmtice ce = command_executer.CommandExecuter(logging_level) 193b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian start = time.time() 20f2a3ef46f75d2196a93d3ed27f4d1fcf22b54fbeLuis Lozano command = 'sleep 20' 213b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian ce.RunCommand(command, command_timeout=timeout, terminated_timeout=timeout) 223b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian end = time.time() 233b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian self.assertTrue(round(end - start) == timeout) 243b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian 25f2a3ef46f75d2196a93d3ed27f4d1fcf22b54fbeLuis Lozano 26f2a3ef46f75d2196a93d3ed27f4d1fcf22b54fbeLuis Lozanoif __name__ == '__main__': 273b85c8bf5c99c0445d71439fa4c0ed86d7747d9eyunlian unittest.main() 28