def test_claim_view_post(self):
"""Test ClaimView"""
self.client.login(**test_auth())
response = self.client.post(
reverse("upload_claim", kwargs={"file_hash": self.upload.sha512}))
self.assertEqual(response.status_code, 302)
self.client.login(**test_auth(superuser=False))
response = self.client.post(
reverse("upload_claim", kwargs={"file_hash": self.upload.sha512}))
self.assertEqual(response.status_code, 302)
def test_legacy_upload(self):
"""Test legacy upload view"""
test_auth()
with open(settings.MEDIA_ROOT + "test3.txt") as _file:
response = self.client.post(
reverse("upload"),
data={
"id": "test",
"imagedata": _file,
"username": "******"
},
)
self.assertEqual(response.status_code, 200)
def test_windows(self):
"""Test windows download"""
self.client.login(**test_auth())
response = self.client.get(
reverse("download-client", kwargs={"client": "windows"}))
self.assertEqual(response["Content-Type"], "application/octet-stream")
self.assertEqual(response.status_code, 200)
def test_browser_upload(self):
"""Test browser upload"""
self.client.login(**test_auth())
with open(settings.MEDIA_ROOT + "test3.txt") as _file:
response = self.client.post(reverse("upload_browser"),
data={"test": _file})
self.assertEqual(response.status_code, 204)
def test_index_collection(self):
"""Test valid collection"""
self.client.login(**test_auth())
Collection.objects.create(name="test",
owner=get_user_model().objects.first())
response = self.client.get(reverse("index") + "?collection=test")
self.assertEqual(response.status_code, 200)
def test_macos(self):
"""Test macos download"""
self.client.login(**test_auth())
response = self.client.get(
reverse("download-client", kwargs={"client": "macos"}))
self.assertEqual(response["Content-Type"],
"application/x-apple-diskimage")
self.assertEqual(response.status_code, 200)
def test_upload_view_post(self):
"""Test ObjectView's post"""
auth = test_auth()
self.client.login(**auth)
test_user = get_user_model().objects.get(username=auth.get("username"))
test_collection = Collection.objects.create(name="test",
owner=test_user)
form_data = {"collection": test_collection}
response = self.client.post(
reverse("upload_view", kwargs={"file_hash": self.upload.sha512}),
data=form_data,
)
self.assertEqual(response.status_code, 302)
self.assertEqual(
response.url,
reverse("upload_view", kwargs={"file_hash": self.upload.sha512}),
)
def test_delete_view_invalid(self):
"""Test DeleteView's post but with invalid permissions"""
self.client.login(**test_auth(superuser=False))
response = self.client.post(
reverse("upload_delete", kwargs={"file_hash": self.upload.sha512}))
self.assertEqual(response.status_code, 302)
def test_delete_view_post(self):
"""Test DeleteView's post"""
self.client.login(**test_auth())
response = self.client.post(
reverse("upload_delete", kwargs={"file_hash": self.upload.sha512}))
self.assertEqual(response.status_code, 302)
def test_claim_view(self):
"""Test ClaimView"""
self.client.login(**test_auth())
response = self.client.get(
reverse("upload_claim", kwargs={"file_hash": self.upload.sha512}))
self.assertEqual(response.status_code, 200)
def test_sharex(self):
"""Test shareX"""
self.client.login(**test_auth())
response = self.client.get(
reverse("download-client", kwargs={"client": "sharex"}))
self.assertEqual(response.status_code, 200)
def test_index_collection_invalid(self):
"""Test invalid collection"""
self.client.login(**test_auth())
response = self.client.get(reverse("index") + "?collection=aa")
self.assertEqual(response.status_code, 404)
def test_index_paginator(self):
"""Test invalid paginator page"""
self.client.login(**test_auth())
response = self.client.get(reverse("index") + "?page=3")
self.assertEqual(response.status_code, 200)
def test_index_not_super(self):
"""Test default index page as a normal user"""
self.client.login(**test_auth(superuser=False))
response = self.client.get(reverse("index"))
self.assertEqual(response.status_code, 200)
def test_index(self):
"""Test default index page"""
self.client.login(**test_auth())
response = self.client.get(reverse("index"))
self.assertEqual(response.status_code, 200)
