sdk/platform-tools/systrace/catapult/devil/devil/android/flag_changer_test.py

   1 #!/usr/bin/env python
   2 # Copyright 2017 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import posixpath
   7 import unittest
   8
   9 from devil.android import flag_changer
  10
  11
  12 _CMDLINE_FILE = 'chrome-command-line'
  13
  14
  15 class _FakeDevice(object):
  16   def __init__(self):
  17     self.build_type = 'user'
  18     self.has_root = True
  19     self.file_system = {}
  20
  21   def HasRoot(self):
  22     return self.has_root
  23
  24   def PathExists(self, filepath):
  25     return filepath in self.file_system
  26
  27   def RemovePath(self, path, **_kwargs):
  28     self.file_system.pop(path)
  29
  30   def WriteFile(self, path, contents, **_kwargs):
  31     self.file_system[path] = contents
  32
  33   def ReadFile(self, path, **_kwargs):
  34     return self.file_system[path]
  35
  36
  37 class FlagChangerTest(unittest.TestCase):
  38   def setUp(self):
  39     self.device = _FakeDevice()
  40     # pylint: disable=protected-access
  41     self.cmdline_path = posixpath.join(flag_changer._CMDLINE_DIR, _CMDLINE_FILE)
  42     self.cmdline_path_legacy = posixpath.join(
  43         flag_changer._CMDLINE_DIR_LEGACY, _CMDLINE_FILE)
  44
  45   def testFlagChanger_removeLegacyCmdLine(self):
  46     self.device.WriteFile(self.cmdline_path_legacy, 'chrome --old --stuff')
  47     self.assertTrue(self.device.PathExists(self.cmdline_path_legacy))
  48
  49     changer = flag_changer.FlagChanger(self.device, 'chrome-command-line')
  50     self.assertEquals(
  51         changer._cmdline_path,  # pylint: disable=protected-access
  52         self.cmdline_path)
  53     self.assertFalse(self.device.PathExists(self.cmdline_path_legacy))
  54
  55   def testFlagChanger_mustBeFileName(self):
  56     with self.assertRaises(ValueError):
  57       flag_changer.FlagChanger(self.device, '/data/local/chrome-command-line')
  58
  59
  60 class ParseSerializeFlagsTest(unittest.TestCase):
  61   def _testQuoteFlag(self, flag, expected_quoted_flag):
  62     # Start with an unquoted flag, check that it's quoted as expected.
  63     # pylint: disable=protected-access
  64     quoted_flag = flag_changer._QuoteFlag(flag)
  65     self.assertEqual(quoted_flag, expected_quoted_flag)
  66     # Check that it survives a round-trip.
  67     parsed_flags = flag_changer._ParseFlags('_ %s' % quoted_flag)
  68     self.assertEqual(len(parsed_flags), 1)
  69     self.assertEqual(flag, parsed_flags[0])
  70
  71   def testQuoteFlag_simple(self):
  72     self._testQuoteFlag('--simple-flag', '--simple-flag')
  73
  74   def testQuoteFlag_withSimpleValue(self):
  75     self._testQuoteFlag('--key=value', '--key=value')
  76
  77   def testQuoteFlag_withQuotedValue1(self):
  78     self._testQuoteFlag('--key=valueA valueB', '--key="valueA valueB"')
  79
  80   def testQuoteFlag_withQuotedValue2(self):
  81     self._testQuoteFlag(
  82         '--key=this "should" work', r'--key="this \"should\" work"')
  83
  84   def testQuoteFlag_withQuotedValue3(self):
  85     self._testQuoteFlag(
  86         "--key=this is 'fine' too", '''--key="this is 'fine' too"''')
  87
  88   def testQuoteFlag_withQuotedValue4(self):
  89     self._testQuoteFlag(
  90         "--key='I really want to keep these quotes'",
  91         '''--key="'I really want to keep these quotes'"''')
  92
  93   def testQuoteFlag_withQuotedValue5(self):
  94     self._testQuoteFlag(
  95         "--this is a strange=flag", '"--this is a strange=flag"')
  96
  97   def testQuoteFlag_withEmptyValue(self):
  98     self._testQuoteFlag('--some-flag=', '--some-flag=')
  99
 100   def _testParseCmdLine(self, command_line, expected_flags):
 101     # Start with a command line, check that flags are parsed as expected.
 102     # pylint: disable=protected-access
 103     flags = flag_changer._ParseFlags(command_line)
 104     self.assertItemsEqual(flags, expected_flags)
 105
 106     # Check that flags survive a round-trip.
 107     # Note: Although new_command_line and command_line may not match, they
 108     # should describe the same set of flags.
 109     new_command_line = flag_changer._SerializeFlags(flags)
 110     new_flags = flag_changer._ParseFlags(new_command_line)
 111     self.assertItemsEqual(new_flags, expected_flags)
 112
 113   def testParseCmdLine_simple(self):
 114     self._testParseCmdLine(
 115         'chrome --foo --bar="a b" --baz=true --fine="ok"',
 116         ['--foo', '--bar=a b', '--baz=true', '--fine=ok'])
 117
 118   def testParseCmdLine_withFancyQuotes(self):
 119     self._testParseCmdLine(
 120         r'''_ --foo="this 'is' ok"
 121               --bar='this \'is\' too'
 122               --baz="this \'is\' tricky"
 123         ''',
 124         ["--foo=this 'is' ok",
 125          "--bar=this 'is' too",
 126          r"--baz=this \'is\' tricky"])
 127
 128   def testParseCmdLine_withUnterminatedQuote(self):
 129     self._testParseCmdLine(
 130         '_ --foo --bar="I forgot something',
 131         ['--foo', '--bar=I forgot something'])
 132
 133
 134 if __name__ == '__main__':
 135   unittest.main(verbosity=2)