Oprava testdat KorekturovanePDF

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))