1#!/usr/bin/env python3
2#
3# Copyright 2018, The Android Open Source Project
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9#     http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
17"""Unittests for atest_arg_parser."""
18
19
20import unittest
21
22import atest_arg_parser
23
24
25class AtestArgParserUnittests(unittest.TestCase):
26    """Unit tests for atest_arg_parser.py"""
27
28    def test_get_args(self):
29        """Test get_args(): flatten a nested list. """
30        parser = atest_arg_parser.AtestArgParser()
31        parser.add_argument('-t', '--test', help='Run the tests.')
32        parser.add_argument('-b', '--build', help='Run a build.')
33        parser.add_argument('--generate-baseline', help='Generate a baseline.')
34        test_args = ['-t', '--test',
35                     '-b', '--build',
36                     '--generate-baseline',
37                     '-h', '--help'].sort()
38        self.assertEqual(test_args, parser.get_args().sort())
39
40
41if __name__ == '__main__':
42    unittest.main()
43