feat: add airflow_byoid_uri field to Cloud Composer API (#172)

* feat: add airflow_byoid_uri field to Cloud Composer API PiperOrigin-RevId: 527102116 Source-Link: googleapis/googleapis@baf6171 Source-Link: googleapis/googleapis-gen@5b90e61 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNWI5MGU2MTdlNWFmOTIyMTMyYjY5MTg1NzViZDNiNWNmMDA4YTc5OSJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * feat: add airflow_byoid_uri field to Cloud Composer API PiperOrigin-RevId: 527102364 Source-Link: googleapis/googleapis@6a48073 Source-Link: googleapis/googleapis-gen@a161ce2 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYTE2MWNlMmViNjlmZDNhNWNiNTM0Yjk1MTFkMDk5OTFkM2FmZWIyZiJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
googleapis · Apr 26, 2023 · a402b20 · a402b20
1 parent c281d83
commit a402b20
Show file tree

Hide file tree

Showing 6 changed files with 32 additions and 2 deletions.
diff --git a/google/cloud/orchestration/airflow/service_v1/types/environments.py b/google/cloud/orchestration/airflow/service_v1/types/environments.py
@@ -543,6 +543,13 @@ class EnvironmentConfig(proto.Message):
             Output only. The URI of the Apache Airflow Web UI hosted
             within this environment (see `Airflow web
             interface </composer/docs/how-to/accessing/airflow-web-interface>`__).
+        airflow_byoid_uri (str):
+            Output only. The 'bring your own identity' variant of the
+            URI of the Apache Airflow Web UI hosted within this
+            environment, to be accessed with external identities using
+            workforce identity federation (see `Access environments with
+            workforce identity
+            federation </composer/docs/composer-2/access-environments-with-workforce-identity-federation>`__).
         master_authorized_networks_config (google.cloud.orchestration.airflow.service_v1.types.MasterAuthorizedNetworksConfig):
             Optional. The configuration options for GKE
             cluster master authorized networks. By default
@@ -642,6 +649,10 @@ class EnvironmentSize(proto.Enum):
         proto.STRING,
         number=6,
     )
+    airflow_byoid_uri: str = proto.Field(
+        proto.STRING,
+        number=20,
+    )
     master_authorized_networks_config: "MasterAuthorizedNetworksConfig" = proto.Field(
         proto.MESSAGE,
         number=17,

diff --git a/google/cloud/orchestration/airflow/service_v1beta1/types/environments.py b/google/cloud/orchestration/airflow/service_v1beta1/types/environments.py
@@ -669,6 +669,13 @@ class EnvironmentConfig(proto.Message):
             Output only. The URI of the Apache Airflow Web UI hosted
             within this environment (see `Airflow web
             interface </composer/docs/how-to/accessing/airflow-web-interface>`__).
+        airflow_byoid_uri (str):
+            Output only. The 'bring your own identity' variant of the
+            URI of the Apache Airflow Web UI hosted within this
+            environment, to be accessed with external identities using
+            workforce identity federation (see `Access environments with
+            workforce identity
+            federation </composer/docs/composer-2/access-environments-with-workforce-identity-federation>`__).
         encryption_config (google.cloud.orchestration.airflow.service_v1beta1.types.EncryptionConfig):
             Optional. The encryption options for the
             Cloud Composer environment and its dependencies.
@@ -784,6 +791,10 @@ class EnvironmentSize(proto.Enum):
         proto.STRING,
         number=6,
     )
+    airflow_byoid_uri: str = proto.Field(
+        proto.STRING,
+        number=21,
+    )
     encryption_config: "EncryptionConfig" = proto.Field(
         proto.MESSAGE,
         number=12,

diff --git a/...les/generated_samples/snippet_metadata_google.cloud.orchestration.airflow.service.v1.json b/...les/generated_samples/snippet_metadata_google.cloud.orchestration.airflow.service.v1.json
@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-orchestration-airflow",
-    "version": "1.7.1"
+    "version": "0.1.0"
   },
   "snippets": [
     {

diff --git a/...enerated_samples/snippet_metadata_google.cloud.orchestration.airflow.service.v1beta1.json b/...enerated_samples/snippet_metadata_google.cloud.orchestration.airflow.service.v1beta1.json
@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-orchestration-airflow-service",
-    "version": "1.7.1"
+    "version": "0.1.0"
   },
   "snippets": [
     {

diff --git a/tests/unit/gapic/service_v1/test_environments.py b/tests/unit/gapic/service_v1/test_environments.py
@@ -2516,6 +2516,7 @@ def test_create_environment_rest(request_type):
             },
             "environment_size": 1,
             "airflow_uri": "airflow_uri_value",
+            "airflow_byoid_uri": "airflow_byoid_uri_value",
             "master_authorized_networks_config": {
                 "enabled": True,
                 "cidr_blocks": [
@@ -2708,6 +2709,7 @@ def test_create_environment_rest_bad_request(
             },
             "environment_size": 1,
             "airflow_uri": "airflow_uri_value",
+            "airflow_byoid_uri": "airflow_byoid_uri_value",
             "master_authorized_networks_config": {
                 "enabled": True,
                 "cidr_blocks": [
@@ -3330,6 +3332,7 @@ def test_update_environment_rest(request_type):
             },
             "environment_size": 1,
             "airflow_uri": "airflow_uri_value",
+            "airflow_byoid_uri": "airflow_byoid_uri_value",
             "master_authorized_networks_config": {
                 "enabled": True,
                 "cidr_blocks": [
@@ -3522,6 +3525,7 @@ def test_update_environment_rest_bad_request(
             },
             "environment_size": 1,
             "airflow_uri": "airflow_uri_value",
+            "airflow_byoid_uri": "airflow_byoid_uri_value",
             "master_authorized_networks_config": {
                 "enabled": True,
                 "cidr_blocks": [

diff --git a/tests/unit/gapic/service_v1beta1/test_environments.py b/tests/unit/gapic/service_v1beta1/test_environments.py
@@ -2796,6 +2796,7 @@ def test_create_environment_rest(request_type):
             "database_config": {"machine_type": "machine_type_value"},
             "web_server_config": {"machine_type": "machine_type_value"},
             "airflow_uri": "airflow_uri_value",
+            "airflow_byoid_uri": "airflow_byoid_uri_value",
             "encryption_config": {"kms_key_name": "kms_key_name_value"},
             "maintenance_window": {
                 "start_time": {"seconds": 751, "nanos": 543},
@@ -2991,6 +2992,7 @@ def test_create_environment_rest_bad_request(
             "database_config": {"machine_type": "machine_type_value"},
             "web_server_config": {"machine_type": "machine_type_value"},
             "airflow_uri": "airflow_uri_value",
+            "airflow_byoid_uri": "airflow_byoid_uri_value",
             "encryption_config": {"kms_key_name": "kms_key_name_value"},
             "maintenance_window": {
                 "start_time": {"seconds": 751, "nanos": 543},
@@ -3616,6 +3618,7 @@ def test_update_environment_rest(request_type):
             "database_config": {"machine_type": "machine_type_value"},
             "web_server_config": {"machine_type": "machine_type_value"},
             "airflow_uri": "airflow_uri_value",
+            "airflow_byoid_uri": "airflow_byoid_uri_value",
             "encryption_config": {"kms_key_name": "kms_key_name_value"},
             "maintenance_window": {
                 "start_time": {"seconds": 751, "nanos": 543},
@@ -3893,6 +3896,7 @@ def test_update_environment_rest_bad_request(
             "database_config": {"machine_type": "machine_type_value"},
             "web_server_config": {"machine_type": "machine_type_value"},
             "airflow_uri": "airflow_uri_value",
+            "airflow_byoid_uri": "airflow_byoid_uri_value",
             "encryption_config": {"kms_key_name": "kms_key_name_value"},
             "maintenance_window": {
                 "start_time": {"seconds": 751, "nanos": 543},