Home | History | Annotate | Download | only in metrics 
      1 #!/usr/bin/env python
      2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
      3 # Use of this source code is governed by a BSD-style license that can be
      4 # found in the LICENSE file.
      5 
      6 """Tests for count_ifdefs.
      7 """
      8 
      9 import os
     10 import unittest
     11 
     12 import count_ifdefs
     13 
     14 
     15 class CountIfdefsTest(unittest.TestCase):
     16 
     17   def setUp(self):
     18     self.root = os.path.join(os.path.dirname(__file__), 'testdata')
     19 
     20   def testNormal(self):
     21     count = count_ifdefs.CountIfdefs('OS_[A-Z]+', self.root)
     22     self.failUnless(count == 6)
     23 
     24   def testSkipTests(self):
     25     count = count_ifdefs.CountIfdefs('OS_[A-Z]+', self.root, True)
     26     self.failUnless(count == 4)
     27 
     28 
     29 if __name__ == '__main__':
     30   unittest.main()
     31