[Vim/YCM] Identify ObjC and ObjC++ headers.

tools/vim/tests/chromium.ycm_extra_conf_unittest.py

 #!/usr/bin/env python
 
 # Copyright 2015 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.
 
 """Tests for chromium.ycm_extra_conf.
 
 These tests should be getting picked up by the PRESUBMIT.py in /tools/vim.
 Currently the tests only run on Linux and require 'ninja' to be available on
@@ skipping 42 matching lines @@
     statinfo = os.stat(path)
     os.chmod(path, statinfo.st_mode | stat.S_IXUSR)
 
 def GetLastLangFlag(flags):
   lastLang = None
   for i, flag in enumerate(flags):
       if flag =='-x':
         lastLang = flags[i+1]
   return lastLang
 
+def TestLanguage(test_file, language):
+  def test(self):
+    result = self.ycm_extra_conf.FlagsForFile(
+        os.path.join(self.chrome_root, test_file))
+    self.assertTrue(result)
+    self.assertEqual(GetLastLangFlag(result['flags']), language)
+  return test
+
 class Chromium_ycmExtraConfTest(unittest.TestCase):
 
   def SetUpFakeChromeTreeBelowPath(self):
     """Create fake Chromium source tree under self.test_root.
 
     The fake source tree has the following contents:
 
     <self.test_root>
       |  .gclient
       |
@@ skipping 152 matching lines @@
         '-std=c++11',
         '-x', 'c++',
         '-I[SRC]',
         '-Wno-unknown-warning-option',
         '-I[OUT]/a',
         '-isysroot',
         '/mac.sdk',
         '-I[OUT]/tag-default'
         ])
 
-  def testGetFlagsForFileForUnknownObjcFile(self):
-    result = self.ycm_extra_conf.FlagsForFile(
-        os.path.join(self.chrome_root, 'nonexistent.m'))
-    self.assertTrue(result)
-    self.assertEqual(GetLastLangFlag(result['flags']), 'objective-c')
-
-  def testGetFlagsForFileForUnknownObjcppFile(self):
-    result = self.ycm_extra_conf.FlagsForFile(
-        os.path.join(self.chrome_root, 'nonexistent.mm'))
-    self.assertTrue(result)
-    self.assertEqual(GetLastLangFlag(result['flags']), 'objective-c++')
+  testGetFlagsForFileForKnownObjcFile = TestLanguage(
+      'eight.m', 'objective-c')
+  testGetFlagsForFileForKnownObjcHeaderFile = TestLanguage(
+      'eight.h', 'objective-c')
+  testGetFlagsForFileForUnknownObjcFile = TestLanguage(
+      'nonexistent.m', 'objective-c')
+  testGetFlagsForFileForKnownObjcppFile = TestLanguage(
+      'nine.mm', 'objective-c++')
+  testGetFlagsForFileForKnownObjcppHeaderFile = TestLanguage(
+      'nine.h', 'objective-c++')
+  testGetFlagsForFileForUnknownObjcppFile = TestLanguage(
+      'nonexistent.mm', 'objective-c++')
 
   def testGetFlagsForFileForUnknownHeaderFile(self):
     result = self.ycm_extra_conf.FlagsForFile(
         os.path.join(self.chrome_root, 'nonexistent.h'))
     self.assertTrue(result)
     self.assertTrue('do_cache' in result)
     self.assertTrue(result['do_cache'])
     self.assertTrue('flags' in result)
     self.assertEquals(self.NormalizeStringsInList(result['flags']), [
         '-DUSE_CLANG_COMPLETER',
@@ skipping 148 matching lines @@
         '-Wno-unknown-warning-option',
         '-I[OUT]/a',
         '--sysroot=[SRC]/build/sysroot-image',
         ])
 
 if __name__ == '__main__':
   if not os.path.isfile('chromium.ycm_extra_conf.py'):
     print('The test must be run from src/tools/vim/ directory')
     sys.exit(1)
   unittest.main()