diff --git a/codespell_lib/tests/test_basic.py b/codespell_lib/tests/test_basic.py index c0285412f9..38fdbe0b21 100644 --- a/codespell_lib/tests/test_basic.py +++ b/codespell_lib/tests/test_basic.py @@ -10,9 +10,20 @@ import tempfile import warnings -from nose.tools import assert_equal, assert_true +from nose.tools import with_setup, assert_equal, assert_true -from codespell_lib import main +import codespell_lib as cs + + +def reload_codespell_lib(): + try: + reload # Python 2.7 + except NameError: + try: + from importlib import reload # Python 3.4+ + except ImportError: + from imp import reload # Python 3.0 - 3.3 + reload(cs._codespell) def run_codespell(args=(), cwd=None): @@ -34,102 +45,102 @@ def test_command(): def test_basic(): """Test some basic functionality""" - assert_equal(main('_does_not_exist_'), 0) + assert_equal(cs.main('_does_not_exist_'), 0) with tempfile.NamedTemporaryFile('w') as f: with CaptureStdout() as sio: - assert_equal(main('-D', 'foo', f.name), 1) # missing dictionary + assert_equal(cs.main('-D', 'foo', f.name), 1) # missing dictionary assert_true('cannot find dictionary' in sio[1]) - assert_equal(main(f.name), 0) # empty file + assert_equal(cs.main(f.name), 0) # empty file f.write('this is a test file\n') f.flush() - assert_equal(main(f.name), 0) # good + assert_equal(cs.main(f.name), 0) # good f.write('abandonned\n') f.flush() - assert_equal(main(f.name), 1) # bad + assert_equal(cs.main(f.name), 1) # bad f.write('abandonned\n') f.flush() - assert_equal(main(f.name), 2) # worse + assert_equal(cs.main(f.name), 2) # worse with TemporaryDirectory() as d: with open(op.join(d, 'bad.txt'), 'w') as f: f.write('abandonned\nAbandonned\nABANDONNED\nAbAnDoNnEd') - assert_equal(main(d), 4) + assert_equal(cs.main(d), 4) with CaptureStdout() as sio: - assert_equal(main('-w', d), 0) + assert_equal(cs.main('-w', d), 0) assert_true('FIXED:' in sio[1]) with open(op.join(d, 'bad.txt')) as f: new_content = f.read() - assert_equal(main(d), 0) + assert_equal(cs.main(d), 0) assert_equal(new_content, 'abandoned\nAbandoned\nABANDONED\nabandoned') with open(op.join(d, 'bad.txt'), 'w') as f: f.write('abandonned abandonned\n') - assert_equal(main(d), 2) + assert_equal(cs.main(d), 2) with CaptureStdout() as sio: - assert_equal(main('-q', '16', '-w', d), 0) + assert_equal(cs.main('-q', '16', '-w', d), 0) assert_equal(sio[0], '') - assert_equal(main(d), 0) + assert_equal(cs.main(d), 0) # empty directory os.mkdir(op.join(d, 'test')) - assert_equal(main(d), 0) + assert_equal(cs.main(d), 0) # hidden file with open(op.join(d, 'test.txt'), 'w') as f: f.write('abandonned\n') - assert_equal(main(op.join(d, 'test.txt')), 1) + assert_equal(cs.main(op.join(d, 'test.txt')), 1) os.rename(op.join(d, 'test.txt'), op.join(d, '.test.txt')) - assert_equal(main(op.join(d, '.test.txt')), 0) + assert_equal(cs.main(op.join(d, '.test.txt')), 0) def test_interactivity(): """Test interaction""" with tempfile.NamedTemporaryFile('w') as f: - assert_equal(main(f.name), 0) # empty file + assert_equal(cs.main(f.name), 0) # empty file f.write('abandonned\n') f.flush() - assert_equal(main('-i', '-1', f.name), 1) # bad + assert_equal(cs.main('-i', '-1', f.name), 1) # bad with FakeStdin('y\n'): - assert_equal(main('-i', '3', f.name), 1) + assert_equal(cs.main('-i', '3', f.name), 1) with CaptureStdout() as sio: with FakeStdin('n\n'): - assert_equal(main('-w', '-i', '3', f.name), 0) + assert_equal(cs.main('-w', '-i', '3', f.name), 0) assert_true('==>' in sio[0]) with CaptureStdout(): with FakeStdin('x\ny\n'): - assert_equal(main('-w', '-i', '3', f.name), 0) - assert_equal(main(f.name), 0) + assert_equal(cs.main('-w', '-i', '3', f.name), 0) + assert_equal(cs.main(f.name), 0) with tempfile.NamedTemporaryFile('w') as f: f.write('abandonned\n') f.flush() - assert_equal(main(f.name), 1) + assert_equal(cs.main(f.name), 1) with CaptureStdout(): with FakeStdin(' '): # blank input -> Y - assert_equal(main('-w', '-i', '3', f.name), 0) - assert_equal(main(f.name), 0) + assert_equal(cs.main('-w', '-i', '3', f.name), 0) + assert_equal(cs.main(f.name), 0) # multiple options with tempfile.NamedTemporaryFile('w') as f: f.write('ackward\n') f.flush() - assert_equal(main(f.name), 1) + assert_equal(cs.main(f.name), 1) with CaptureStdout(): with FakeStdin(' \n'): # blank input -> nothing - assert_equal(main('-w', '-i', '3', f.name), 0) - assert_equal(main(f.name), 1) + assert_equal(cs.main('-w', '-i', '3', f.name), 0) + assert_equal(cs.main(f.name), 1) with CaptureStdout(): with FakeStdin('0\n'): # blank input -> nothing - assert_equal(main('-w', '-i', '3', f.name), 0) - assert_equal(main(f.name), 0) + assert_equal(cs.main('-w', '-i', '3', f.name), 0) + assert_equal(cs.main(f.name), 0) with open(f.name, 'r') as f_read: assert_equal(f_read.read(), 'awkward\n') f.seek(0) f.write('ackward\n') f.flush() - assert_equal(main(f.name), 1) + assert_equal(cs.main(f.name), 1) with CaptureStdout() as sio: with FakeStdin('x\n1\n'): # blank input -> nothing - assert_equal(main('-w', '-i', '3', f.name), 0) + assert_equal(cs.main('-w', '-i', '3', f.name), 0) assert_true('a valid option' in sio[0]) - assert_equal(main(f.name), 0) + assert_equal(cs.main(f.name), 0) with open(f.name, 'r') as f: assert_equal(f.read(), 'backward\n') @@ -138,24 +149,46 @@ def test_summary(): """Test summary functionality""" with tempfile.NamedTemporaryFile('w') as f: with CaptureStdout() as sio: - main(f.name) + cs.main(f.name) for ii in range(2): assert_equal(sio[ii], '') # no output with CaptureStdout() as sio: - main(f.name, '--summary') + cs.main(f.name, '--summary') assert_equal(sio[1], '') # stderr assert_true('SUMMARY' in sio[0]) assert_equal(len(sio[0].split('\n')), 5) # no output f.write('abandonned\nabandonned') f.flush() with CaptureStdout() as sio: - main(f.name, '--summary') + cs.main(f.name, '--summary') assert_equal(sio[1], '') # stderr assert_true('SUMMARY' in sio[0]) assert_equal(len(sio[0].split('\n')), 7) assert_true('abandonned' in sio[0].split()[-2]) +@with_setup(reload_codespell_lib, reload_codespell_lib) +def test_ignore_dictionary(): + """Test ignore dictionary functionality""" + with TemporaryDirectory() as d: + with open(op.join(d, 'bad.txt'), 'w') as f: + f.write('abandonned\nabondon\n') + with tempfile.NamedTemporaryFile('w') as f: + f.write('abandonned\n') + f.flush() + assert_equal(cs.main('-I', f.name, d), 1) + + +@with_setup(reload_codespell_lib, reload_codespell_lib) +def test_custom_regex(): + """Test custom word regex""" + with TemporaryDirectory() as d: + with open(op.join(d, 'bad.txt'), 'w') as f: + f.write('abandonned_abondon\n') + assert_equal(cs.main(d), 0) + assert_equal(cs.main('-r', "[a-z]+", d), 2) + + def test_exclude_file(): """Test exclude file functionality""" with TemporaryDirectory() as d: @@ -164,8 +197,8 @@ def test_exclude_file(): with tempfile.NamedTemporaryFile('w') as f: f.write('abandonned 1\n') f.flush() - assert_equal(main(d), 2) - assert_equal(main('-x', f.name, d), 1) + assert_equal(cs.main(d), 2) + assert_equal(cs.main('-x', f.name, d), 1) def test_encoding(): @@ -173,24 +206,24 @@ def test_encoding(): # Some simple Unicode things with tempfile.NamedTemporaryFile('wb') as f: # with CaptureStdout() as sio: - assert_equal(main(f.name), 0) + assert_equal(cs.main(f.name), 0) f.write(u'naïve\n'.encode('utf-8')) f.flush() - assert_equal(main(f.name), 0) - assert_equal(main('-e', f.name), 0) + assert_equal(cs.main(f.name), 0) + assert_equal(cs.main('-e', f.name), 0) f.write(u'naieve\n'.encode('utf-8')) f.flush() - assert_equal(main(f.name), 1) + assert_equal(cs.main(f.name), 1) # Binary file warning with tempfile.NamedTemporaryFile('wb') as f: - assert_equal(main(f.name), 0) + assert_equal(cs.main(f.name), 0) f.write(b'\x00\x00naiive\x00\x00') f.flush() with CaptureStdout() as sio: - assert_equal(main(f.name), 0) + assert_equal(cs.main(f.name), 0) assert_true('WARNING: Binary file' in sio[1]) with CaptureStdout() as sio: - assert_equal(main('-q', '2', f.name), 0) + assert_equal(cs.main('-q', '2', f.name), 0) assert_equal(sio[1], '') @@ -199,19 +232,19 @@ def test_ignore(): with TemporaryDirectory() as d: with open(op.join(d, 'good.txt'), 'w') as f: f.write('this file is okay') - assert_equal(main(d), 0) + assert_equal(cs.main(d), 0) with open(op.join(d, 'bad.txt'), 'w') as f: f.write('abandonned') - assert_equal(main(d), 1) - assert_equal(main('--skip=bad*', d), 0) - assert_equal(main('--skip=bad.txt', d), 0) + assert_equal(cs.main(d), 1) + assert_equal(cs.main('--skip=bad*', d), 0) + assert_equal(cs.main('--skip=bad.txt', d), 0) subdir = op.join(d, 'ignoredir') os.mkdir(subdir) with open(op.join(subdir, 'bad.txt'), 'w') as f: f.write('abandonned') - assert_equal(main(d), 2) - assert_equal(main('--skip=bad*', d), 0) - assert_equal(main('--skip=*ignoredir*', d), 1) + assert_equal(cs.main(d), 2) + assert_equal(cs.main('--skip=bad*', d), 0) + assert_equal(cs.main('--skip=*ignoredir*', d), 1) class TemporaryDirectory(object):