1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
|
import unittest
import sys
from contextlib import contextmanager
from os import path
sys.path.append(path.abspath(path.join(path.dirname(__file__),
'..', '..', 'tools')))
try:
from StringIO import StringIO
except ImportError:
from io import StringIO
from checkimports import is_valid
@contextmanager
def captured_output():
tmp_out, tmp_err = StringIO(), StringIO()
old_out, old_err = sys.stdout, sys.stderr
try:
sys.stdout, sys.stderr = tmp_out, tmp_err
yield sys.stdout, sys.stderr
finally:
sys.stdout, sys.stderr = old_out, old_err
tmp_out.close()
tmp_err.close()
class CheckImportsTest(unittest.TestCase):
fixturesDir = path.join(path.dirname(__file__), '..', '..',
'test', 'fixtures', 'tools', 'checkimports')
def test_unused_and_unsorted(self):
with captured_output() as (out, err):
self.assertEqual(is_valid(path.join(self.fixturesDir, 'invalid.cc')),
False)
output = out.getvalue()
self.assertIn('does not use "Local"', output);
self.assertIn('using statements aren\'t sorted in', output);
self.assertIn('Line 1: Actual: v8::MaybeLocal, Expected: v8::Array',
output);
self.assertIn('Line 2: Actual: v8::Array, Expected: v8::Local',
output);
self.assertIn('Line 3: Actual: v8::Local, Expected: v8::MaybeLocal',
output);
def test_unused_complex(self):
with captured_output() as (out, err):
self.assertEqual(is_valid(path.join(self.fixturesDir, 'maybe.cc')),
False)
output = out.getvalue()
self.assertIn('does not use "Local"', output);
def test_unused_simple(self):
with captured_output() as (out, err):
self.assertEqual(is_valid(path.join(self.fixturesDir, 'unused.cc')),
False)
output = out.getvalue()
self.assertIn('does not use "Context"', output);
def test_unsorted(self):
with captured_output() as (out, err):
self.assertEqual(is_valid(path.join(self.fixturesDir, 'unsorted.cc')),
False)
output = out.getvalue()
self.assertIn('using statements aren\'t sorted in', output);
self.assertIn('Line 1: Actual: v8::MaybeLocal, Expected: v8::Array',
output);
self.assertIn('Line 2: Actual: v8::Array, Expected: v8::MaybeLocal',
output);
def test_valid(self):
with captured_output() as (out, err):
self.assertEqual(is_valid(path.join(self.fixturesDir, 'valid.cc')),
True)
output = out.getvalue()
self.assertEqual(output, '');
if __name__ == '__main__':
unittest.main()
|