Examples with Invocation org.apache.dubbo.rpc.Invocation used on opensource projects

Example 71 with Invocation

use of org.apache.dubbo.rpc.Invocation in project dubbo by alibaba.

the class AccessKeyAuthenticatorTest method testSignForRequest.

@Test
void testSignForRequest() {
    URL url = URL.valueOf("dubbo://10.10.10.10:2181").addParameter(Constants.ACCESS_KEY_ID_KEY, "ak").addParameter(CommonConstants.APPLICATION_KEY, "test").addParameter(Constants.SECRET_ACCESS_KEY_KEY, "sk");
    Invocation invocation = new RpcInvocation();
    AccessKeyAuthenticator helper = mock(AccessKeyAuthenticator.class);
    doCallRealMethod().when(helper).sign(invocation, url);
    when(helper.getSignature(eq(url), eq(invocation), eq("sk"), anyString())).thenReturn("dubbo");
    AccessKeyPair accessKeyPair = mock(AccessKeyPair.class);
    when(accessKeyPair.getSecretKey()).thenReturn("sk");
    when(helper.getAccessKeyPair(invocation, url)).thenReturn(accessKeyPair);
    helper.sign(invocation, url);
    assertEquals(String.valueOf(invocation.getAttachment(CommonConstants.CONSUMER)), url.getParameter(CommonConstants.APPLICATION_KEY));
    assertNotNull(invocation.getAttachments().get(Constants.REQUEST_SIGNATURE_KEY));
    assertEquals(invocation.getAttachments().get(Constants.REQUEST_SIGNATURE_KEY), "dubbo");
}

Example 72 with Invocation

use of org.apache.dubbo.rpc.Invocation in project dubbo by alibaba.

the class AccessKeyAuthenticatorTest method testGetSignatureWithParameter.

@Test
void testGetSignatureWithParameter() {
    URL url = mock(URL.class);
    when(url.getParameter(Constants.PARAMETER_SIGNATURE_ENABLE_KEY, false)).thenReturn(true);
    Invocation invocation = mock(Invocation.class);
    String secretKey = "123456";
    Object[] params = { "dubbo", new ArrayList() };
    when(invocation.getArguments()).thenReturn(params);
    AccessKeyAuthenticator helper = new AccessKeyAuthenticator();
    String signature = helper.getSignature(url, invocation, secretKey, String.valueOf(System.currentTimeMillis()));
    assertNotNull(signature);
    Object[] fakeParams = { "dubbo1", new ArrayList<>() };
    when(invocation.getArguments()).thenReturn(fakeParams);
    String signature1 = helper.getSignature(url, invocation, secretKey, String.valueOf(System.currentTimeMillis()));
    assertNotEquals(signature, signature1);
}

Example 73 with Invocation

use of org.apache.dubbo.rpc.Invocation in project dubbo by alibaba.

the class AccessKeyAuthenticatorTest method testAuthenticateRequest.

@Test
void testAuthenticateRequest() throws RpcAuthenticationException {
    URL url = URL.valueOf("dubbo://10.10.10.10:2181").addParameter(Constants.ACCESS_KEY_ID_KEY, "ak").addParameter(CommonConstants.APPLICATION_KEY, "test").addParameter(Constants.SECRET_ACCESS_KEY_KEY, "sk");
    Invocation invocation = new RpcInvocation();
    invocation.setAttachment(Constants.ACCESS_KEY_ID_KEY, "ak");
    invocation.setAttachment(Constants.REQUEST_SIGNATURE_KEY, "dubbo");
    invocation.setAttachment(Constants.REQUEST_TIMESTAMP_KEY, String.valueOf(System.currentTimeMillis()));
    invocation.setAttachment(CommonConstants.CONSUMER, "test");
    AccessKeyAuthenticator helper = mock(AccessKeyAuthenticator.class);
    doCallRealMethod().when(helper).authenticate(invocation, url);
    when(helper.getSignature(eq(url), eq(invocation), eq("sk"), anyString())).thenReturn("dubbo");
    AccessKeyPair accessKeyPair = mock(AccessKeyPair.class);
    when(accessKeyPair.getSecretKey()).thenReturn("sk");
    when(helper.getAccessKeyPair(invocation, url)).thenReturn(accessKeyPair);
    assertDoesNotThrow(() -> helper.authenticate(invocation, url));
}

Example 74 with Invocation

use of org.apache.dubbo.rpc.Invocation in project dubbo by alibaba.

the class ConsumerSignFilterTest method testAuthEnabled.

@Test
void testAuthEnabled() {
    URL url = URL.valueOf("dubbo://10.10.10.10:2181").addParameter(Constants.ACCESS_KEY_ID_KEY, "ak").addParameter(Constants.SECRET_ACCESS_KEY_KEY, "sk").addParameter(CommonConstants.APPLICATION_KEY, "test").addParameter(Constants.SERVICE_AUTH, true);
    Invoker invoker = mock(Invoker.class);
    Invocation invocation = mock(Invocation.class);
    when(invoker.getUrl()).thenReturn(url);
    ConsumerSignFilter consumerSignFilter = new ConsumerSignFilter();
    consumerSignFilter.invoke(invoker, invocation);
    verify(invocation, times(1)).setAttachment(eq(Constants.REQUEST_SIGNATURE_KEY), anyString());
}

Example 75 with Invocation

use of org.apache.dubbo.rpc.Invocation in project dubbo by alibaba.

the class ThriftNativeCodec method encodeRequest.

protected void encodeRequest(Channel channel, ChannelBuffer buffer, Request request) throws IOException {
    Invocation invocation = (Invocation) request.getData();
    TProtocol protocol = newProtocol(channel.getUrl(), buffer);
    try {
        protocol.writeMessageBegin(new TMessage(invocation.getMethodName(), TMessageType.CALL, thriftSeq.getAndIncrement()));
        protocol.writeStructBegin(new TStruct(invocation.getMethodName() + "_args"));
        for (int i = 0; i < invocation.getParameterTypes().length; i++) {
            Class<?> type = invocation.getParameterTypes()[i];
        }
    } catch (TException e) {
        throw new IOException(e.getMessage(), e);
    }
}