mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-07 07:18:27 +01:00
Batuhan Taskaya
9aec8f1926
Intermediate PR in removing path_info from our code base. It is completely removed from the main codebase and is still being used in tests, which will be migrated in the followups. For fs-specific simple path manipulation, we've introduced `fs.path` helper (similar to `os.path`). Fixes #6774
77 lines
2.0 KiB
Python
77 lines
2.0 KiB
Python
# pylint:disable=abstract-method
|
|
import os
|
|
from wsgiref.simple_server import make_server
|
|
|
|
import pytest
|
|
from funcy import cached_property, first
|
|
from wsgidav.wsgidav_app import WsgiDAVApp
|
|
|
|
from tests.utils.httpd import run_server_on_thread
|
|
|
|
from .base import Base
|
|
from .path_info import WebDAVURLInfo
|
|
|
|
AUTH = {"user1": {"password": "password1"}}
|
|
|
|
|
|
class Webdav(Base, WebDAVURLInfo):
|
|
@staticmethod
|
|
def get_url(port): # pylint: disable=arguments-differ
|
|
return f"webdav://localhost:{port}"
|
|
|
|
@cached_property
|
|
def client(self):
|
|
from webdav4.client import Client
|
|
|
|
user, secrets = first(AUTH.items())
|
|
return Client(
|
|
self.replace(path="").url, auth=(user, secrets["password"])
|
|
)
|
|
|
|
def mkdir(self, mode=0o777, parents=False, exist_ok=False):
|
|
assert mode == 0o777
|
|
parent_dirs = list(reversed(self.parents))[1:] if parents else []
|
|
for d in parent_dirs + [self]:
|
|
path = d.path # pylint: disable=no-member
|
|
if not self.client.exists(path):
|
|
self.client.mkdir(path)
|
|
|
|
def write_bytes(self, contents):
|
|
from io import BytesIO
|
|
|
|
self.client.upload_fileobj(BytesIO(contents), self.path)
|
|
|
|
@property
|
|
def fs_path(self):
|
|
return self.path.lstrip("/")
|
|
|
|
|
|
@pytest.fixture
|
|
def webdav_server(test_config, tmp_path_factory):
|
|
test_config.requires("webdav")
|
|
|
|
host, port = "localhost", 0
|
|
directory = os.fspath(tmp_path_factory.mktemp("http"))
|
|
dirmap = {"/": directory}
|
|
|
|
app = WsgiDAVApp(
|
|
{
|
|
"host": host,
|
|
"port": port,
|
|
"provider_mapping": dirmap,
|
|
"simple_dc": {"user_mapping": {"*": AUTH}},
|
|
}
|
|
)
|
|
server = make_server(host, port, app)
|
|
with run_server_on_thread(server) as httpd:
|
|
yield httpd
|
|
|
|
|
|
@pytest.fixture
|
|
def webdav(webdav_server):
|
|
url = Webdav.get_url(webdav_server.server_port)
|
|
ret = Webdav(url)
|
|
user, secrets = first(AUTH.items())
|
|
ret.config = {"url": url, "user": user, **secrets}
|
|
yield ret
|