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 reactor.core.publisher.Mono;
import we.constants.CommonConstants; import we.constants.CommonConstants;
import we.fizz.input.*; import we.fizz.input.*;
import we.fizz.input.extension.dubbo.DubboRPCResponse;
import we.proxy.grpc.GrpcGenericService; import we.proxy.grpc.GrpcGenericService;
import we.proxy.grpc.GrpcInterfaceDeclaration; import we.proxy.grpc.GrpcInterfaceDeclaration;
@@ -42,7 +41,7 @@ public class GrpcInput extends RPCInput implements IInput {
Mono<Object> proxyResponse = proxy.send(body, declaration, contextAttachment); Mono<Object> proxyResponse = proxy.send(body, declaration, contextAttachment);
return proxyResponse.flatMap(cr->{ return proxyResponse.flatMap(cr->{
DubboRPCResponse response = new DubboRPCResponse(); GRPCResponse response = new GRPCResponse();
String responseStr = JSON.toJSONString(cr); String responseStr = JSON.toJSONString(cr);
response.setBodyMono(Mono.just(responseStr)); response.setBodyMono(Mono.just(responseStr));
return Mono.just(response); 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; package we.fizz.input;
import com.google.common.util.concurrent.ListenableFuture; 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.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.mockito.junit.MockitoJUnitRunner; import org.mockito.junit.MockitoJUnitRunner;
import org.springframework.context.ConfigurableApplicationContext; import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.test.util.ReflectionTestUtils; import org.springframework.test.util.ReflectionTestUtils;
import we.fizz.group.FastTestGroup;
import we.fizz.Step; import we.fizz.Step;
import we.fizz.StepContext; import we.fizz.StepContext;
import we.fizz.StepResponse; import we.fizz.StepResponse;
import we.fizz.input.extension.grpc.GrpcInput; import we.fizz.input.extension.grpc.GrpcInput;
import we.fizz.input.extension.grpc.GrpcInputConfig; import we.fizz.input.extension.grpc.GrpcInputConfig;
import we.proxy.dubbo.ApacheDubboGenericService;
import we.proxy.grpc.GrpcGenericService; import we.proxy.grpc.GrpcGenericService;
import we.proxy.grpc.GrpcInterfaceDeclaration; import we.proxy.grpc.GrpcInterfaceDeclaration;
import we.proxy.grpc.client.GrpcProxyClient; import we.proxy.grpc.client.GrpcProxyClient;
@@ -30,7 +27,8 @@ import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*; import static org.mockito.Mockito.*;
@RunWith(MockitoJUnitRunner.class) @RunWith(MockitoJUnitRunner.class)
public class GrpcInputTests { @Category(FastTestGroup.class)
public class GrpcInputMockTests {
private static final String URL ="localhost:8090"; private static final String URL ="localhost:8090";
private static final String SERVICE_NAME = "com.fizzgate.test"; private static final String SERVICE_NAME = "com.fizzgate.test";
private static final String METHOD_NAME = "method"; private static final String METHOD_NAME = "method";
@@ -38,7 +36,7 @@ public class GrpcInputTests {
private static final Object[] RIGHT = new Object[]{}; private static final Object[] RIGHT = new Object[]{};
private ApacheDubboGenericService proxy; private GrpcGenericService proxy;
// @Before // @Before
// public void setup(){ // public void setup(){
// ApacheDubboGenericProxyTests test = new ApacheDubboGenericProxyTests(); // ApacheDubboGenericProxyTests test = new ApacheDubboGenericProxyTests();