def test_upload_should_return_401_if_checksum_is_wrong(self):
invalid_values_with_expected_error_code = {
b'sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89':
ErrorCode.MESSAGE_FILES_CHECKSUM_WRONG_TYPE,
'': ErrorCode.MESSAGE_FILES_CHECKSUM_EMPTY,
'95a0f391c7ad86686ab1366bcd519ba5ab3cce89':
ErrorCode.MESSAGE_FILES_CHECKSUM_WRONG_FORMAT,
':95a0f391c7ad86686ab1366bcd519ba5ab3cce89':
ErrorCode.MESSAGE_FILES_CHECKSUM_INVALID_ALGORITHM,
'sha1:95a0f391c7ad86686ab1366bcd519ba5amONj':
ErrorCode.MESSAGE_FILES_CHECKSUM_INVALID_SHA1_HASH,
'sha1:': ErrorCode.MESSAGE_FILES_CHECKSUM_INVALID_SHA1_HASH,
}
for invalid_value, error_code in invalid_values_with_expected_error_code.items(
):
file1 = FileTransferToken.FileInfo(
path=
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend',
checksum='sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89',
size=1024,
)
file2 = FileTransferToken.FileInfo(
path=
'blenderrr/benchmark/test_task/scene-Helicopter-27-cycles.blend',
checksum=invalid_value,
size=1024,
)
self.upload_token.files = [file1, file2]
self.upload_token.sig = None
golem_upload_token = dump(self.upload_token,
settings.CONCENT_PRIVATE_KEY,
settings.CONCENT_PUBLIC_KEY)
encoded_token = b64encode(golem_upload_token).decode()
response = self.client.post(
'{}{}'.format(
reverse('gatekeeper:upload'),
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend'
),
HTTP_AUTHORIZATION='Golem ' + encoded_token,
HTTP_CONCENT_AUTH=self.header_concent_auth,
content_type='application/x-www-form-urlencoded',
HTTP_X_Golem_Messages=settings.GOLEM_MESSAGES_VERSION,
)
self.assertIsInstance(response, JsonResponse)
self.assertEqual(response.status_code, 401)
self.assertIn('message', response.json().keys())
self.assertIn('error_code', response.json().keys())
self.assertEqual("application/json", response["Content-Type"])
self.assertEqual(response.json()["error_code"], error_code.value)
def test_upload_should_return_401_if_specific_file_info_data(self):
file = FileTransferToken.FileInfo(
path=get_storage_result_file_path(task_id=self._get_uuid(),
subtask_id=self._get_uuid()),
size=1,
checksum='sha1:356a192b7913b04c54574d18c28d46e6395428ab\n',
)
self.upload_token.files = [file]
golem_upload_token = dump(self.upload_token,
settings.CONCENT_PRIVATE_KEY,
settings.CONCENT_PUBLIC_KEY)
encoded_token = b64encode(golem_upload_token).decode()
response = self.client.post(
'{}{}'.format(
reverse('gatekeeper:upload'),
get_storage_result_file_path(
task_id=self._get_uuid(),
subtask_id=self._get_uuid(),
),
),
HTTP_AUTHORIZATION='Golem ' + encoded_token,
HTTP_CONCENT_AUTH=self.header_concent_auth,
content_type='application/x-www-form-urlencoded',
HTTP_X_Golem_Messages=settings.GOLEM_MESSAGES_VERSION,
)
self.assertIsInstance(response, JsonResponse)
self.assertEqual(response.status_code, 401)
self.assertIn('message', response.json().keys())
self.assertEqual("application/json", response["Content-Type"])
def test_download_should_return_401_if_size_of_file_is_wrong(self):
invalid_values_with_expected_error_code = {
None: ErrorCode.MESSAGE_FILES_SIZE_EMPTY,
'number': ErrorCode.MESSAGE_FILES_SIZE_WRONG_TYPE,
'-2': ErrorCode.MESSAGE_FILES_SIZE_NEGATIVE,
'1.0': ErrorCode.MESSAGE_FILES_SIZE_WRONG_TYPE,
}
for invalid_value, error_code in invalid_values_with_expected_error_code.items(
):
file1 = FileTransferToken.FileInfo(
path=
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend',
checksum='sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89',
size=1024,
)
file2 = FileTransferToken.FileInfo(
path=
'blenderrr/benchmark/test_task/scene-Helicopter-27-cycles.blend',
checksum='sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89',
size=invalid_value,
)
self.download_token.files = [file1, file2]
self.download_token.sig = None
golem_download_token = dump(self.download_token,
settings.CONCENT_PRIVATE_KEY,
settings.CONCENT_PUBLIC_KEY)
encoded_token = b64encode(golem_download_token).decode()
response = self.client.get(
'{}{}'.format(
reverse('gatekeeper:download'),
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend'
),
HTTP_AUTHORIZATION='Golem ' + encoded_token,
HTTP_CONCENT_AUTH=self.header_concent_auth,
)
self.assertIsInstance(response, JsonResponse)
self.assertEqual(response.status_code, 401)
self.assertIn('message', response.json().keys())
self.assertEqual("application/json", response["Content-Type"])
self.assertEqual(response.json()["error_code"], error_code.value)
def create_file_info(
file_path: str,
package_hash: str,
size: int,
category: FileTransferToken.FileInfo.Category,
) -> FileTransferToken.FileInfo:
assert isinstance(category, FileTransferToken.FileInfo.Category)
return FileTransferToken.FileInfo(path=file_path,
checksum=package_hash,
size=size,
category=category)
def test_upload_should_return_401_if_file_paths_are_not_unique(self):
file1 = FileTransferToken.FileInfo(
path='blender/benchmark/test_task/scene-Helicopter-27-cycles.blend',
checksum='sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89',
size=1024,
)
file2 = FileTransferToken.FileInfo(
path='blender/benchmark/test_task/scene-Helicopter-27-cycles.blend',
checksum='sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89',
size=1024,
)
self.upload_token.files = [file1, file2]
assert file1 == file2
golem_upload_token = dump(self.upload_token,
settings.CONCENT_PRIVATE_KEY,
settings.CONCENT_PUBLIC_KEY)
encoded_token = b64encode(golem_upload_token).decode()
response = self.client.post(
'{}{}'.format(
reverse('gatekeeper:upload'),
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend'
),
HTTP_AUTHORIZATION='Golem ' + encoded_token,
HTTP_CONCENT_AUTH=self.header_concent_auth,
content_type='application/x-www-form-urlencoded',
HTTP_X_Golem_Messages=settings.GOLEM_MESSAGES_VERSION,
)
self.assertIsInstance(response, JsonResponse)
self.assertEqual(response.status_code, 401)
self.assertIn('message', response.json().keys())
self.assertIn('error_code', response.json().keys())
self.assertEqual("application/json", response["Content-Type"])
self.assertEqual(response.json()["error_code"],
ErrorCode.MESSAGE_FILES_PATHS_NOT_UNIQUE.value)
def test_download_should_return_401_if_file_categories_are_not_unique_across_file_info_list(
self):
file_1 = FileTransferToken.FileInfo(
path='blender/benchmark/test_task/scene-Helicopter-27-cycles.blend',
checksum='sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89',
size=1024,
category=FileTransferToken.FileInfo.Category.results)
file_2 = FileTransferToken.FileInfo(
path='blender/benchmark/test_task/scene-Helicopter-28-cycles.blend',
checksum='sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89',
size=1024,
category=FileTransferToken.FileInfo.Category.results)
self.download_token.files = [file_1, file_2]
golem_download_token = dump(self.download_token,
settings.CONCENT_PRIVATE_KEY,
settings.CONCENT_PUBLIC_KEY)
encoded_token = b64encode(golem_download_token).decode()
response = self.client.get(
'{}{}'.format(
reverse('gatekeeper:download'),
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend'
),
HTTP_AUTHORIZATION='Golem ' + encoded_token,
HTTP_CONCENT_AUTH=self.header_concent_auth,
content_type='application/x-www-form-urlencoded',
)
self.assertIsInstance(response, JsonResponse)
self.assertEqual(response.status_code, 401)
self.assertIn('message', response.json().keys())
self.assertIn('error_code', response.json().keys())
self.assertEqual("application/json", response["Content-Type"])
self.assertEqual(response.json()["error_code"],
ErrorCode.MESSAGE_FILES_CATEGORY_NOT_UNIQUE.value)
def test_upload_should_return_401_if_newlines_in_message_path(self):
invalid_values_with_expected_error_code = {
'blen\nder/benchmark/test_task/scene-Helicopter-2\n7-cycles.blend':
ErrorCode.MESSAGE_FILES_PATH_NOT_LISTED_IN_FILES,
'\nblender/benchmark/test_task/scene-Helicopter-27-cycles.blend':
ErrorCode.MESSAGE_FILES_PATH_NOT_LISTED_IN_FILES,
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend\n':
ErrorCode.MESSAGE_FILES_PATH_NOT_LISTED_IN_FILES,
}
for value, error_code in invalid_values_with_expected_error_code.items(
):
file = FileTransferToken.FileInfo(
path=value,
checksum='sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89',
size=1024,
)
self.upload_token.files = [file]
self.upload_token.sig = None
golem_upload_token = dump(self.upload_token,
settings.CONCENT_PRIVATE_KEY,
settings.CONCENT_PUBLIC_KEY)
encoded_token = b64encode(golem_upload_token).decode()
response = self.client.post(
'{}{}'.format(
reverse('gatekeeper:upload'),
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend'
),
HTTP_AUTHORIZATION='Golem ' + encoded_token,
HTTP_CONCENT_AUTH=self.header_concent_auth,
content_type='application/x-www-form-urlencoded',
HTTP_X_Golem_Messages=settings.GOLEM_MESSAGES_VERSION,
)
self.assertIsInstance(response, JsonResponse)
self.assertEqual(response.status_code, 401)
self.assertIn('message', response.json().keys())
self.assertIn('error_code', response.json().keys())
self.assertEqual("application/json", response["Content-Type"])
self.assertEqual(response.json()["error_code"], error_code.value)
def test_download_should_return_400_if_newlines_in_message_checksum(self):
invalid_values_with_expected_error_code = {
's\nha1:95a0f391c7ad866\n86ab1366bcd519ba5ab3cce89':
ErrorCode.MESSAGE_FILES_CHECKSUM_INVALID_ALGORITHM,
'\nsha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89':
ErrorCode.MESSAGE_FILES_CHECKSUM_INVALID_ALGORITHM,
'sha1:95a0f391c7ad86686ab1366bcd519ba5ab3cce89\n':
ErrorCode.MESSAGE_FILES_CHECKSUM_INVALID_SHA1_HASH,
}
for value, error_code in invalid_values_with_expected_error_code.items(
):
file = FileTransferToken.FileInfo(
path=
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend',
checksum=value,
size=1024,
)
self.download_token.files = [file]
self.download_token.sig = None
golem_download_token = dump(self.download_token,
settings.CONCENT_PRIVATE_KEY,
settings.CONCENT_PUBLIC_KEY)
encoded_token = b64encode(golem_download_token).decode()
response = self.client.get(
'{}{}'.format(
reverse('gatekeeper:download'),
'blender/benchmark/test_task/scene-Helicopter-27-cycles.blend'
),
HTTP_AUTHORIZATION='Golem ' + encoded_token,
HTTP_CONCENT_AUTH=self.header_concent_auth,
)
self.assertIsInstance(response, JsonResponse)
self.assertEqual(response.status_code, 401)
self.assertIn('message', response.json().keys())
self.assertEqual("application/json", response["Content-Type"])
self.assertEqual(response.json()["error_code"], error_code.value)