mojo/devtools/common/devtoolslib/shell_arguments_unittest.py

# Copyright 2015 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 imp
import os.path
import sys
import unittest

try:
  imp.find_module("devtoolslib")
except ImportError:
  sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from devtoolslib.shell_arguments import AppendToArgument


class AppendToArgumentTest(unittest.TestCase):
  """Tests AppendToArgument()."""

  def testAppendToEmpty(self):
    arguments = []
    key = '--something='
    value = 'val'
    expected_result = ['--something=val']
    self.assertEquals(expected_result, AppendToArgument(arguments, key, value))

  def testAppendToNonEmpty(self):
    arguments = ['--other']
    key = '--something='
    value = 'val'
    expected_result = ['--other', '--something=val']
    self.assertEquals(expected_result, AppendToArgument(arguments, key, value))

  def testAppendToExisting(self):
    arguments = ['--something=old_val']
    key = '--something='
    value = 'val'
    expected_result = ['--something=old_val,val']
    self.assertEquals(expected_result, AppendToArgument(arguments, key, value))


if __name__ == "__main__":
  unittest.main()