def __init__(self,
project_id,
location,
body,
gcp_conn_id='google_cloud_default',
api_version='v1',
zip_path=None,
validate_body=True,
*args, **kwargs):
self.project_id = project_id
self.location = location
self.full_location = 'projects/{}/locations/{}'.format(self.project_id,
self.location)
self.body = body
self.gcp_conn_id = gcp_conn_id
self.api_version = api_version
self.zip_path = zip_path
self.zip_path_preprocessor = ZipPathPreprocessor(body, zip_path)
self._field_validator = None
if validate_body:
self._field_validator = GcpBodyFieldValidator(CLOUD_FUNCTION_VALIDATION,
api_version=api_version)
self._hook = GcfHook(gcp_conn_id=self.gcp_conn_id, api_version=self.api_version)
self._validate_inputs()
super(GcfFunctionDeployOperator, self).__init__(*args, **kwargs)
def __init__(self,
name,
gcp_conn_id='google_cloud_default',
api_version='v1',
*args, **kwargs):
self.name = name
self.gcp_conn_id = gcp_conn_id
self.api_version = api_version
self._validate_inputs()
self.hook = GcfHook(gcp_conn_id=self.gcp_conn_id, api_version=self.api_version)
super(GcfFunctionDeleteOperator, self).__init__(*args, **kwargs)
def setUp(self):
with mock.patch(
'airflow.contrib.hooks.gcp_api_base_hook.GoogleCloudBaseHook.__init__',
new=mock_base_gcp_hook_no_default_project_id):
self.gcf_function_hook_no_project_id = GcfHook(gcp_conn_id='test',
api_version='v1')
