headless: Make API generator more robust against weird enum literals

headless/lib/browser/client_api_generator_unittest.py

 # Copyright 2016 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 argparse
 import client_api_generator
 import shutil
 import sys
 import tempfile
 import unittest
@@ skipping 17 matching lines @@
     self.assertEqual(client_api_generator.DashToCamelCase('foo-bar'), 'FooBar')
     self.assertEqual(client_api_generator.DashToCamelCase('foo-'), 'Foo')
     self.assertEqual(client_api_generator.DashToCamelCase('-bar'), 'Bar')
 
   def test_CamelCaseToHackerStyle(self):
     self.assertEqual(client_api_generator.CamelCaseToHackerStyle('FooBar'),
                      'foo_bar')
     self.assertEqual(client_api_generator.CamelCaseToHackerStyle('LoLoLoL'),
                      'lo_lo_lol')
 
-  def test_MangleEnum(self):
-    self.assertEqual(client_api_generator.MangleEnum('FOO'), 'FOO')
-    self.assertEqual(client_api_generator.MangleEnum('NULL'), 'NONE')
+  def test_SanitizeLiteralEnum(self):
+    self.assertEqual(client_api_generator.SanitizeLiteral('foo'), 'foo')
+    self.assertEqual(client_api_generator.SanitizeLiteral('null'), 'none')
+    self.assertEqual(client_api_generator.SanitizeLiteral('Infinity'),
+                                                          'InfinityValue')
 
   def test_PatchFullQualifiedRefs(self):
     json_api = {
       'domains': [
         {
           'domain': 'domain0',
           '$ref': 'reference',
         },
         {
           'domain': 'domain1',
@@ skipping 474 matching lines @@
 
 
 if __name__ == '__main__':
   cmdline_parser = argparse.ArgumentParser()
   cmdline_parser.add_argument('--stamp')
   args = cmdline_parser.parse_args()
   unittest.main(verbosity=2, exit=False, argv=sys.argv[:1])
   if args.stamp:
     with open(args.stamp, 'a') as f:
       pass