def test_workstation_update_view(client):
w = WorkstationFactory()
user = UserFactory()
w.add_editor(user=user)
title = "my Workstation"
description = "my AWESOME workstation"
assert w.title != title
assert w.description is None
response = get_view_for_user(
client=client,
method=client.post,
viewname="workstations:update",
reverse_kwargs={"slug": w.slug},
user=user,
data={
"title": title,
"description": description
},
)
w.refresh_from_db()
assert response.status_code == 302
assert w.title == title
assert w.description == description
def test_workstationimage_create(client):
u2 = UserFactory()
VerificationFactory(user=u2, is_verified=True)
w1 = WorkstationFactory()
w2 = WorkstationFactory()
w2.add_editor(user=u2)
user_upload = UserUploadFactory(filename="test_image.tar.gz", creator=u2)
user_upload.status = user_upload.StatusChoices.COMPLETED
user_upload.save()
assert w1.workstationimage_set.count() == 0
assert w2.workstationimage_set.count() == 0
response = get_view_for_user(
client=client,
method=client.post,
viewname="workstations:image-create",
reverse_kwargs={"slug": w2.slug},
user=u2,
data={
"user_upload": user_upload.pk,
"creator": u2.pk,
"workstation": w2.pk,
"initial_path": "a",
"websocket_port": 1337,
"http_port": 1234,
},
)
assert response.status_code == 302
w1.refresh_from_db()
w2.refresh_from_db()
assert w1.workstationimage_set.count() == 0
w2_images = w2.workstationimage_set.all()
assert len(w2_images) == 1
assert w2_images[0].creator == u2
assert w2_images[0].websocket_port == 1337
assert w2_images[0].http_port == 1234
assert w2_images[0].user_upload == user_upload
assert w2_images[0].initial_path == "a"
def test_group_deletion_reverse(group):
ws = WorkstationFactory()
users_group = ws.users_group
editors_group = ws.editors_group
assert users_group
assert editors_group
getattr(ws, group).delete()
with pytest.raises(ObjectDoesNotExist):
users_group.refresh_from_db()
with pytest.raises(ObjectDoesNotExist):
editors_group.refresh_from_db()
with pytest.raises(ObjectDoesNotExist):
ws.refresh_from_db()
def test_workstationimage_create(client):
u2 = UserFactory()
w1 = WorkstationFactory()
w2 = WorkstationFactory()
w2.add_editor(user=u2)
staged_file = StagedFileFactory(file__filename="example.tar.gz")
assert w1.workstationimage_set.count() == 0
assert w2.workstationimage_set.count() == 0
response = get_view_for_user(
client=client,
method=client.post,
viewname="workstations:image-create",
reverse_kwargs={"slug": w2.slug},
user=u2,
data={
"chunked_upload": staged_file.file_id,
"initial_path": "a",
"websocket_port": 1337,
"http_port": 1234,
},
)
assert response.status_code == 302
w1.refresh_from_db()
w2.refresh_from_db()
assert w1.workstationimage_set.count() == 0
w2_images = w2.workstationimage_set.all()
assert len(w2_images) == 1
assert w2_images[0].creator == u2
assert w2_images[0].websocket_port == 1337
assert w2_images[0].http_port == 1234
assert w2_images[0].staged_image_uuid == staged_file.file_id
assert w2_images[0].initial_path == "a"
def test_workstationimage_create(client):
UserFactory()
u2 = UserFactory(is_staff=True)
w1 = WorkstationFactory()
w2 = WorkstationFactory()
staged_file = StagedFileFactory(file__filename="example.tar.gz")
assert w1.workstationimage_set.count() == 0
assert w2.workstationimage_set.count() == 0
response = get_view_for_user(
client=client,
method=client.post,
viewname="workstations:image-create",
reverse_kwargs={"slug": w2.slug},
user=u2,
data={
"chunked_upload": staged_file.file_id,
"initial_path": "",
"websocket_port": 1337,
"http_port": 1234,
},
)
assert response.status_code == 302
w1.refresh_from_db()
w2.refresh_from_db()
assert w1.workstationimage_set.count() == 0
w2_images = w2.workstationimage_set.all()
assert len(w2_images) == 1
assert w2_images[0].creator == u2
assert w2_images[0].websocket_port == 1337
assert w2_images[0].http_port == 1234
assert w2_images[0].staged_image_uuid == staged_file.file_id
assert w2_images[0].initial_path == ""
def test_workstation_update_view(client):
w = WorkstationFactory()
user = UserFactory(is_staff=True)
title = "my Workstation"
description = "my AWESOME workstation"
assert w.title != title
assert w.description is None
response = get_view_for_user(
client=client,
method=client.post,
viewname="workstations:update",
reverse_kwargs={"slug": w.slug},
user=user,
data={"title": title, "description": description},
)
w.refresh_from_db()
assert response.status_code == 302
assert w.title == title
assert w.description == description