summaryrefslogtreecommitdiffstats
path: root/tools/generate_stubs/generate_stubs_unittest.py
diff options
context:
space:
mode:
Diffstat (limited to 'tools/generate_stubs/generate_stubs_unittest.py')
-rwxr-xr-xtools/generate_stubs/generate_stubs_unittest.py18
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):