from tests import BaseTestCase
from md_pdf.build.pandoc import output_html, build_document
from md_pdf.utils import remove_dir
import os


class OutputHTMLTestCase(BaseTestCase):
    output_dir = 'test-output'

    def setUp(self):
        super().setUp()
        os.makedirs(self.output_dir)

    def tearDown(self):
        super().tearDown()
        remove_dir(self.output_dir)

    def test_outputs_file(self):
        self.assertFalse(os.path.isfile(os.path.join(self.output_dir, 'output.html')))
        output_html('test', self.output_dir)
        self.assertTrue(os.path.isfile(os.path.join(self.output_dir, 'output.html')))

    def test_outputs_correct_data(self):
        output_html('test', self.output_dir)
        with open(os.path.join(self.output_dir, 'output.html')) as f:
            self.assertEqual(f.read(), 'test')


class BuildDocumentTestCase(BaseTestCase):
    def test_parses_markdown(self):
        doc = build_document('# test', None)
        self.assertIn('<h1 id="test">test</h1>', doc)

    def test_bibliography(self):
        bibliography = {
            'references': 'test-files/bib.yaml',
            'csl': 'chicago-author-date'
        }
        with open('test-files/2-pandoc.md') as f:
            test_content = f.read()
        doc = build_document(test_content, bibliography)
        self.assertIn(
            '<span class="citation">Doe (2005, 2006, 30; see also Doe and Roe 2007)</span> says blah.',
            doc
        )
        self.assertIn('Doe, John. 2005.', doc)
        self.assertIn('<div id="refs" class="references">', doc)