mirror of
https://github.com/venthur/blag.git
synced 2025-11-25 20:52:43 +00:00
90 lines
2.2 KiB
Python
90 lines
2.2 KiB
Python
# remove when we don't support py38 anymore
|
|
from __future__ import annotations
|
|
from argparse import Namespace
|
|
from typing import Iterator, Callable
|
|
from tempfile import TemporaryDirectory
|
|
import os
|
|
|
|
import pytest
|
|
from jinja2 import Environment, Template
|
|
|
|
from blag import blag, quickstart
|
|
|
|
|
|
@pytest.fixture
|
|
def environment(cleandir: str) -> Iterator[Environment]:
|
|
site = {
|
|
'base_url': 'site base_url',
|
|
'title': 'site title',
|
|
'description': 'site description',
|
|
'author': 'site author',
|
|
}
|
|
env = blag.environment_factory('templates', globals_=dict(site=site))
|
|
yield env
|
|
|
|
|
|
@pytest.fixture
|
|
def page_template(environment: Environment) -> Iterator[Template]:
|
|
yield environment.get_template('page.html')
|
|
|
|
|
|
@pytest.fixture
|
|
def article_template(environment: Environment) -> Iterator[Template]:
|
|
yield environment.get_template('article.html')
|
|
|
|
|
|
@pytest.fixture
|
|
def index_template(environment: Environment) -> Iterator[Template]:
|
|
yield environment.get_template('index.html')
|
|
|
|
|
|
@pytest.fixture
|
|
def archive_template(environment: Environment) -> Iterator[Template]:
|
|
yield environment.get_template('archive.html')
|
|
|
|
|
|
@pytest.fixture
|
|
def tags_template(environment: Environment) -> Iterator[Template]:
|
|
yield environment.get_template('tags.html')
|
|
|
|
|
|
@pytest.fixture
|
|
def tag_template(environment: Environment) -> Iterator[Template]:
|
|
yield environment.get_template('tag.html')
|
|
|
|
|
|
@pytest.fixture
|
|
def cleandir() -> Iterator[str]:
|
|
"""Create a temporary working directory and cwd."""
|
|
config = """
|
|
[main]
|
|
base_url = https://example.com/
|
|
title = title
|
|
description = description
|
|
author = a. u. thor
|
|
"""
|
|
|
|
with TemporaryDirectory() as dir:
|
|
os.mkdir(f'{dir}/build')
|
|
with open(f'{dir}/config.ini', 'w') as fh:
|
|
fh.write(config)
|
|
# change directory
|
|
old_cwd = os.getcwd()
|
|
os.chdir(dir)
|
|
quickstart.copy_default_theme()
|
|
yield dir
|
|
# and change back afterwards
|
|
os.chdir(old_cwd)
|
|
|
|
|
|
@pytest.fixture
|
|
def args(cleandir: Callable[[], Iterator[str]]) -> Iterator[Namespace]:
|
|
|
|
args = Namespace(
|
|
input_dir='content',
|
|
output_dir='build',
|
|
static_dir='static',
|
|
template_dir='templates',
|
|
)
|
|
yield args
|