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)