def test_storage_client_creation(self, mock_client, mock_get_connetion,
mock_get_creds_and_project_id,
mock_client_info):
hook = gcs.GCSHook()
result = hook.get_conn()
# test that Storage Client is called with required arguments
mock_client.assert_called_once_with(client_info="CLIENT_INFO",
credentials="CREDENTIALS",
project="PROJECT_ID")
self.assertEqual(mock_client.return_value, result)
def setUp(self):
with mock.patch(BASE_STRING.format('CloudBaseHook.__init__')):
self.gcs_hook = gcs.GCSHook(google_cloud_storage_conn_id='test')
# generate a 384KiB test file (larger than the minimum 256KiB multipart chunk size)
self.testfile = tempfile.NamedTemporaryFile(delete=False)
self.testfile.write(b"x" * 393216)
self.testfile.flush()
self.testdata_bytes = b"x" * 393216
self.testdata_str = "x" * 393216
def setUp(self):
with mock.patch(GCS_STRING.format("CloudBaseHook.__init__"),
new=mock_base_gcp_hook_default_project_id):
self.gcs_hook = gcs.GCSHook(google_cloud_storage_conn_id="test")
def setUp(self):
with mock.patch(
GCS_STRING.format('GoogleBaseHook.__init__'),
new=mock_base_gcp_hook_default_project_id,
):
self.gcs_hook = gcs.GCSHook(gcp_conn_id='test')
