mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 15:28:27 +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.
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
# pylint:disable=abstract-method
|
|
import locale
|
|
import os
|
|
import uuid
|
|
|
|
import pytest
|
|
import requests
|
|
|
|
from dvc.path_info import HTTPURLInfo
|
|
|
|
from .base import Base
|
|
|
|
|
|
class HTTP(Base, HTTPURLInfo):
|
|
@staticmethod
|
|
def get_url(port): # pylint: disable=arguments-differ
|
|
dname = str(uuid.uuid4())
|
|
return f"http://127.0.0.1:{port}/{dname}"
|
|
|
|
def mkdir(self, mode=0o777, parents=False, exist_ok=False):
|
|
assert mode == 0o777
|
|
assert parents
|
|
assert not exist_ok
|
|
|
|
def write_bytes(self, contents):
|
|
assert isinstance(contents, bytes)
|
|
response = requests.post(self.url, data=contents)
|
|
assert response.status_code == 200
|
|
|
|
def write_text(self, contents, encoding=None, errors=None):
|
|
if not encoding:
|
|
encoding = locale.getpreferredencoding(False)
|
|
assert errors is None
|
|
self.write_bytes(contents.encode(encoding))
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def http_server(test_config, tmp_path_factory):
|
|
from tests.utils.httpd import StaticFileServer
|
|
|
|
test_config.requires("http")
|
|
directory = os.fspath(tmp_path_factory.mktemp("http"))
|
|
with StaticFileServer(directory=directory) as httpd:
|
|
yield httpd
|
|
|
|
|
|
@pytest.fixture
|
|
def http(http_server):
|
|
yield HTTP(HTTP.get_url(http_server.server_port))
|