mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Saugat Pachhai
5faae2a6e9
Some of the tests are run by default, other requires
`--enable-{remote}` to run. The remotes that are run by default
also have a `--disable-{remote}` flag. All of the remotes can be
enabled using `--all` flag.
To use this functionality in the fixture,
`test_config` fixture has to be used, and requires to call
`test_config.requires(remote)`.
In an individual test, you can simply use `pytest.mark.remote`
to make them skip or run.
63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
# pylint:disable=abstract-method
|
|
import os
|
|
from pathlib import Path
|
|
from wsgiref.simple_server import make_server
|
|
|
|
import pytest
|
|
from funcy import first
|
|
from wsgidav.wsgidav_app import WsgiDAVApp
|
|
|
|
from dvc.path_info import WebDAVURLInfo
|
|
from dvc.utils.fs import makedirs
|
|
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}"
|
|
|
|
def mkdir(self, mode=0o777, parents=False, exist_ok=False):
|
|
assert parents
|
|
makedirs(self.path, exist_ok=exist_ok)
|
|
|
|
def write_bytes(self, contents):
|
|
Path(self.path).write_bytes(contents)
|
|
|
|
def write_text(self, contents, encoding=None, errors=None):
|
|
Path(self.path).write_text(contents, encoding=encoding, errors=errors)
|
|
|
|
|
|
@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
|