Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add trpc-spring junit #35

Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1,116 @@
/*
* Tencent is pleased to support the open source community by making tRPC available.
*
* Copyright (C) 2023 THL A29 Limited, a Tencent company.
* All rights reserved.
*
* If you have downloaded a copy of the tRPC source code from Tencent,
* please note that tRPC source code is licensed under the Apache 2.0 License,
* A copy of the Apache 2.0 License can be found in the LICENSE file.
*/

package com.tencent.trpc.spring.context.configuration;

import com.tencent.trpc.core.common.ConfigManager;
import com.tencent.trpc.core.common.config.ClientConfig;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.List;
import org.junit.Assert;
import org.junit.Test;


public class AddFilterTRpcConfigManagerCustomizerTest {


private static final String KEY = "OrderKey:b131d74c753540db859c626c564aa8ce";

@Test
public void testConstructor() {
AddFilterTRpcConfigManagerCustomizer addFilterTRpcConfigManagerCustomizer = new AddFilterTRpcConfigManagerCustomizer();
Assert.assertNotNull(addFilterTRpcConfigManagerCustomizer);
}

@Test
public void testAddClientFilters() {
AddFilterTRpcConfigManagerCustomizer addFilterTRpcConfigManagerCustomizer = new AddFilterTRpcConfigManagerCustomizer();
AddFilterTRpcConfigManagerCustomizer customizer = addFilterTRpcConfigManagerCustomizer.addClientFilters(
"filter1", "filter2");
Assert.assertEquals(addFilterTRpcConfigManagerCustomizer, customizer);
}

@Test
public void testAddServerFilters() {
AddFilterTRpcConfigManagerCustomizer addFilterTRpcConfigManagerCustomizer = new AddFilterTRpcConfigManagerCustomizer();
AddFilterTRpcConfigManagerCustomizer customizer = addFilterTRpcConfigManagerCustomizer.addServerFilters(
"filter1", "filter2");
Assert.assertEquals(addFilterTRpcConfigManagerCustomizer, customizer);
}

@Test
public void testCustomize() {
AddFilterTRpcConfigManagerCustomizer addFilterTRpcConfigManagerCustomizer = new AddFilterTRpcConfigManagerCustomizer();

ConfigManager instance = ConfigManager.getInstance();
ClientConfig clientConfig = new ClientConfig();
List<String> list = Arrays.asList("filter1", "filter2");
clientConfig.setFilters(list);
instance.setClientConfig(clientConfig);
addFilterTRpcConfigManagerCustomizer.customize(instance);
List<String> filters = instance.getClientConfig().getFilters();

Assert.assertEquals(filters.size(), list.size());
Assert.assertEquals(list, instance.getClientConfig().getFilters());
}

@Test
public void testGetOrder() {
AddFilterTRpcConfigManagerCustomizer addFilterTRpcConfigManagerCustomizer = new AddFilterTRpcConfigManagerCustomizer2();
Assert.assertEquals(Integer.MAX_VALUE, addFilterTRpcConfigManagerCustomizer.getOrder());
Integer value = 1024;
System.setProperty(KEY, String.valueOf(value));
Assert.assertEquals((long) value, addFilterTRpcConfigManagerCustomizer.getOrder());
}

static final class AddFilterTRpcConfigManagerCustomizer2 extends AddFilterTRpcConfigManagerCustomizer {

@Override
public int getOrder() {
return System.getProperty(KEY) == null ? super.getOrder() : Integer.parseInt(System.getProperty(KEY));
}
}

@Test
public void testEmptyIfNull() throws NoSuchMethodException, InvocationTargetException, IllegalAccessException {

AddFilterTRpcConfigManagerCustomizer customizer = new AddFilterTRpcConfigManagerCustomizer();

Method method = customizer.getClass().getDeclaredMethod("emptyIfNull", List.class);
boolean accessible = method.isAccessible();
method.setAccessible(true);
List<String> list1 = Arrays.asList("filter1", "filter2", "filter3");
Object invoke = method.invoke(customizer, list1);
if (invoke instanceof List) {
List<String> resList = (List<String>) invoke;
Assert.assertEquals(list1.size(), resList.size());
}
method.setAccessible(accessible);
}

@Test
public void testMerge() throws NoSuchMethodException, InvocationTargetException, IllegalAccessException {
AddFilterTRpcConfigManagerCustomizer customizer = new AddFilterTRpcConfigManagerCustomizer();
List<String> list1 = Arrays.asList("filter1", "filter2", "filter3");
List<String> list2 = Arrays.asList("filter4", "filter5", "filter6");
Method method = customizer.getClass().getDeclaredMethod("merge", List.class, List.class);
boolean accessible = method.isAccessible();
method.setAccessible(true);
Object invoke = method.invoke(customizer, list1, list2);
if (invoke instanceof List) {
List<String> resList = (List<String>) invoke;
Assert.assertEquals(list1.size() + list2.size(), resList.size());
}
method.setAccessible(accessible);
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ public void testMessageToMessage() {
public void testUnsupported() {
DefaultExceptionResultTransformer transform = new DefaultExceptionResultTransformer();
MyResponse myResponse = new MyResponse("testObjectToMessage", "123456");
Object result = transform.transform(myResponse, List.class);
transform.transform(myResponse, List.class);
Assert.fail();
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -35,10 +35,10 @@ public void testInvocableExceptionHandler() throws Exception {
HashSet<InvocableExceptionHandler> set = new HashSet<>();
set.add(handler1);
Assert.assertEquals(handler1.hashCode(), handler2.hashCode());
Assert.assertEquals(true, handler1.equals(handler1));
Assert.assertEquals(true, handler1.equals(handler2));
Assert.assertEquals(false, handler1.equals(bean));
Assert.assertEquals(true, set.contains(handler2));
Assert.assertEquals(handler1, handler1);
Assert.assertEquals(handler1, handler2);
Assert.assertNotEquals(handler1, bean);
Assert.assertTrue(set.contains(handler2));
Assert.assertEquals(MyExceptionHandle.class, handler1.getTargetType());
}

Expand All @@ -60,7 +60,8 @@ public void testHandleException() throws Throwable {
Assert.assertEquals("myBean1", methodParameters[0].getParameterName());
Assert.assertEquals("e", methodParameters[1].getParameterName());
Assert.assertEquals("method11111", methodParameters[2].getParameterName());
Assert.assertEquals(true, String.class.isAssignableFrom(result.getClass()));
assert result != null;
Assert.assertTrue(String.class.isAssignableFrom(result.getClass()));
Assert.assertEquals("name1_IllegalArgumentException_targetMethod", result);
}

Expand Down
Loading