15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#!/usr/bin/env python
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# Copyright (c) 2012 The Chromium Authors. All rights reserved.
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# Use of this source code is governed by a BSD-style license that can be
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)# found in the LICENSE file.
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)"""Tests for count_ifdefs.
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)"""
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import os
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import unittest
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)import count_ifdefs
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class CountIfdefsTest(unittest.TestCase):
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  def setUp(self):
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    self.root = os.path.join(os.path.dirname(__file__), 'testdata')
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  def testNormal(self):
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    count = count_ifdefs.CountIfdefs('OS_[A-Z]+', self.root)
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    self.failUnless(count == 6)
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  def testSkipTests(self):
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    count = count_ifdefs.CountIfdefs('OS_[A-Z]+', self.root, True)
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    self.failUnless(count == 4)
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)if __name__ == '__main__':
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unittest.main()
31