cookiecutter

1
"""Collection of tests around loading extensions."""
2

3
import pytest
4

5
from cookiecutter.environment import StrictEnvironment
6
from cookiecutter.exceptions import UnknownExtension
7

8

9
def test_env_should_raise_for_unknown_extension() -> None:
10
    """Test should raise if extension not installed in system."""
11
    context = {'cookiecutter': {'_extensions': ['foobar']}}
12

13
    with pytest.raises(UnknownExtension) as err:
14
        StrictEnvironment(context=context, keep_trailing_newline=True)
15

16
    assert 'Unable to load extension: ' in str(err.value)
17

18

19
def test_env_should_come_with_default_extensions() -> None:
20
    """Verify default extensions loaded with StrictEnvironment."""
21
    env = StrictEnvironment(keep_trailing_newline=True)
22
    assert 'cookiecutter.extensions.JsonifyExtension' in env.extensions
23
    assert 'cookiecutter.extensions.RandomStringExtension' in env.extensions
24
    assert 'cookiecutter.extensions.SlugifyExtension' in env.extensions
25
    assert 'cookiecutter.extensions.TimeExtension' in env.extensions
26
    assert 'cookiecutter.extensions.UUIDExtension' in env.extensions
27