@@ -62,7 +62,7 @@ public void reset() {
6262 @ Override
6363 public void listWorkflows (
6464 ListWorkflowsRequest request , StreamObserver <ListWorkflowsResponse > responseObserver ) {
65- Object response = responses .remove ();
65+ Object response = responses .poll ();
6666 if (response instanceof ListWorkflowsResponse ) {
6767 requests .add (request );
6868 responseObserver .onNext (((ListWorkflowsResponse ) response ));
@@ -74,15 +74,15 @@ public void listWorkflows(
7474 new IllegalArgumentException (
7575 String .format (
7676 "Unrecognized response type %s for method ListWorkflows, expected %s or %s" ,
77- response .getClass ().getName (),
77+ response == null ? "null" : response .getClass ().getName (),
7878 ListWorkflowsResponse .class .getName (),
7979 Exception .class .getName ())));
8080 }
8181 }
8282
8383 @ Override
8484 public void getWorkflow (GetWorkflowRequest request , StreamObserver <Workflow > responseObserver ) {
85- Object response = responses .remove ();
85+ Object response = responses .poll ();
8686 if (response instanceof Workflow ) {
8787 requests .add (request );
8888 responseObserver .onNext (((Workflow ) response ));
@@ -94,7 +94,7 @@ public void getWorkflow(GetWorkflowRequest request, StreamObserver<Workflow> res
9494 new IllegalArgumentException (
9595 String .format (
9696 "Unrecognized response type %s for method GetWorkflow, expected %s or %s" ,
97- response .getClass ().getName (),
97+ response == null ? "null" : response .getClass ().getName (),
9898 Workflow .class .getName (),
9999 Exception .class .getName ())));
100100 }
@@ -103,7 +103,7 @@ public void getWorkflow(GetWorkflowRequest request, StreamObserver<Workflow> res
103103 @ Override
104104 public void createWorkflow (
105105 CreateWorkflowRequest request , StreamObserver <Operation > responseObserver ) {
106- Object response = responses .remove ();
106+ Object response = responses .poll ();
107107 if (response instanceof Operation ) {
108108 requests .add (request );
109109 responseObserver .onNext (((Operation ) response ));
@@ -115,7 +115,7 @@ public void createWorkflow(
115115 new IllegalArgumentException (
116116 String .format (
117117 "Unrecognized response type %s for method CreateWorkflow, expected %s or %s" ,
118- response .getClass ().getName (),
118+ response == null ? "null" : response .getClass ().getName (),
119119 Operation .class .getName (),
120120 Exception .class .getName ())));
121121 }
@@ -124,7 +124,7 @@ public void createWorkflow(
124124 @ Override
125125 public void deleteWorkflow (
126126 DeleteWorkflowRequest request , StreamObserver <Operation > responseObserver ) {
127- Object response = responses .remove ();
127+ Object response = responses .poll ();
128128 if (response instanceof Operation ) {
129129 requests .add (request );
130130 responseObserver .onNext (((Operation ) response ));
@@ -136,7 +136,7 @@ public void deleteWorkflow(
136136 new IllegalArgumentException (
137137 String .format (
138138 "Unrecognized response type %s for method DeleteWorkflow, expected %s or %s" ,
139- response .getClass ().getName (),
139+ response == null ? "null" : response .getClass ().getName (),
140140 Operation .class .getName (),
141141 Exception .class .getName ())));
142142 }
@@ -145,7 +145,7 @@ public void deleteWorkflow(
145145 @ Override
146146 public void updateWorkflow (
147147 UpdateWorkflowRequest request , StreamObserver <Operation > responseObserver ) {
148- Object response = responses .remove ();
148+ Object response = responses .poll ();
149149 if (response instanceof Operation ) {
150150 requests .add (request );
151151 responseObserver .onNext (((Operation ) response ));
@@ -157,7 +157,7 @@ public void updateWorkflow(
157157 new IllegalArgumentException (
158158 String .format (
159159 "Unrecognized response type %s for method UpdateWorkflow, expected %s or %s" ,
160- response .getClass ().getName (),
160+ response == null ? "null" : response .getClass ().getName (),
161161 Operation .class .getName (),
162162 Exception .class .getName ())));
163163 }
0 commit comments