build/android/devil/utils/cmd_helper_test.py

   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
   3 # found in the LICENSE file.
   4
   5 """Tests for the cmd_helper module."""
   6
   7 import unittest
   8 import subprocess
   9
  10 from devil.utils import cmd_helper
  11
  12
  13 class CmdHelperSingleQuoteTest(unittest.TestCase):
  14
  15   def testSingleQuote_basic(self):
  16     self.assertEquals('hello',
  17                       cmd_helper.SingleQuote('hello'))
  18
  19   def testSingleQuote_withSpaces(self):
  20     self.assertEquals("'hello world'",
  21                       cmd_helper.SingleQuote('hello world'))
  22
  23   def testSingleQuote_withUnsafeChars(self):
  24     self.assertEquals("""'hello'"'"'; rm -rf /'""",
  25                       cmd_helper.SingleQuote("hello'; rm -rf /"))
  26
  27   def testSingleQuote_dontExpand(self):
  28     test_string = 'hello $TEST_VAR'
  29     cmd = 'TEST_VAR=world; echo %s' % cmd_helper.SingleQuote(test_string)
  30     self.assertEquals(test_string,
  31                       cmd_helper.GetCmdOutput(cmd, shell=True).rstrip())
  32
  33
  34 class CmdHelperDoubleQuoteTest(unittest.TestCase):
  35
  36   def testDoubleQuote_basic(self):
  37     self.assertEquals('hello',
  38                       cmd_helper.DoubleQuote('hello'))
  39
  40   def testDoubleQuote_withSpaces(self):
  41     self.assertEquals('"hello world"',
  42                       cmd_helper.DoubleQuote('hello world'))
  43
  44   def testDoubleQuote_withUnsafeChars(self):
  45     self.assertEquals('''"hello\\"; rm -rf /"''',
  46                       cmd_helper.DoubleQuote('hello"; rm -rf /'))
  47
  48   def testSingleQuote_doExpand(self):
  49     test_string = 'hello $TEST_VAR'
  50     cmd = 'TEST_VAR=world; echo %s' % cmd_helper.DoubleQuote(test_string)
  51     self.assertEquals('hello world',
  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