paperwork: Improve performance of random_certifications testing strategy
Some checks failed
Ruff / ruff (push) Successful in 22s
Test / test (push) Failing after 3m56s

This commit is contained in:
Adam Goldsmith 2024-02-17 15:55:56 -05:00
parent 6cf520fdf9
commit 8d0730bf70

View File

@ -1,3 +1,5 @@
from itertools import chain
from django.contrib.auth import get_user_model from django.contrib.auth import get_user_model
from django.contrib.auth.models import Permission from django.contrib.auth.models import Permission
from django.contrib.contenttypes.models import ContentType from django.contrib.contenttypes.models import ContentType
@ -78,29 +80,42 @@ class InstructorOrVendorReportTestCase(PermissionRequiredViewTestCaseMixin, Test
@st.composite @st.composite
def random_certifications(draw): def random_certifications(
departments = draw(st.lists(from_model(Department), min_size=1)) draw,
definitions = draw( ) -> list[Certification]:
st.lists( def certifications(version: CertificationVersion):
return st.lists(
from_model( from_model(
CertificationDefinition, department=st.sampled_from(departments) Certification,
number=st.none(),
certification_version=st.just(version),
), ),
min_size=1, max_size=10,
) )
def versions_with_certifications(definition: CertificationDefinition):
return st.lists(
from_model(CertificationVersion, definition=st.just(definition)).flatmap(
certifications
),
max_size=2,
) )
certification_versions = draw(
st.lists( def definitions_with_versions(department: Department):
from_model(CertificationVersion, definition=st.sampled_from(definitions)), return st.lists(
min_size=1, from_model(CertificationDefinition, department=st.just(department)).flatmap(
) versions_with_certifications
),
max_size=2,
) )
return draw( return draw(
st.lists( st.lists(
from_model( from_model(Department).flatmap(definitions_with_versions),
Certification, max_size=2,
number=st.none(), ).map(
certification_version=st.sampled_from(certification_versions), lambda x: list(
chain.from_iterable(chain.from_iterable(chain.from_iterable(x)))
) )
) )
) )
@ -111,7 +126,7 @@ class CertifiersReportTestCase(PermissionRequiredViewTestCaseMixin, TestCase):
path = "/paperwork/certifiers" path = "/paperwork/certifiers"
@given(certifications=random_certifications()) @given(certifications=random_certifications())
def test_certifers_report(self, certifications: list[Certification]) -> None: def test_certifiers_report(self, certifications: list[Certification]) -> None:
self.client.force_login(self.user_with_permission) self.client.force_login(self.user_with_permission)
response = self.client.get(self.path) response = self.client.get(self.path)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)