feat: support for OrgPolicy dry runs

google/cloud/orgpolicy_v2/types/orgpolicy.py

@@ -18,6 +18,7 @@
 import proto  # type: ignore
 
 from google.cloud.orgpolicy_v2.types import constraint
+from google.protobuf import field_mask_pb2  # type: ignore
 from google.protobuf import timestamp_pb2  # type: ignore
 from google.type import expr_pb2  # type: ignore
 
@@ -66,6 +67,11 @@ class Policy(proto.Message):
             Policy.
         alternate (google.cloud.orgpolicy_v2.types.AlternatePolicySpec):
             Deprecated.
+        dry_run_spec (google.cloud.orgpolicy_v2.types.PolicySpec):
+            dry-run policy.
+            Audit-only policy, can be used to monitor how
+            the policy would have impacted the existing and
+            future resources if it's enforced.
     """
 
     name: str = proto.Field(
@@ -82,6 +88,11 @@ class Policy(proto.Message):
         number=3,
         message="AlternatePolicySpec",
     )
+    dry_run_spec: "PolicySpec" = proto.Field(
+        proto.MESSAGE,
+        number=4,
+        message="PolicySpec",
+    )
 
 
 class AlternatePolicySpec(proto.Message):
@@ -500,13 +511,23 @@ class UpdatePolicyRequest(proto.Message):
     Attributes:
         policy (google.cloud.orgpolicy_v2.types.Policy):
             Required. ``Policy`` to update.
+        update_mask (google.protobuf.field_mask_pb2.FieldMask):
+            Field mask used to specify the fields to be overwritten in
+            the policy by the set. The fields specified in the
+            update_mask are relative to the policy, not the full
+            request.
     """
 
     policy: "Policy" = proto.Field(
         proto.MESSAGE,
         number=1,
         message="Policy",
     )
+    update_mask: field_mask_pb2.FieldMask = proto.Field(
+        proto.MESSAGE,
+        number=3,
+        message=field_mask_pb2.FieldMask,
+    )
 
 
 class DeletePolicyRequest(proto.Message):

samples/generated_samples/snippet_metadata_google.cloud.orgpolicy.v2.json

@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-org-policy",
-    "version": "1.6.0"
+    "version": "0.1.0"
   },
   "snippets": [
     {

scripts/fixup_orgpolicy_v2_keywords.py

@@ -45,7 +45,7 @@ class orgpolicyCallTransformer(cst.CSTTransformer):
         'get_policy': ('name', ),
         'list_constraints': ('parent', 'page_size', 'page_token', ),
         'list_policies': ('parent', 'page_size', 'page_token', ),
-        'update_policy': ('policy', ),
+        'update_policy': ('policy', 'update_mask', ),
     }
 
     def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode:

tests/unit/gapic/orgpolicy_v2/test_org_policy.py

@@ -44,6 +44,7 @@
 from google.cloud.orgpolicy_v2.types import constraint
 from google.cloud.orgpolicy_v2.types import orgpolicy
 from google.oauth2 import service_account
+from google.protobuf import field_mask_pb2  # type: ignore
 from google.protobuf import timestamp_pb2  # type: ignore
 from google.type import expr_pb2  # type: ignore
 import google.auth