Support all possible bitflags when computing the PSL DAFSA

net/tools/dafsa/make_dafsa_unittest.py

 #!/usr/bin/env python
 # Copyright 2014 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 sys
 import unittest
 import make_dafsa
 
@@ skipping 11 matching lines @@
     self.assertRaises(make_dafsa.InputError, make_dafsa.parse_gperf, infile3)
 
   def testBadValues(self):
     """Tests exception is thrown when value is out of range."""
     infile1 = [ '%%', 'a, -1', '%%' ]
     self.assertRaises(make_dafsa.InputError, make_dafsa.parse_gperf, infile1)
 
     infile2 = [ '%%', 'a, x', '%%' ]
     self.assertRaises(make_dafsa.InputError, make_dafsa.parse_gperf, infile2)
 
-    infile3 = [ '%%', 'a,  3', '%%' ]
-    self.assertRaises(make_dafsa.InputError, make_dafsa.parse_gperf, infile3)
-
-    infile4 = [ '%%', 'a,  6', '%%' ]
-    self.assertRaises(make_dafsa.InputError, make_dafsa.parse_gperf, infile4)
-
-    infile5 = [ '%%', 'a,  12', '%%' ]
+    infile5 = [ '%%', 'a, 12', '%%' ]
     self.assertRaises(make_dafsa.InputError, make_dafsa.parse_gperf, infile5)
 
   def testValues(self):
     """Tests legal values are accepted."""
     infile1 = [ '%%', 'a, 0', '%%' ]
     words1 = [ 'a0' ]
     self.assertEqual(make_dafsa.parse_gperf(infile1), words1)
 
     infile2 = [ '%%', 'a, 1', '%%' ]
     words2 = [ 'a1' ]
     self.assertEqual(make_dafsa.parse_gperf(infile2), words2)
 
     infile3 = [ '%%', 'a, 2', '%%' ]
     words3 = [ 'a2' ]
     self.assertEqual(make_dafsa.parse_gperf(infile3), words3)
 
-    infile4 = [ '%%', 'a, 4', '%%' ]
-    words4 = [ 'a4' ]
+    infile4 = [ '%%', 'a, 3', '%%' ]
+    words4 = [ 'a3' ]
     self.assertEqual(make_dafsa.parse_gperf(infile4), words4)
 
+    infile5 = [ '%%', 'a, 4', '%%' ]
+    words5 = [ 'a4' ]
+    self.assertEqual(make_dafsa.parse_gperf(infile5), words5)
+
+    infile6 = [ '%%', 'a, 6', '%%' ]
+    words6 = [ 'a6' ]
+    self.assertEqual(make_dafsa.parse_gperf(infile6), words6)
+
   def testOneWord(self):
     """Tests a single key can be parsed."""
     infile = [ '%%', 'apa, 1', '%%' ]
     words = [ 'apa1' ]
     self.assertEqual(make_dafsa.parse_gperf(infile), words)
 
   def testTwoWords(self):
     """Tests a sequence of keys can be parsed."""
     infile = [ '%%', 'apa, 1', 'bepa.com, 2', '%%' ]
     words = [ 'apa1', 'bepa.com2' ]
@@ skipping 679 matching lines @@
     infile = [ '%%', 'aa, 1', 'bbb, 2', 'baa, 1', '%%' ]
     bytes = [ 0x02, 0x83, 0xE2, 0x02, 0x83, 0x61, 0x61, 0x81, 0x62, 0x62,
               0x82 ]
     outfile = make_dafsa.to_cxx(bytes)
     self.assertEqual(make_dafsa.words_to_cxx(make_dafsa.parse_gperf(infile)),
                       outfile)
 
 
 if __name__ == '__main__':
   unittest.main()