add repeated enums

marwan-at-work · Feb 22, 2020 · 4ff8162 · 4ff8162
1 parent 7907fab
commit 4ff8162
Show file tree

Hide file tree

Showing 9 changed files with 383 additions and 142 deletions.
diff --git a/e2e/e2e_test.go b/e2e/e2e_test.go
@@ -43,7 +43,8 @@ func TestTrafficJam(t *testing.T) {
 		"operationName": "q",
 		"variables": {
 			"req": {
-				"color": "GREEN"
+				"color": "GREEN",
+				"trafficLights": ["YELLOW", "RED"]
 			}
 		},
 		"query": "query q($req: TrafficJamReq) {\n  trafficJam(req: $req) {\n next }\n}\n"
@@ -52,6 +53,7 @@ func TestTrafficJam(t *testing.T) {
 	h.ServeHTTP(w, req)
 
 	require.Equal(t, e2e.TrafficLight_GREEN, s.trafficJamReq.GetColor(), "Expected GraphQL request to populate Twirp Object")
+	require.Equal(t, []e2e.TrafficLight{e2e.TrafficLight_YELLOW, e2e.TrafficLight_RED}, s.trafficJamReq.GetTrafficLights(), "Expected repeated enums to be equal")
 
 	expected := `{"data":{"trafficJam":{"next":"YELLOW"}}}`
 

diff --git a/e2e/service.pb.go b/e2e/service.pb.go
diff --git a/e2e/service.proto b/e2e/service.proto
@@ -13,7 +13,7 @@ service Service {
   rpc Bread(BreadReq) returns (BreadResp);
   rpc ChangeMe(ChangeMeReq) returns (ChangeMeResp) {
     option (twirpql.options.rpc) = {
-      mutation: true;
+      mutation: true
     };
   };
 }
@@ -28,6 +28,7 @@ message HelloResp {
 
 message TrafficJamReq {
   TrafficLight color = 1;
+  repeated TrafficLight trafficLights = 2;
 }
 
 enum TrafficLight {