def test_annotationset_creation(client, settings):
# Override the celery settings
settings.task_eager_propagates = (True,)
settings.task_always_eager = (True,)
user = UserFactory(is_staff=True)
client.login(username=user.username, password=SUPER_SECURE_TEST_PASSWORD)
imageset = ChallengeFactory().imageset_set.get(phase=ImageSet.TRAINING)
url = reverse(
"datasets:annotationset-create",
kwargs={
"challenge_short_name": imageset.challenge.short_name,
"base_pk": imageset.pk,
},
)
url, kwargs = get_http_host(url=url, kwargs={})
response = client.get(url, **kwargs)
assert response.status_code == 200
response = client.post(
url, data={"kind": AnnotationSet.GROUNDTRUTH}, **kwargs
)
assert response.status_code == 302
annotationset = AnnotationSet.objects.get(
base=imageset, kind=AnnotationSet.GROUNDTRUTH
)
assert len(annotationset.images.all()) == 0
assert annotationset.creator == user
assert response.url == reverse(
"datasets:annotationset-add-images",
kwargs={
"challenge_short_name": annotationset.base.challenge.short_name,
"pk": annotationset.pk,
},
)
images = ["image10x10x10.zraw", "image10x10x10.mhd"]
session, uploaded_images = create_raw_upload_image_session(
images, annotationset=annotationset
)
url, kwargs = get_http_host(
url=annotationset.get_absolute_url(), kwargs={}
)
response = client.get(url, **kwargs)
assert "image10x10x10.mhd" in response.rendered_content
assert str(session.pk) in response.rendered_content
annotationset.refresh_from_db()
assert len(annotationset.images.all()) == 1
def test_annotationset_creation(client, settings):
# Override the celery settings
settings.task_eager_propagates = (True, )
settings.task_always_eager = (True, )
user = UserFactory(is_staff=True)
client.login(username=user.username, password=SUPER_SECURE_TEST_PASSWORD)
imageset = ChallengeFactory().imageset_set.get(phase=ImageSet.TRAINING)
url = reverse(
"datasets:annotationset-create",
kwargs={
"challenge_short_name": imageset.challenge.short_name,
"base_pk": imageset.pk,
},
)
url, kwargs = get_http_host(url=url, kwargs={})
response = client.get(url, **kwargs)
assert response.status_code == 200
response = client.post(url,
data={"kind": AnnotationSet.GROUNDTRUTH},
**kwargs)
assert response.status_code == 302
annotationset = AnnotationSet.objects.get(base=imageset,
kind=AnnotationSet.GROUNDTRUTH)
assert len(annotationset.images.all()) == 0
assert annotationset.creator == user
assert response.url == reverse(
"datasets:annotationset-add-images",
kwargs={
"challenge_short_name": annotationset.base.challenge.short_name,
"pk": annotationset.pk,
},
)
images = ["image10x10x10.zraw", "image10x10x10.mhd"]
session, uploaded_images = create_raw_upload_image_session(
images,
annotationset=annotationset,
linked_task=add_images_to_annotationset,
)
url, kwargs = get_http_host(url=annotationset.get_absolute_url(),
kwargs={})
response = client.get(url, **kwargs)
assert "image10x10x10.mhd" in response.rendered_content
assert str(session.pk) in response.rendered_content
annotationset.refresh_from_db()
assert len(annotationset.images.all()) == 1
def test_imageset_add_images(client, settings):
# Override the celery settings
settings.task_eager_propagates = (True,)
settings.task_always_eager = (True,)
user = UserFactory(is_staff=True)
client.login(username=user.username, password=SUPER_SECURE_TEST_PASSWORD)
challenge = ChallengeFactory()
imageset = ImageSet.objects.get(
challenge=challenge, phase=ImageSet.TRAINING
)
assert len(imageset.images.all()) == 0
images = ["image10x10x10.zraw", "image10x10x10.mhd"]
session, uploaded_images = create_raw_upload_image_session(
images, imageset=imageset
)
url, kwargs = get_http_host(url=imageset.get_absolute_url(), kwargs={})
response = client.get(url, **kwargs)
assert "image10x10x10.mhd" in response.rendered_content
assert str(session.pk) in response.rendered_content
imageset.refresh_from_db()
assert len(imageset.images.all()) == 1
def test_imageset_add_images(client, settings):
# Override the celery settings
settings.task_eager_propagates = (True, )
settings.task_always_eager = (True, )
user = UserFactory(is_staff=True)
client.login(username=user.username, password=SUPER_SECURE_TEST_PASSWORD)
challenge = ChallengeFactory()
imageset = ImageSet.objects.get(challenge=challenge,
phase=ImageSet.TRAINING)
assert len(imageset.images.all()) == 0
images = ["image10x10x10.zraw", "image10x10x10.mhd"]
session, uploaded_images = create_raw_upload_image_session(
images, imageset=imageset)
url, kwargs = get_http_host(url=imageset.get_absolute_url(), kwargs={})
response = client.get(url, **kwargs)
assert "image10x10x10.mhd" in response.rendered_content
assert str(session.pk) in response.rendered_content
imageset.refresh_from_db()
assert len(imageset.images.all()) == 1
def _view_url(self, user, url):
self._login(user)
url, kwargs = get_http_host(url=url, kwargs={})
response = self.client.get(url, **kwargs)
if user is None:
username = "******"
else:
username = user.username
return response, username
def test_autocomplete_num_queries(self, client, django_assert_num_queries):
archive = ArchiveFactory()
admin = UserFactory()
archive.add_editor(admin)
user = UserFactory()
url = reverse("users-autocomplete", kwargs={})
client.login(username=user.username,
password=SUPER_SECURE_TEST_PASSWORD)
method = client.get
url, kwargs = get_http_host(url=url,
kwargs={"data": {
"q": user.username
}})
try:
with django_assert_num_queries(20) as _:
method(url, **kwargs)
finally:
client.logout()