fix: Fix REST tests generation for repeated enums (#1421)

Co-authored-by: Anthonios Partheniou <[email protected]>

Author: vam-google

packages/gapic-generator/gapic/schema/wrappers.py

@@ -137,7 +137,11 @@ def recursive_mock_original_type(field):
 
             if field.enum:
                 # First Truthy value, fallback to the first value
-                return next((v for v in field.type.values if v.number), field.type.values[0]).number
+                answer = next(
+                    (v for v in field.type.values if v.number), field.type.values[0]).number
+                if field.repeated:
+                    answer = [answer]
+                return answer
 
             answer = field.primitive_mock() or None
 

packages/gapic-generator/tests/unit/schema/wrappers/test_field.py

@@ -463,6 +463,34 @@ def test_mock_value_original_type_enum():
     assert empty_field.mock_value_original_type == 0
 
 
+def test_mock_value_original_type_enum_repeated():
+    mollusc_field = make_field(
+        name="class",
+        enum=make_enum(
+            name="Class",
+            values=[
+                ("UNKNOWN", 0),
+                ("BIVALVE", 2),
+                ("CEPHALOPOD", 3),
+            ],
+        ),
+        label=3,
+    )
+
+    assert mollusc_field.mock_value_original_type == [2]
+
+    empty_field = make_field(
+        name="empty",
+        enum=make_enum(
+            name="Empty",
+            values=[("UNKNOWN", 0)],
+        ),
+        label=3,
+    )
+
+    assert empty_field.mock_value_original_type == [0]
+
+
 @pytest.mark.parametrize(
     "mock_method,expected",
     [