third_party/pigweed/repo/pw_arduino_build/py/builder_test.py

   1 # Copyright 2020 The Pigweed Authors
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License"); you may not
   4 # use this file except in compliance with the License. You may obtain a copy of
   5 # the License at
   6 #
   7 #     https://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  11 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  12 # License for the specific language governing permissions and limitations under
  13 # the License.
  14 """Tests for arduinobuilder."""
  15
  16 import shlex
  17 import unittest
  18 from parameterized import parameterized  # type: ignore
  19
  20
  21 class TestShellArgumentSplitting(unittest.TestCase):
  22     """Tests to ensure shlex.split handles expected use cases."""
  23
  24     @parameterized.expand([
  25         (
  26             "remove-both-quotes",
  27             """ -DUSB_CONFIG_POWER=100 """
  28             """ '-DUSB_MANUFACTURER="Adafruit LLC"' """
  29             """ '-DUSB_PRODUCT="Adafruit PyGamer Advance M4"' """
  30             """ "-I$HOME/samd/1.6.2/cores/arduino/TinyUSB" """,
  31             [
  32                 """ -DUSB_CONFIG_POWER=100 """.strip(),
  33                 """ -DUSB_MANUFACTURER="Adafruit LLC" """.strip(),
  34                 """ -DUSB_PRODUCT="Adafruit PyGamer Advance M4" """.strip(),
  35                 """ -I$HOME/samd/1.6.2/cores/arduino/TinyUSB """.strip(),
  36             ]
  37         )
  38     ]) # yapf: disable
  39     def test_split_arguments_and_remove_quotes(self, unused_test_name,
  40                                                input_string, expected):
  41         """Test splitting a string into a list of arguments with quotes removed.
  42         """
  43         result = shlex.split(input_string)
  44         self.assertEqual(result, expected)
  45
  46
  47 if __name__ == '__main__':
  48     unittest.main()