mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-09 23:10:24 +01:00
test: remove nested contextmanagers, return False on exists()
This commit is contained in:
@@ -173,6 +173,8 @@ class RemoteGS(RemoteBASE):
|
||||
if path_info.path == fname:
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
def _upload(self, from_file, to_info, name=None, no_progress_bar=True):
|
||||
bucket = self.gs.bucket(to_info.bucket)
|
||||
_upload_to_bucket(
|
||||
|
||||
+18
-30
@@ -235,26 +235,20 @@ class S3Mocked:
|
||||
get_url = staticmethod(get_aws_url)
|
||||
|
||||
@classmethod
|
||||
@contextmanager
|
||||
def remote(cls):
|
||||
@contextmanager
|
||||
def inner():
|
||||
with mock_s3():
|
||||
remote = RemoteS3(None, {"url": cls.get_url()})
|
||||
yield remote
|
||||
|
||||
return inner()
|
||||
with mock_s3():
|
||||
remote = RemoteS3(None, {"url": cls.get_url()})
|
||||
yield remote
|
||||
|
||||
@classmethod
|
||||
@contextmanager
|
||||
def put_objects(cls, remote, objects):
|
||||
@contextmanager
|
||||
def inner():
|
||||
s3 = cls.get_client(remote)
|
||||
s3.create_bucket(Bucket="dvc-test")
|
||||
for key, body in objects.items():
|
||||
cls.put_object(remote, key, body)
|
||||
yield
|
||||
|
||||
return inner()
|
||||
s3 = cls.get_client(remote)
|
||||
s3.create_bucket(Bucket="dvc-test")
|
||||
for key, body in objects.items():
|
||||
cls.put_object(remote, key, body)
|
||||
yield
|
||||
|
||||
@staticmethod
|
||||
def get_client(remote):
|
||||
@@ -275,24 +269,18 @@ class GCP:
|
||||
get_url = staticmethod(get_gcp_url)
|
||||
|
||||
@classmethod
|
||||
@contextmanager
|
||||
def remote(cls):
|
||||
@contextmanager
|
||||
def inner():
|
||||
remote = RemoteGS(None, {"url": cls.get_url()})
|
||||
yield remote
|
||||
|
||||
return inner()
|
||||
remote = RemoteGS(None, {"url": cls.get_url()})
|
||||
yield remote
|
||||
|
||||
@classmethod
|
||||
@contextmanager
|
||||
def put_objects(cls, remote, objects):
|
||||
@contextmanager
|
||||
def inner():
|
||||
for key, body in objects.items():
|
||||
cls.put_object(remote, key, body)
|
||||
yield
|
||||
cls.remove(remote, objects.keys())
|
||||
|
||||
return inner()
|
||||
for key, body in objects.items():
|
||||
cls.put_object(remote, key, body)
|
||||
yield
|
||||
cls.remove(remote, objects.keys())
|
||||
|
||||
@classmethod
|
||||
def put_object(cls, remote, key, body):
|
||||
|
||||
Reference in New Issue
Block a user