paperwork: Improve performance of random_certifications testing strategy
This commit is contained in:
parent
6cf520fdf9
commit
8d0730bf70
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user