diff options
Diffstat (limited to 'tests/test_pycco.py')
| -rw-r--r-- | tests/test_pycco.py | 28 |
1 files changed, 24 insertions, 4 deletions
diff --git a/tests/test_pycco.py b/tests/test_pycco.py index 53fa74f..845bfe8 100644 --- a/tests/test_pycco.py +++ b/tests/test_pycco.py @@ -1,16 +1,24 @@ from __future__ import absolute_import + import copy import os +import os.path import tempfile import time -import os.path + import pytest -from hypothesis import given, example -from hypothesis.strategies import lists, text, booleans, choices, none import pycco.generate_index as generate_index -from pycco.languages import supported_languages import pycco.main as p +from hypothesis import assume, example, given +from hypothesis.strategies import booleans, choices, lists, none, text +from pycco.languages import supported_languages + +try: + from unittest.mock import patch +except ImportError: + from mock import patch + PYTHON = supported_languages['.py'] @@ -164,6 +172,18 @@ def test_process(preserve_paths, index, choice): language=lang_name) +@patch('pygments.lexers.guess_lexer') +def test_process_skips_unknown_languages(mock_guess_lexer): + class Name: + name = 'this language does not exist' + mock_guess_lexer.return_value = Name() + + with pytest.raises(ValueError): + p.process(['LICENSE'], outdir=tempfile.gettempdir(), skip=False) + + p.process(['LICENSE'], outdir=tempfile.gettempdir(), skip=True) + + one_or_more_chars = text(min_size=1, max_size=255) paths = lists(one_or_more_chars, min_size=1, max_size=30) @given( |