from tempfile import NamedTemporaryFile

from helpers import absolute_sample_path
from tools import dumppdf


def run(filename, options=None):
    absolute_path = absolute_sample_path(filename)
    with NamedTemporaryFile() as output_file:
        if options:
            s = 'dumppdf -o%s %s %s' % (output_file.name, options, absolute_path)
        else:
            s = 'dumppdf -o%s %s' % (output_file.name, absolute_path)
        dumppdf.main(s.split(' '))


class TestDumpPDF():
    def test_1(self):
        run('jo.pdf', '-t -a')
        run('simple1.pdf', '-t -a')
        run('simple2.pdf', '-t -a')
        run('simple3.pdf', '-t -a')

    def test_2(self):
        run('nonfree/dmca.pdf', '-t -a')

    def test_3(self):
        run('nonfree/f1040nr.pdf')

    def test_4(self):
        run('nonfree/i1040nr.pdf')

    def test_5(self):
        run('nonfree/kampo.pdf', '-t -a')

    def test_6(self):
        run('nonfree/naacl06-shinyama.pdf', '-t -a')