Source

articledump / testing / test_load_rst.py

Full commit
from articledump.restructuredtext import load_rst, load_all
import datetime
import pprint

content = """\
Test something
================

:date: 2011-10-10
"""

def test_load(tmpdir):
    testfile = tmpdir/'test.rst'
    testfile.write(content)
    info = load_rst(testfile, root=tmpdir)
    assert info.title == 'Test something'
    assert info.slug == 'test-something'
    assert info.date == datetime.date(2011, 10, 10)

    testfile.write(content + ':slug: shortone\n')

    info = load_rst(testfile, root=tmpdir)
    assert info.title == 'Test something'
    assert info.slug == 'shortone'
    assert info.date == datetime.date(2011, 10, 10)
    pprint.pprint(info.build_html_parts())


def test_load_all(tmpdir):
    for name in ['ab', 'ac', 'ad']:
        tmpdir.join(name + '.rst').write('test')

    all_info = list(load_all(tmpdir))