diff options
Diffstat (limited to 'tools/generate_stubs/generate_stubs_unittest.py')
-rwxr-xr-x | tools/generate_stubs/generate_stubs_unittest.py | 18 |
1 files changed, 16 insertions, 2 deletions
diff --git a/tools/generate_stubs/generate_stubs_unittest.py b/tools/generate_stubs/generate_stubs_unittest.py index 10936f1..13a2bc1 100755 --- a/tools/generate_stubs/generate_stubs_unittest.py +++ b/tools/generate_stubs/generate_stubs_unittest.py @@ -1,5 +1,5 @@ #!/usr/bin/env python -# Copyright (c) 2011 The Chromium Authors. All rights reserved. +# Copyright (c) 2012 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. @@ -19,10 +19,11 @@ errors or positioning. Most of that should be caught really fast anyways during any attempt to use a badly behaving script. """ +import generate_stubs as gs import re import StringIO +import sys import unittest -import generate_stubs as gs def _MakeSignature(return_type, name, params): @@ -141,6 +142,19 @@ EXPORTS self.assertTrue(re.search(pattern, contents), msg='Expected match of "%s" in %s' % (pattern, contents)) + def testQuietRun(self): + output = StringIO.StringIO() + gs.QuietRun([sys.executable, + '-c', 'print "line 1 and suffix\\nline 2"'], + write_to=output) + self.assertEqual('line 1 and suffix\nline 2\n', output.getvalue()) + + output = StringIO.StringIO() + gs.QuietRun([sys.executable, + '-c', 'print "line 1 and suffix\\nline 2"'], + filter='line 1', write_to=output) + self.assertEqual('line 2\n', output.getvalue()) + class PosixStubWriterUnittest(unittest.TestCase): def setUp(self): |