mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-05 23:09:32 +01:00
26 lines
759 B
Python
26 lines
759 B
Python
import mock
|
|
from unittest import TestCase
|
|
|
|
from dvc.remote.gs import RemoteGS
|
|
|
|
|
|
class TestRemoteGS(TestCase):
|
|
BUCKET = "bucket"
|
|
PREFIX = "prefix"
|
|
URL = "gs://{}/{}".format(BUCKET, PREFIX)
|
|
PROJECT = "PROJECT"
|
|
CONFIG = {"projectname": PROJECT, "url": URL}
|
|
|
|
def test_init(self):
|
|
remote = RemoteGS(None, self.CONFIG)
|
|
self.assertEqual(remote.url, self.URL)
|
|
self.assertEqual(remote.prefix, self.PREFIX)
|
|
self.assertEqual(remote.bucket, self.BUCKET)
|
|
self.assertEqual(remote.projectname, self.PROJECT)
|
|
|
|
@mock.patch("google.cloud.storage.Client")
|
|
def test_gs(self, mock_client):
|
|
remote = RemoteGS(None, self.CONFIG)
|
|
remote.gs()
|
|
mock_client.assert_called_with(self.PROJECT)
|