mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 15:28:34 +01:00
Merge pull request #1530 from efiop/master
remote: gs: don't forget to set projectname
This commit is contained in:
+1
-1
@@ -42,7 +42,7 @@ class RemoteGS(RemoteBase):
|
||||
|
||||
@property
|
||||
def gs(self):
|
||||
return storage.Client()
|
||||
return storage.Client(self.projectname)
|
||||
|
||||
def get_md5(self, bucket, path):
|
||||
import base64
|
||||
|
||||
@@ -0,0 +1,25 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user