mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 15:28:34 +01:00
48 lines
981 B
Python
48 lines
981 B
Python
# pylint:disable=abstract-method
|
|
import os
|
|
import uuid
|
|
|
|
import pytest
|
|
|
|
from dvc.path_info import CloudURLInfo
|
|
from dvc.utils import env2bool
|
|
|
|
from .base import Base
|
|
|
|
TEST_OSS_REPO_BUCKET = "dvc-test"
|
|
|
|
|
|
class OSS(Base, CloudURLInfo):
|
|
@staticmethod
|
|
def should_test():
|
|
do_test = env2bool("DVC_TEST_OSS", undefined=None)
|
|
if do_test is not None:
|
|
return do_test
|
|
|
|
return (
|
|
os.getenv("OSS_ENDPOINT")
|
|
and os.getenv("OSS_ACCESS_KEY_ID")
|
|
and os.getenv("OSS_ACCESS_KEY_SECRET")
|
|
)
|
|
|
|
@staticmethod
|
|
def _get_storagepath():
|
|
return f"{TEST_OSS_REPO_BUCKET}/{uuid.uuid4()}"
|
|
|
|
@staticmethod
|
|
def get_url():
|
|
return f"oss://{OSS._get_storagepath()}"
|
|
|
|
|
|
@pytest.fixture
|
|
def oss():
|
|
if not OSS.should_test():
|
|
pytest.skip("no oss running")
|
|
yield OSS(OSS.get_url())
|
|
|
|
|
|
@pytest.fixture
|
|
def oss_remote(tmp_dir, dvc, oss):
|
|
tmp_dir.add_remote(config=oss.config)
|
|
yield oss
|