mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
5f63b3a7bb
* chore: drop deprecated tables and remove unused code * isort imports * migration: check for table existence before drop
108 lines
3.5 KiB
Python
108 lines
3.5 KiB
Python
from io import BytesIO
|
|
from test.fixtures import *
|
|
|
|
import pytest
|
|
from mock import Mock
|
|
|
|
from data.userfiles import DelegateUserfiles, Userfiles
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"prefix,path,expected",
|
|
[
|
|
("test", "foo", "test/foo"),
|
|
("test", "bar", "test/bar"),
|
|
("test", "/bar", "test/bar"),
|
|
("test", "../foo", "test/foo"),
|
|
("test", "foo/bar/baz", "test/baz"),
|
|
("test", "foo/../baz", "test/baz"),
|
|
(None, "foo", "foo"),
|
|
(None, "foo/bar/baz", "baz"),
|
|
],
|
|
)
|
|
def test_filepath(prefix, path, expected):
|
|
userfiles = DelegateUserfiles(None, None, "local_us", prefix)
|
|
assert userfiles.get_file_id_path(path) == expected
|
|
|
|
|
|
def test_lookup_userfile(app, client):
|
|
uuid = "deadbeef-dead-beef-dead-beefdeadbeef"
|
|
bad_uuid = "deadduck-dead-duck-dead-duckdeadduck"
|
|
upper_uuid = "DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF"
|
|
|
|
def _stream_read_file(locations, path):
|
|
if path.find(uuid) > 0 or path.find(upper_uuid) > 0:
|
|
return BytesIO(b"hello world")
|
|
|
|
raise IOError("Not found!")
|
|
|
|
storage_mock = Mock()
|
|
storage_mock.stream_read_file = _stream_read_file
|
|
|
|
app.config["USERFILES_PATH"] = "foo"
|
|
Userfiles(
|
|
app, distributed_storage=storage_mock, path="mockuserfiles", handler_name="mockuserfiles"
|
|
)
|
|
|
|
rv = client.open("/mockuserfiles/" + uuid, method="GET")
|
|
assert rv.status_code == 200
|
|
|
|
rv = client.open("/mockuserfiles/" + upper_uuid, method="GET")
|
|
assert rv.status_code == 200
|
|
|
|
rv = client.open("/mockuserfiles/" + bad_uuid, method="GET")
|
|
assert rv.status_code == 404
|
|
|
|
# NOTE: With the changes to the path converter for the extended repository names regex,
|
|
# this route conflicts with one of the route using the <reponame> regex.
|
|
# i.e /mockuserfiles/foo/bar/baz would match another url route.
|
|
# In order to 404 on this path, The Userfiles' path would need the full
|
|
# /mockuserfiles/foo/bar as a path for the Flask url_rule.
|
|
rv = client.open("/mockuserfiles/foo/bar/baz", method="GET")
|
|
# assert rv.status_code == 404
|
|
assert rv.status_code == 308
|
|
|
|
|
|
def test_lookup_userfile_with_nested_path(app, client):
|
|
uuid = "deadbeef-dead-beef-dead-beefdeadbeef"
|
|
bad_uuid = "deadduck-dead-duck-dead-duckdeadduck"
|
|
upper_uuid = "DEADBEEF-DEAD-BEEF-DEAD-BEEFDEADBEEF"
|
|
|
|
def _stream_read_file(locations, path):
|
|
if path.find(uuid) > 0 or path.find(upper_uuid) > 0:
|
|
return BytesIO(b"hello world")
|
|
|
|
raise IOError("Not found!")
|
|
|
|
storage_mock = Mock()
|
|
storage_mock.stream_read_file = _stream_read_file
|
|
|
|
Userfiles(
|
|
app,
|
|
distributed_storage=storage_mock,
|
|
path="mockuserfiles/foo/bar",
|
|
handler_name="mockuserfiles2",
|
|
)
|
|
|
|
rv = client.open("/mockuserfiles/foo/bar/" + uuid, method="GET")
|
|
assert rv.status_code == 200
|
|
|
|
rv = client.open("/mockuserfiles/foo/bar/" + upper_uuid, method="GET")
|
|
assert rv.status_code == 200
|
|
|
|
rv = client.open("/mockuserfiles/foo/bar/" + bad_uuid, method="GET")
|
|
assert rv.status_code == 404
|
|
|
|
rv = client.open("/mockuserfiles/foo/bar/baz", method="GET")
|
|
assert rv.status_code == 404
|
|
|
|
# The follwing path should conflict with the reponame regex
|
|
rv = client.open("/mockuserfiles/" + uuid, method="GET")
|
|
assert rv.status_code == 308
|
|
|
|
rv = client.open("/mockuserfiles/" + upper_uuid, method="GET")
|
|
assert rv.status_code == 308
|
|
|
|
rv = client.open("/mockuserfiles/" + bad_uuid, method="GET")
|
|
assert rv.status_code == 308
|