Skip to content

Commit

Permalink
Merge pull request #7 from jdufresne/mock-argv
Browse files Browse the repository at this point in the history
Use mock to override sys.argv instead of re-implementing it.
  • Loading branch information
tk0miya committed May 15, 2016
2 parents 2b95668 + c087d52 commit 1af907b
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions run_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,25 +71,21 @@ def test_has_no_coding_headers(self):
self.assertEqual(ret[0][1], 0)
self.assertTrue(ret[0][2].startswith('C101 '))

@patch.object(sys, 'argv', [])
def test_default_encoding(self):
try:
_argv = sys.argv
sys.argv = []
get_style_guide(parse_argv=True) # parse arguments
self.assertEqual(CodingChecker.encodings, ['latin-1', 'utf-8'])
finally:
sys.argv = _argv
if hasattr(CodingChecker, 'encodings'):
del CodingChecker.encodings

@patch.object(sys, 'argv', ['', '--accept-encodings=utf-8,utf-16'])
def test_change_encoding(self):
try:
_argv = sys.argv
sys.argv = ['', '--accept-encodings=utf-8,utf-16']
get_style_guide(parse_argv=True) # parse arguments
self.assertEqual(CodingChecker.encodings, ['utf-8', 'utf-16'])
finally:
sys.argv = _argv
if hasattr(CodingChecker, 'encodings'):
del CodingChecker.encodings

Expand Down

0 comments on commit 1af907b

Please sign in to comment.