import unittest
import time
import os
import tempfile


class MainTest(unittest.TestCase):
    def setUp(self):
        self.tic = time.time()  # todo this is sooo ugly

        self.test_dir = tempfile.mkdtemp()

    def tearDown(self):
        self.toc = time.time()
        t = self.toc - self.tic
        print('Time: %.3f' % (t))

    def test_copy_from_zip(self):
        import preparepdf

        expected_files = [
            '123456_F.pdf',
            '123457_O.pdf',
            '125412_T.pdf']

        # Prepare parameter
        in_zip = './submissions.zip'
        sheet_csv = "./Bewertungen.csv"

        out_dir = os.path.join(self.test_dir, 'out')
        os.mkdir(out_dir)

        tmp_dir = os.path.join(self.test_dir, 'tmp')
        os.mkdir(tmp_dir)

        # Call function
        preparepdf.main(["-i", in_zip, "-o", out_dir, "-c", sheet_csv])

        # Assert output
        created_files = os.listdir(out_dir)
        created_files.sort()
        self.assertEqual(expected_files, created_files)