From a6220e8d50b12d61618f164a9582ac29b3222d02 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jon=C3=A1=C5=A1=20Havelka?= <jonas.havelka@volny.cz>
Date: Fri, 2 Aug 2024 20:05:19 +0200
Subject: [PATCH] =?UTF-8?q?Typov=C3=A9=20anotace=20a=20dal=C5=A1=C3=AD=20d?=
 =?UTF-8?q?etaily=20v=20generov=C3=A1n=C3=AD=20testdat=20k=20soustredeni?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 seminar/testutils.py     |  4 ++--
 soustredeni/testutils.py | 22 +++++++++++++++++++---
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/seminar/testutils.py b/seminar/testutils.py
index b3888fa9..2de60884 100644
--- a/seminar/testutils.py
+++ b/seminar/testutils.py
@@ -839,10 +839,10 @@ def create_test_data(size = 6, rnd = None):
 	gen_ulohy_k_tematum(rnd, rocniky, rocnik_cisla, rocnik_temata, organizatori, resitele)
 
 	#generování soustředění
-	soustredeni = gen_soustredeni(size, rnd, resitele, organizatori)
+	soustredeni = gen_soustredeni(size, resitele, organizatori, rnd=rnd)
 
 	#generování konfer
-	konfery = gen_konfery(size, rnd, organizatori, resitele, soustredeni)
+	konfery = gen_konfery(size, organizatori, resitele, soustredeni, rnd=rnd)
 
 	# vytvoreni pdf ke korekturam
 	create_test_pdf(rnd, organizatori)
diff --git a/soustredeni/testutils.py b/soustredeni/testutils.py
index 12a48378..51bbb135 100644
--- a/soustredeni/testutils.py
+++ b/soustredeni/testutils.py
@@ -1,16 +1,25 @@
 import logging
 import datetime
+import random
+from typing import Sequence
 
 import lorem
 
 from .models import Soustredeni, Konfera
 import seminar.models.tvorba as am
+import personalni.models as pm
 
 logger = logging.getLogger(__name__)
 
 
-def gen_soustredeni(size, rnd, resitele, organizatori):
-	logger.info('Generuji soustředění...')
+def gen_soustredeni(
+		size: int,
+		resitele: Sequence[pm.Resitel],
+		organizatori: Sequence[pm.Organizator],
+		rnd: random.Random = None,
+) -> Sequence[Soustredeni]:
+	logger.info('Generuji soustředění (size={})...')
+	rnd = rnd or random.Random(x=42)
 
 	soustredeni = []
 	for _ in range(1, 10):  # FIXME Tu range si změňte jak chcete, nevím, co přesně znamená size (asi Anet?)
@@ -35,8 +44,15 @@ def gen_soustredeni(size, rnd, resitele, organizatori):
 	return soustredeni
 
 
-def gen_konfery(size, rnd, organizatori, resitele, soustredeni):
+def gen_konfery(
+		size: int,
+		organizatori: Sequence[pm.Organizator],
+		resitele: Sequence[pm.Resitel],
+		soustredeni: Sequence[Soustredeni],
+		rnd: random.Random = None,
+) -> Sequence[Konfera]:
 	logger.info('Generuji konfery (size={})...'.format(size))
+	rnd = rnd or random.Random(x=42)
 
 	konfery = []
 	for _ in range(1, size):  # FIXME Tu range si změňte jak chcete, nevím, co přesně znamená size (asi Anet?)