Home | History | Annotate | Download | only in android 
      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)
    136