cookiecutter

1
"""
2
tests_output_folder.
3

4
Test formerly known from a unittest residing in test_generate.py named
5
TestOutputFolder.test_output_folder
6
"""
7

8
import os
9
from pathlib import Path
10

11
import pytest
12

13
from cookiecutter import exceptions, generate, utils
14

15

16
@pytest.fixture(scope='function')
17
def remove_output_folder():
18
    """Remove the output folder after test."""
19
    yield
20
    if os.path.exists('output_folder'):
21
        utils.rmtree('output_folder')
22

23

24
@pytest.mark.usefixtures('clean_system', 'remove_output_folder')
25
def test_output_folder() -> None:
26
    """Tests should correctly create content, as output_folder does not yet exist."""
27
    context = generate.generate_context(
28
        context_file='tests/test-output-folder/cookiecutter.json'
29
    )
30
    generate.generate_files(context=context, repo_dir='tests/test-output-folder')
31

32
    something = """Hi!
33
My name is Audrey Greenfeld.
34
It is 2014.
35
"""
36
    something2 = Path('output_folder/something.txt').read_text()
37
    assert something == something2
38

39
    in_folder = "The color is green and the letter is D.\n"
40
    in_folder2 = Path('output_folder/folder/in_folder.txt').read_text()
41
    assert in_folder == in_folder2
42

43
    assert os.path.isdir('output_folder/im_a.dir')
44
    assert os.path.isfile('output_folder/im_a.dir/im_a.file.py')
45

46

47
@pytest.mark.usefixtures('clean_system', 'remove_output_folder')
48
def test_exception_when_output_folder_exists() -> None:
49
    """Tests should raise error as output folder created before `generate_files`."""
50
    context = generate.generate_context(
51
        context_file='tests/test-output-folder/cookiecutter.json'
52
    )
53
    output_folder = context['cookiecutter']['test_name']
54

55
    if not os.path.exists(output_folder):
56
        os.makedirs(output_folder)
57
    with pytest.raises(exceptions.OutputDirExistsException):
58
        generate.generate_files(context=context, repo_dir='tests/test-output-folder')
59