diff --git a/korektury/testutils.py b/korektury/testutils.py index f9bd009d..03c91ee7 100644 --- a/korektury/testutils.py +++ b/korektury/testutils.py @@ -35,23 +35,26 @@ def create_test_pdf(rnd, organizatori): # TODO silent ghostscript (vypisuje odstavec za každou stránku…) - KorekturovanePDF.objects.create( - nazev='B', komentar='Neuronové sítě', org=rnd.choice(organizatori), pdf=gen_filename(filename='B.pdf') + korekturovane_pdf = KorekturovanePDF.objects.create( + nazev='B', komentar='Neuronové sítě', pdf=gen_filename(filename='B.pdf') ) + korekturovane_pdf.orgove.set((rnd.choice(organizatori),)) - KorekturovanePDF.objects.create( - nazev='A', komentar='M&M: Jak řešit?', org=rnd.choice(organizatori), pdf=gen_filename(filename='A.pdf') + korekturovane_pdf = KorekturovanePDF.objects.create( + nazev='A', komentar='M&M: Jak řešit?', pdf=gen_filename(filename='A.pdf') ) + korekturovane_pdf.orgove.set(rnd.sample(organizatori, 2)) - korekturovane_pdf = KorekturovanePDF.objects.create( - nazev='A', komentar='M&M: Jak řešit?', org=rnd.choice(organizatori), pdf=gen_filename(filename='A.pdf'), + KorekturovanePDF.objects.create( + nazev='A', komentar='M&M: Jak řešit?', pdf=gen_filename(filename='A.pdf'), status='zanaseni' ) - KorekturovanePDF.objects.create( - nazev='A', komentar='M&M: Jak řešit?', org=rnd.choice(organizatori), pdf=gen_filename(filename='A.pdf'), + korekturovane_pdf = KorekturovanePDF.objects.create( + nazev='A', komentar='M&M: Jak řešit?', pdf=gen_filename(filename='A.pdf'), status='zastarale' ) + korekturovane_pdf.orgove.set((rnd.choice(organizatori),)) except OSError as e: # TODO najít správné chyby, které vyhazují různé systémy při neexistenci ImageMagick, nebo knihoven logger.error(str(e))