mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
73 lines
1.9 KiB
Python
73 lines
1.9 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 dvc.path_info import WebDAVURLInfo
|
|
from tests.utils.httpd import run_server_on_thread
|
|
|
|
from .base import Base
|
|
|
|
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 webdav3.client import Client
|
|
|
|
user, secrets = first(AUTH.items())
|
|
return Client(
|
|
{
|
|
"webdav_hostname": self.replace(path="").url,
|
|
"webdav_login": user,
|
|
"webdav_password": 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]:
|
|
self.client.mkdir(d.path) # pylint: disable=no-member
|
|
|
|
def write_bytes(self, contents):
|
|
self.client.upload_to(contents, self.path)
|
|
|
|
|
|
@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
|