Examples with MockUp mockit.MockUp used on opensource projects

Example 6 with MockUp

use of mockit.MockUp in project java-chassis by ServiceComb.

the class TestHighwayServerConnection method testRequestError.

@Test
public void testRequestError() throws Exception {
    header.setMsgType(MsgType.REQUEST);
    Buffer headerBuffer = createBuffer(requestHeaderSchema, header);
    Buffer bodyBuffer = Buffer.buffer();
    Holder<Boolean> holder = new Holder<>(false);
    new MockUp<HighwayServerInvoke>() {

        @Mock
        public boolean init(NetSocket netSocket, long msgId, RequestHeader header, Buffer bodyBuffer) {
            return false;
        }
    };
    connection.handle(0, headerBuffer, bodyBuffer);
    Assert.assertEquals(null, connection.getProtocol());
    Assert.assertEquals(null, connection.getZipName());
    Assert.assertEquals(false, holder.value);
}

Example 7 with MockUp

use of mockit.MockUp in project java-chassis by ServiceComb.

the class MockUtil method mockMicroserviceMetaManager.

public void mockMicroserviceMetaManager() {
    new MockUp<MicroserviceMetaManager>() {

        @Mock
        public SchemaMeta ensureFindSchemaMeta(String schemaId) {
            SchemaMeta schemaMeta = Mockito.mock(SchemaMeta.class);
            OperationMeta operationMeta = Mockito.mock(OperationMeta.class);
            Mockito.when(schemaMeta.ensureFindOperation(null)).thenReturn(operationMeta);
            Method method = this.getClass().getMethods()[0];
            Mockito.when(operationMeta.getMethod()).thenReturn(method);
            OperationProtobuf operationProtobuf = Mockito.mock(OperationProtobuf.class);
            Mockito.when(operationMeta.getExtData("protobuf")).thenReturn(operationProtobuf);
            Executor lExecutor = Mockito.mock(Executor.class);
            Mockito.when(operationMeta.getExecutor()).thenReturn(lExecutor);
            return schemaMeta;
        }
    };
}

Example 8 with MockUp

use of mockit.MockUp in project java-chassis by ServiceComb.

the class TestGrpcServerInvoke method testOnProviderResponseSuccess.

@Test
public void testOnProviderResponseSuccess() {
    //boolean status = false;
    try {
        MockUtil.getInstance().mockMicroserviceMetaManager();
        mockForExecutorManager();
        mockForOperationMeta();
        new MockUp<GrpcServerInvoke>() {

            @Mock
            private void sendFailResponse(Throwable throwable) {
            // Nothing to do Just For mock  
            }
        };
        Invocation invocation = Mockito.mock(Invocation.class);
        Response response = Mockito.mock(Response.class);
        Mockito.when(response.isSuccessed()).thenReturn(true);
        Mockito.when(response.getResult()).thenReturn("test");
        Mockito.when(routingContext.response()).thenReturn(Mockito.mock(HttpServerResponse.class));
        Deencapsulation.setField(lGrpcServerInvoke, "routingContext", routingContext);
        Deencapsulation.invoke(lGrpcServerInvoke, "onProviderResponse", invocation, response);
    } catch (Exception e) {
    // status = true;
    }
//Assert.assertTrue(status);
}

Example 9 with MockUp

use of mockit.MockUp in project java-chassis by ServiceComb.

the class TestGrpcTransport method testSend.

@Test
public void testSend() {
    boolean status = false;
    Invocation invocation = Mockito.mock(Invocation.class);
    AsyncResponse asyncResp = Mockito.mock(AsyncResponse.class);
    OperationMeta operationMeta = Mockito.mock(OperationMeta.class);
    Mockito.when(invocation.getOperationMeta()).thenReturn(operationMeta);
    OperationProtobuf operationProtobuf = Mockito.mock(OperationProtobuf.class);
    Mockito.when(operationMeta.getExtData("protobuf")).thenReturn(operationProtobuf);
    Endpoint lEndpoint = Mockito.mock(Endpoint.class);
    Mockito.when(invocation.getEndpoint()).thenReturn(lEndpoint);
    WrapSchema lWrapSchema = Mockito.mock(WrapSchema.class);
    Mockito.when(operationProtobuf.getRequestSchema()).thenReturn(lWrapSchema);
    IpPort ipPort = Mockito.mock(IpPort.class);
    Mockito.when(lEndpoint.getAddress()).thenReturn(ipPort);
    Mockito.when(ipPort.getHostOrIp()).thenReturn("127.0.0.1");
    Mockito.when(ipPort.getPort()).thenReturn(80);
    new MockUp<ClientPoolManager<HttpClientWithContext>>() {

        @Mock
        public HttpClientWithContext findThreadBindClientPool() {
            return Mockito.mock(HttpClientWithContext.class);
        }
    };
    try {
        transport.send(invocation, asyncResp);
    } catch (Exception e) {
        status = true;
    }
    Assert.assertFalse(status);
}

Example 10 with MockUp

use of mockit.MockUp in project java-chassis by ServiceComb.

the class TestPath method testQueryVarParamWriter.

@Test
public void testQueryVarParamWriter() throws Exception {
    boolean status = true;
    new MockUp<RestParam>() {

        @Mock
        public String getParamName() {
            return "queryVar";
        }
    };
    new MockUp<QueryVarParamWriter>() {

        @Mock
        protected Object getParamValue(Object[] args) {
            return args[0];
        }
    };
    Parameter parameter = new QueryParameter();
    QueryVarParamWriter writer = new QueryVarParamWriter('&', new RestParam(0, parameter, String.class));
    try {
        verify(writer, "T", "&queryVar=T");
        verify(writer, null, "&");
        verify(writer, new String[] { "a", "b" }, "&queryVar=a&queryVar=b");
        verify(writer, new String[] { "a", null, "b" }, "&queryVar=a&queryVar=&queryVar=b");
        verify(writer, Arrays.asList("Lars", "Simon"), "&queryVar=Lars&queryVar=Simon");
        verify(writer, "测试", "&queryVar=%E6%B5%8B%E8%AF%95");
        verify(writer, "a b", "&queryVar=a+b");
        verify(writer, "a+b", "&queryVar=a%2Bb");
    } catch (Exception e) {
        status = false;
    }
    Assert.assertTrue(status);
}