1#!/usr/bin/env python
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 test_mapping"""
18
19import unittest
20
21import test_mapping
22import unittest_constants as uc
23
24
25class TestMappingUnittests(unittest.TestCase):
26    """Unit tests for test_mapping.py"""
27
28    def test_parsing(self):
29        """Test creating TestDetail object"""
30        detail = test_mapping.TestDetail(uc.TEST_MAPPING_TEST)
31        self.assertEqual(uc.TEST_MAPPING_TEST['name'], detail.name)
32        self.assertEqual([], detail.options)
33
34    def test_parsing_with_option(self):
35        """Test creating TestDetail object with option configured"""
36        detail = test_mapping.TestDetail(uc.TEST_MAPPING_TEST_WITH_OPTION)
37        self.assertEqual(uc.TEST_MAPPING_TEST_WITH_OPTION['name'], detail.name)
38        self.assertEqual(uc.TEST_MAPPING_TEST_WITH_OPTION_STR, str(detail))
39
40    def test_parsing_with_bad_option(self):
41        """Test creating TestDetail object with bad option configured"""
42        with self.assertRaises(Exception) as context:
43            test_mapping.TestDetail(uc.TEST_MAPPING_TEST_WITH_BAD_OPTION)
44        self.assertEqual(
45            'Each option can only have one key.', str(context.exception))
46
47
48if __name__ == '__main__':
49    unittest.main()
50