| OLD | NEW |
| 1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 """Tests for the cmd_helper module.""" | 5 """Tests for the cmd_helper module.""" |
| 6 | 6 |
| 7 import unittest | 7 import unittest |
| 8 import subprocess |
| 8 | 9 |
| 9 from pylib import cmd_helper | 10 from pylib import cmd_helper |
| 10 | 11 |
| 11 # TODO(jbudorick) Make these tests run on the bots. | |
| 12 | |
| 13 | 12 |
| 14 class CmdHelperSingleQuoteTest(unittest.TestCase): | 13 class CmdHelperSingleQuoteTest(unittest.TestCase): |
| 15 | 14 |
| 16 def testSingleQuote_basic(self): | 15 def testSingleQuote_basic(self): |
| 17 self.assertEquals('hello', | 16 self.assertEquals('hello', |
| 18 cmd_helper.SingleQuote('hello')) | 17 cmd_helper.SingleQuote('hello')) |
| 19 | 18 |
| 20 def testSingleQuote_withSpaces(self): | 19 def testSingleQuote_withSpaces(self): |
| 21 self.assertEquals("'hello world'", | 20 self.assertEquals("'hello world'", |
| 22 cmd_helper.SingleQuote('hello world')) | 21 cmd_helper.SingleQuote('hello world')) |
| (...skipping 21 matching lines...) Expand all Loading... |
| 44 | 43 |
| 45 def testDoubleQuote_withUnsafeChars(self): | 44 def testDoubleQuote_withUnsafeChars(self): |
| 46 self.assertEquals('''"hello\\"; rm -rf /"''', | 45 self.assertEquals('''"hello\\"; rm -rf /"''', |
| 47 cmd_helper.DoubleQuote('hello"; rm -rf /')) | 46 cmd_helper.DoubleQuote('hello"; rm -rf /')) |
| 48 | 47 |
| 49 def testSingleQuote_doExpand(self): | 48 def testSingleQuote_doExpand(self): |
| 50 test_string = 'hello $TEST_VAR' | 49 test_string = 'hello $TEST_VAR' |
| 51 cmd = 'TEST_VAR=world; echo %s' % cmd_helper.DoubleQuote(test_string) | 50 cmd = 'TEST_VAR=world; echo %s' % cmd_helper.DoubleQuote(test_string) |
| 52 self.assertEquals('hello world', | 51 self.assertEquals('hello world', |
| 53 cmd_helper.GetCmdOutput(cmd, shell=True).rstrip()) | 52 cmd_helper.GetCmdOutput(cmd, shell=True).rstrip()) |
| 53 |
| 54 |
| 55 class CmdHelperIterCmdOutputLinesTest(unittest.TestCase): |
| 56 """Test IterCmdOutputLines with some calls to the unix 'seq' command.""" |
| 57 |
| 58 def testIterCmdOutputLines_success(self): |
| 59 for num, line in enumerate( |
| 60 cmd_helper.IterCmdOutputLines(['seq', '10']), 1): |
| 61 self.assertEquals(num, int(line)) |
| 62 |
| 63 def testIterCmdOutputLines_exitStatusFail(self): |
| 64 with self.assertRaises(subprocess.CalledProcessError): |
| 65 for num, line in enumerate( |
| 66 cmd_helper.IterCmdOutputLines('seq 10 && false', shell=True), 1): |
| 67 self.assertEquals(num, int(line)) |
| 68 # after reading all the output we get an exit status of 1 |
| 69 |
| 70 def testIterCmdOutputLines_exitStatusIgnored(self): |
| 71 for num, line in enumerate( |
| 72 cmd_helper.IterCmdOutputLines('seq 10 && false', shell=True, |
| 73 check_status=False), 1): |
| 74 self.assertEquals(num, int(line)) |
| 75 |
| 76 def testIterCmdOutputLines_exitStatusSkipped(self): |
| 77 for num, line in enumerate( |
| 78 cmd_helper.IterCmdOutputLines('seq 10 && false', shell=True), 1): |
| 79 self.assertEquals(num, int(line)) |
| 80 # no exception will be raised because we don't attempt to read past |
| 81 # the end of the output and, thus, the status never gets checked |
| 82 if num == 10: |
| 83 break |
| OLD | NEW |