media/libaom/test_cmakeparser.py

   1 # This Source Code Form is subject to the terms of the Mozilla Public
   2 # License, v. 2.0. If a copy of the MPL was not distributed with this
   3 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   4 from pyparsing import ParseException
   5 import unittest
   6
   7 import cmakeparser as cp
   8
   9 class TestCMakeParser(unittest.TestCase):
  10     def test_arguments(self):
  11         self.assertEqual(cp.arguments.parseString('1').asList(), ['1'])
  12         self.assertEqual(cp.arguments.parseString('(1 2)').asList(),
  13                          ['(', '1', '2', ')'])
  14
  15     def test_command(self):
  16         self.assertEqual(cp.command.parseString('blah()').asList(),
  17                          [['blah', '(', ')']])
  18         self.assertEqual(cp.command.parseString('blah(1)').asList(),
  19                          [['blah', '(', '1', ')']])
  20         self.assertEqual(cp.command.parseString('blah(1 (2 3))').asList(),
  21                          [['blah', '(', '1', '(', '2', '3', ')', ')']])
  22
  23     def test_evaluate_boolean(self):
  24         self.assertTrue(cp.evaluate_boolean({}, ['TRUE']))
  25         self.assertFalse(cp.evaluate_boolean({}, ['NOT', 'TRUE']))
  26         self.assertFalse(cp.evaluate_boolean({}, ['TRUE', 'AND', 'FALSE']))
  27         self.assertTrue(cp.evaluate_boolean({}, ['ABC', 'MATCHES', '^AB']))
  28         self.assertTrue(cp.evaluate_boolean({}, ['TRUE', 'OR', 'FALSE']))
  29         self.assertTrue(cp.evaluate_boolean({}, ['ABC', 'STREQUAL', 'ABC']))
  30         self.assertTrue(cp.evaluate_boolean({'ABC': '1'}, ['ABC']))
  31         self.assertFalse(cp.evaluate_boolean({'ABC': '0'}, ['${ABC}']))
  32         self.assertFalse(cp.evaluate_boolean({}, ['ABC']))
  33         self.assertFalse(cp.evaluate_boolean({}, ['${ABC}']))
  34         self.assertTrue(cp.evaluate_boolean({'YES_ABC': 1, 'VAL': 'ABC'},
  35                                             ['YES_${VAL}']))
  36         self.assertTrue(cp.evaluate_boolean({'ABC': 'DEF', 'DEF': 1},
  37                                             ['${ABC}']))
  38         self.assertTrue(cp.evaluate_boolean({}, ['FALSE', 'OR', '(', 'FALSE', 'OR', 'TRUE', ')']))
  39         self.assertFalse(cp.evaluate_boolean({}, ['FALSE', 'OR', '(', 'FALSE', 'AND', 'TRUE', ')']))
  40
  41     def test_foreach(self):
  42         s = """
  43             set(STUFF A B C D E F)
  44             foreach(item ${STUFF})
  45                 set(YES_${item} 1)
  46             endforeach ()
  47             """
  48         parsed = cp.cmake.parseString(s)
  49         variables = {}
  50         cp.evaluate(variables, [], parsed)
  51         for k in ['A', 'B', 'C', 'D', 'E', 'F']:
  52             self.assertEqual(variables['YES_%s' % k], '1')
  53
  54         s = """
  55             set(STUFF "A;B;C;D;E;F")
  56             foreach(item ${STUFF})
  57                 set(${item} 1)
  58             endforeach ()
  59             """
  60         parsed = cp.cmake.parseString(s)
  61         variables = {}
  62         cp.evaluate(variables, [], parsed)
  63         for k in ['A', 'B', 'C', 'D', 'E', 'F']:
  64             self.assertEqual(variables[k], '1')
  65
  66         s = """
  67             set(STUFF D E F)
  68             foreach(item A B C ${STUFF})
  69                 set(${item} 1)
  70             endforeach ()
  71             """
  72         parsed = cp.cmake.parseString(s)
  73         variables = {}
  74         cp.evaluate(variables, [], parsed)
  75         for k in ['A', 'B', 'C', 'D', 'E', 'F']:
  76             self.assertEqual(variables[k], '1')
  77
  78     def test_list(self):
  79         s = 'list(APPEND TEST 1 1 2 3 5 8 13)'
  80         parsed = cp.cmake.parseString(s)
  81         variables = {}
  82         cache_variables = []
  83         cp.evaluate(variables, cache_variables, parsed)
  84         self.assertEqual(variables['TEST'], '1 1 2 3 5 8 13')
  85         self.assertEqual(len(cache_variables), 0)
  86
  87         s = """
  88             set(TEST 1)
  89             list(APPEND TEST 1 2 3 5 8 13)
  90             """
  91         parsed = cp.cmake.parseString(s)
  92         variables = {}
  93         cache_variables = []
  94         cp.evaluate(variables, cache_variables, parsed)
  95         self.assertEqual(variables['TEST'], '1 1 2 3 5 8 13')
  96         self.assertEqual(len(cache_variables), 0)
  97
  98     def test_malformed_input(self):
  99         self.assertEqual(len(cp.cmake.parseString('func ((A)')), 0)
 100         self.assertEqual(len(cp.cmake.parseString('cmd"arg"(arg2)')), 0)
 101         self.assertRaises(ParseException, cp.command.parseString, 'func ((A)')
 102         self.assertRaises(ParseException, cp.identifier.parseString,
 103                           '-asdlf')
 104         self.assertEqual(cp.identifier.parseString('asd-lf')[0], 'asd')
 105         self.assertRaises(ParseException, cp.quoted_argument.parseString,
 106                           'blah"')
 107         s = """
 108             " blah blah
 109             blah
 110             """
 111         self.assertRaises(ParseException, cp.quoted_argument.parseString,
 112                           s)
 113         self.assertRaises(ParseException, cp.quoted_argument.parseString,
 114                           'asdlf')
 115         self.assertRaises(ParseException, cp.unquoted_argument.parseString,
 116                           '#asdflkj')
 117
 118     def test_parse_if(self):
 119         s = """
 120             if (A)
 121                 B()
 122             elseif (C)
 123                 D()
 124             elseif (E)
 125                 F()
 126             else ()
 127                 H()
 128             endif ()
 129             I()
 130             """
 131         parsed = cp.cmake.parseString(s)
 132         self.assertEqual(len(parsed), 10)
 133         end, conditions = cp.parse_if(parsed, 0)
 134         self.assertEqual(parsed[end][0], 'endif')
 135         self.assertEqual(len(conditions), 4)
 136         self.assertEqual(conditions[0][0], ['A'])
 137         self.assertEqual(conditions[0][1][0], parsed[1])
 138         self.assertEqual(conditions[1][0], ['C'])
 139         self.assertEqual(conditions[1][1][0], parsed[3])
 140         self.assertEqual(conditions[2][0], ['E'])
 141         self.assertEqual(conditions[2][1][0], parsed[5])
 142         self.assertEqual(conditions[3][0], ['TRUE'])
 143         self.assertEqual(conditions[3][1][0], parsed[7])
 144
 145     def test_return(self):
 146         s = """
 147             set(TEST 2)
 148             if (true)
 149                 return()
 150             endif ()
 151             set(TEST 3)
 152             """
 153         parsed = cp.cmake.parseString(s)
 154         variables = {}
 155         cache_variables = []
 156         cp.evaluate(variables, cache_variables, parsed)
 157         self.assertEqual(variables['TEST'], '2')
 158         self.assertEqual(len(cache_variables), 0)
 159
 160     def test_set(self):
 161         s = """set(TEST 2)"""
 162         parsed = cp.cmake.parseString(s)
 163         variables = {}
 164         cache_variables = []
 165         cp.evaluate(variables, cache_variables, parsed)
 166         self.assertEqual(variables['TEST'], '2')
 167         self.assertEqual(len(cache_variables), 0)
 168
 169         s = """set(TEST 3 CACHE "documentation")"""
 170         parsed = cp.cmake.parseString(s)
 171         variables = {}
 172         cache_variables = []
 173         cp.evaluate(variables, cache_variables, parsed)
 174         self.assertEqual(variables['TEST'], '3')
 175         self.assertTrue('TEST' in cache_variables)
 176
 177         s = """set(TEST A B C D)"""
 178         parsed = cp.cmake.parseString(s)
 179         variables = {}
 180         cp.evaluate(variables, [], parsed)
 181         self.assertEqual(variables['TEST'], 'A B C D')
 182
 183         s = """set(TEST ${TEST} E F G H)"""
 184         parsed = cp.cmake.parseString(s)
 185         cp.evaluate(variables, [], parsed)
 186         self.assertEqual(variables['TEST'], 'A B C D E F G H')
 187
 188
 189 if __name__ == '__main__':
 190     unittest.main()