Coverage for tests/test_generation/utils.py: 100%

1from typing import Iterator 

2from pathlib import Path 

3 

4from prisma.generator.utils import resolve_template_path 

5 

6 

7def iter_templates_dir(path: Path) -> Iterator[Path]: 

8 templates = path / 'generator' / 'templates' 

9 assert templates.exists() 

10 

11 for template in templates.iterdir(): 

12 name = template.name 

13 

14 if template.is_dir() or not name.endswith('.py.jinja') or name.startswith('_'): 

15 continue 

16 

17 yield resolve_template_path(path, template.relative_to(templates)) 

18 

19 

20def assert_module_is_clean(path: Path) -> None: 

21 for template in iter_templates_dir(path): 

22 assert not template.exists() 

23 

24 

25def assert_module_not_clean(path: Path) -> None: 

26 for template in iter_templates_dir(path): 

27 assert template.exists()