blob: c4df0ea59cfa519af5f2e1589b522d21babcf750 [file] [log] [blame]
#!/usr/bin/env python
# Copyright 2017 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import posixpath
import unittest
import six
from devil.android import flag_changer
_CMDLINE_FILE = 'chrome-command-line'
class _FakeDevice(object):
def __init__(self):
self.build_type = 'user'
self.has_root = True
self.file_system = {}
def HasRoot(self):
return self.has_root
def PathExists(self, filepath):
return filepath in self.file_system
def RemovePath(self, path, **_kwargs):
self.file_system.pop(path)
def WriteFile(self, path, contents, **_kwargs):
self.file_system[path] = contents
def ReadFile(self, path, **_kwargs):
return self.file_system[path]
class FlagChangerTest(unittest.TestCase):
def setUp(self):
self.device = _FakeDevice()
# pylint: disable=protected-access
self.cmdline_path = posixpath.join(flag_changer._CMDLINE_DIR, _CMDLINE_FILE)
self.cmdline_path_legacy = posixpath.join(flag_changer._CMDLINE_DIR_LEGACY,
_CMDLINE_FILE)
def testFlagChanger_removeAlternateCmdLine(self):
self.device.WriteFile(self.cmdline_path_legacy, 'chrome --old --stuff')
self.assertTrue(self.device.PathExists(self.cmdline_path_legacy))
changer = flag_changer.FlagChanger(self.device, 'chrome-command-line')
self.assertEqual(
changer._cmdline_path, # pylint: disable=protected-access
self.cmdline_path)
self.assertFalse(self.device.PathExists(self.cmdline_path_legacy))
def testFlagChanger_removeAlternateCmdLineLegacyPath(self):
self.device.WriteFile(self.cmdline_path, 'chrome --old --stuff')
self.assertTrue(self.device.PathExists(self.cmdline_path))
changer = flag_changer.FlagChanger(
self.device, 'chrome-command-line', use_legacy_path=True)
self.assertEqual(
changer._cmdline_path, # pylint: disable=protected-access
self.cmdline_path_legacy)
self.assertFalse(self.device.PathExists(self.cmdline_path))
def testFlagChanger_mustBeFileName(self):
with self.assertRaises(ValueError):
flag_changer.FlagChanger(self.device, '/data/local/chrome-command-line')
class ParseSerializeFlagsTest(unittest.TestCase):
def _testQuoteFlag(self, flag, expected_quoted_flag):
# Start with an unquoted flag, check that it's quoted as expected.
# pylint: disable=protected-access
quoted_flag = flag_changer._QuoteFlag(flag)
self.assertEqual(quoted_flag, expected_quoted_flag)
# Check that it survives a round-trip.
parsed_flags = flag_changer._ParseFlags('_ %s' % quoted_flag)
self.assertEqual(len(parsed_flags), 1)
self.assertEqual(flag, parsed_flags[0])
def testQuoteFlag_simple(self):
self._testQuoteFlag('--simple-flag', '--simple-flag')
def testQuoteFlag_withSimpleValue(self):
self._testQuoteFlag('--key=value', '--key=value')
def testQuoteFlag_withQuotedValue1(self):
self._testQuoteFlag('--key=valueA valueB', '--key="valueA valueB"')
def testQuoteFlag_withQuotedValue2(self):
self._testQuoteFlag('--key=this "should" work',
r'--key="this \"should\" work"')
def testQuoteFlag_withQuotedValue3(self):
self._testQuoteFlag("--key=this is 'fine' too",
'''--key="this is 'fine' too"''')
def testQuoteFlag_withQuotedValue4(self):
self._testQuoteFlag("--key='I really want to keep these quotes'",
'''--key="'I really want to keep these quotes'"''')
def testQuoteFlag_withQuotedValue5(self):
self._testQuoteFlag("--this is a strange=flag",
'"--this is a strange=flag"')
def testQuoteFlag_withEmptyValue(self):
self._testQuoteFlag('--some-flag=', '--some-flag=')
def _testParseCmdLine(self, command_line, expected_flags):
# Start with a command line, check that flags are parsed as expected.
# pylint: disable=protected-access
# pylint: disable=no-member
flags = flag_changer._ParseFlags(command_line)
if six.PY2:
self.assertItemsEqual(flags, expected_flags)
else:
self.assertCountEqual(flags, expected_flags)
# Check that flags survive a round-trip.
# Note: Although new_command_line and command_line may not match, they
# should describe the same set of flags.
new_command_line = flag_changer._SerializeFlags(flags)
new_flags = flag_changer._ParseFlags(new_command_line)
if six.PY2:
self.assertItemsEqual(new_flags, expected_flags)
else:
self.assertCountEqual(new_flags, expected_flags)
def testParseCmdLine_simple(self):
self._testParseCmdLine('chrome --foo --bar="a b" --baz=true --fine="ok"',
['--foo', '--bar=a b', '--baz=true', '--fine=ok'])
def testParseCmdLine_withFancyQuotes(self):
self._testParseCmdLine(
r'''_ --foo="this 'is' ok"
--bar='this \'is\' too'
--baz="this \'is\' tricky"
''', [
"--foo=this 'is' ok", "--bar=this 'is' too",
r"--baz=this \'is\' tricky"
])
def testParseCmdLine_withUnterminatedQuote(self):
self._testParseCmdLine('_ --foo --bar="I forgot something',
['--foo', '--bar=I forgot something'])
if __name__ == '__main__':
unittest.main(verbosity=2)