def test_should_return_int_when_set_via_env_var(self): instance = hook.GoogleBaseHook(gcp_conn_id="google_cloud_default") self.assertIsInstance(instance.num_retries, int)
def setUp(self): self.instance = hook.GoogleBaseHook()
def test_should_fallback_when_empty_string_in_env_var(self): instance = hook.GoogleBaseHook(gcp_conn_id="google_cloud_default") self.assertIsInstance(instance.num_retries, int) self.assertEqual(5, instance.num_retries)
def test_should_raise_when_invalid_value_via_env_var(self): instance = hook.GoogleBaseHook(gcp_conn_id="google_cloud_default") with self.assertRaisesRegex( AirflowException, re.escape("The num_retries field should be a integer.")): self.assertIsInstance(instance.num_retries, int)
def setUp(self): with mock.patch( MODULE_NAME + '.GoogleBaseHook.__init__', new=mock_base_gcp_hook_default_project_id, ): self.instance = hook.GoogleBaseHook(gcp_conn_id="google-cloud-default")
def setUp(self): with mock.patch( 'airflow.providers.google.common.hooks.base_google.GoogleBaseHook.__init__', new=mock_base_gcp_hook_default_project_id, ): self.instance = hook.GoogleBaseHook(gcp_conn_id="google-cloud-default")