add grpc mock test

This commit is contained in:
linwaiwai
2021-02-04 10:50:07 +08:00
parent 43b6550de7
commit ee3a5b40e2
4 changed files with 25 additions and 9 deletions

View File

@@ -0,0 +1,15 @@
package we.fizz.input.extension.grpc;
import org.springframework.http.HttpStatus;
import we.fizz.input.RPCResponse;
public class GRPCResponse extends RPCResponse {
private HttpStatus statusCode;
public void setStatus(HttpStatus statusCode) {
this.statusCode = statusCode;
}
public HttpStatus getStatusCode() {
return statusCode;
}
}

View File

@@ -5,7 +5,6 @@ import org.springframework.context.ConfigurableApplicationContext;
import reactor.core.publisher.Mono;
import we.constants.CommonConstants;
import we.fizz.input.*;
import we.fizz.input.extension.dubbo.DubboRPCResponse;
import we.proxy.grpc.GrpcGenericService;
import we.proxy.grpc.GrpcInterfaceDeclaration;
@@ -42,7 +41,7 @@ public class GrpcInput extends RPCInput implements IInput {
Mono<Object> proxyResponse = proxy.send(body, declaration, contextAttachment);
return proxyResponse.flatMap(cr->{
DubboRPCResponse response = new DubboRPCResponse();
GRPCResponse response = new GRPCResponse();
String responseStr = JSON.toJSONString(cr);
response.setBodyMono(Mono.just(responseStr));
return Mono.just(response);

View File

@@ -0,0 +1,4 @@
package we.fizz.group;
public class FastTestGroup {
}

View File

@@ -1,21 +1,18 @@
package we.fizz.input;
import com.google.common.util.concurrent.ListenableFuture;
import io.grpc.ManagedChannel;
import org.apache.dubbo.config.ReferenceConfig;
import org.apache.dubbo.rpc.service.GenericService;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.mockito.junit.MockitoJUnitRunner;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.test.util.ReflectionTestUtils;
import we.fizz.group.FastTestGroup;
import we.fizz.Step;
import we.fizz.StepContext;
import we.fizz.StepResponse;
import we.fizz.input.extension.grpc.GrpcInput;
import we.fizz.input.extension.grpc.GrpcInputConfig;
import we.proxy.dubbo.ApacheDubboGenericService;
import we.proxy.grpc.GrpcGenericService;
import we.proxy.grpc.GrpcInterfaceDeclaration;
import we.proxy.grpc.client.GrpcProxyClient;
@@ -30,7 +27,8 @@ import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;
@RunWith(MockitoJUnitRunner.class)
public class GrpcInputTests {
@Category(FastTestGroup.class)
public class GrpcInputMockTests {
private static final String URL ="localhost:8090";
private static final String SERVICE_NAME = "com.fizzgate.test";
private static final String METHOD_NAME = "method";
@@ -38,7 +36,7 @@ public class GrpcInputTests {
private static final Object[] RIGHT = new Object[]{};
private ApacheDubboGenericService proxy;
private GrpcGenericService proxy;
// @Before
// public void setup(){
// ApacheDubboGenericProxyTests test = new ApacheDubboGenericProxyTests();