# -*- coding: utf-8 -*-
from os.path import dirname, join
from tempfile import TemporaryDirectory

from pelican.generators import ArticlesGenerator
from pelican.tests.support import get_settings, unittest

from .neighbors import neighbors


CUR_DIR = dirname(__file__)


class NeighborsTest(unittest.TestCase):
    def test_neighbors_basic(self):
        with TemporaryDirectory() as tmpdirname:
            generator = _build_article_generator(join(CUR_DIR, '..', 'test_data'), tmpdirname)
            neighbors(generator)
    def test_neighbors_with_single_article(self):
        with TemporaryDirectory() as tmpdirname:
            generator = _build_article_generator(join(CUR_DIR, 'test_data'), tmpdirname)
            neighbors(generator)


def _build_article_generator(content_path, output_path):
    settings = get_settings(filenames={})
    settings['PATH'] = content_path
    context = settings.copy()
    context['generated_content'] = dict()
    context['static_links'] = set()
    article_generator = ArticlesGenerator(
        context=context, settings=settings,
        path=settings['PATH'], theme=settings['THEME'], output_path=output_path)
    article_generator.generate_context()
    return article_generator