github-repos/testable-mock
1
0
Fork 0
mirror of https://github.com/alibaba/testable-mock.git synced 2025-01-25 03:40:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

support access array element by index

Browse Source
This commit is contained in:
金戟 2021-03-18 00:15:01 +08:00
parent af3af6b272
commit 206fc5486e
5 changed files with 90 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
testable-core/src/main/java/com/alibaba/testable/core/tool/OmniAccessor.java
View File

@ -4,6 +4,7 @@ import com.alibaba.testable.core.util.FixSizeMap;
import com.alibaba.testable.core.util.TypeUtil;
import com.sun.deploy.util.StringUtils;
import java.lang.reflect.Array;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Arrays;
@ -147,7 +148,7 @@ public class OmniAccessor {
private static int extraIndexFromQuery(String query) {
return query.endsWith(BRACKET_END)
? Integer.parseInt(query.substring(query.lastIndexOf(BRACKET_START) + 1), query.length() - 1)
? Integer.parseInt(query.substring(query.lastIndexOf(BRACKET_START) + 1, query.length() - 1))
: -1;
}
@ -165,7 +166,12 @@ public class OmniAccessor {
nth = extraIndexFromQuery(querySegments[i]);
field = TypeUtil.getFieldByName(obj.getClass(), name);
field.setAccessible(true);
obj = field.get(obj);
if (field.getType().isArray() && nth >= 0) {
Object f = field.get(obj);
obj = Array.get(f, nth);
} else {
obj = field.get(obj);
}
}
return obj;
}
@ -176,7 +182,18 @@ public class OmniAccessor {
int nth = extraIndexFromQuery(querySegment);
Field field = TypeUtil.getFieldByName(target.getClass(), name);
field.setAccessible(true);
field.set(target, value);
if (field.getType().isArray()) {
Object f = field.get(target);
if (nth >= 0) {
Array.set(f, nth, value);
} else {
for (int i = 0; i < Array.getLength(f); i++) {
Array.set(f, i, value);
}
}
} else {
field.set(target, value);
}
}
}

38
testable-core/src/main/java/com/alibaba/testable/core/tool/OmniConstructor.java
View File

@ -3,10 +3,7 @@ package com.alibaba.testable.core.tool;
import com.alibaba.testable.core.model.Null;
import com.alibaba.testable.core.util.TypeUtil;
import java.lang.reflect.Array;
import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.*;
/**
* @author flin
@ -19,9 +16,14 @@ public class OmniConstructor {
return newPrimitive(clazz);
} else if (clazz.isArray()) {
return newArray(clazz);
} else {
return newObject(clazz);
} else if (clazz.isEnum()) {
return newEnum(clazz);
} else if (clazz.isInterface()) {
return newInterface(clazz);
} else if (Modifier.isAbstract(clazz.getModifiers())) {
return newAbstractClass(clazz);
}
return newObject(clazz);
} catch (NoSuchMethodException e) {
return null;
} catch (IllegalAccessException e) {
@ -30,17 +32,11 @@ public class OmniConstructor {
return null;
} catch (InvocationTargetException e) {
return null;
} catch (ClassCastException e) {
return null;
}
}
public static <T> void appendInstance(Object target, Class<T> clazz) {
appendInstance(target, 1, clazz);
}
public static <T> void appendInstance(Object target, int count, Class<T> clazz) {
return;
}
private static <T> T newObject(Class<T> clazz)
throws NoSuchMethodException, IllegalAccessException, InvocationTargetException, InstantiationException {
Constructor<?> constructor = getBestConstructor(clazz);
@ -53,6 +49,20 @@ public class OmniConstructor {
return (T)ins;
}
private static <T> T newAbstractClass(Class<T> clazz) {
return null;
}
private static <T> T newInterface(Class<T> clazz) {
return null;
}
private static <T> T newEnum(Class<T> clazz)
throws NoSuchMethodException, InvocationTargetException, IllegalAccessException {
T[] constants = clazz.getEnumConstants();
return constants.length > 0 ? constants[0] : null;
}
private static <T> T newArray(Class<T> clazz) {
return (T)Array.newInstance(clazz.getComponentType(), 0);
}

9
testable-core/src/test/java/com/alibaba/testable/core/tool/DemoGrandChild.java
View File

@ -7,4 +7,13 @@ public class DemoGrandChild {
public DemoGrandChild(int i) {
this.i = i;
}
public int get() {
return i;
}
public void set(int i) {
this.i = i;
}
}

6
testable-core/src/test/java/com/alibaba/testable/core/tool/DemoParent.java
View File

@ -4,10 +4,10 @@ public class DemoParent {
public DemoChild c;
private DemoChild[] cs;
public DemoChild[] cs;
public DemoChild.SubChild sc;
private DemoChild.SubChild sc;
public DemoChild.StaticSubChild ssc;
private DemoChild.StaticSubChild ssc;
}

40
testable-core/src/test/java/com/alibaba/testable/core/tool/OmniAccessorTest.java
View File

@ -73,21 +73,49 @@ class OmniAccessorTest {
@Test
void should_get_by_path() {
DemoParent parent = prepareDemoObject();
DemoParent parent = prepareParentObject();
Object obj = PrivateAccessor.<String>invokeStatic(OmniAccessor.class, "getByPath", parent, "c{DemoChild}/gc{DemoGrandChild}", "c/gc");
assertTrue(obj instanceof DemoGrandChild);
assertEquals(0, PrivateAccessor.<Integer>get(obj, "i"));
assertEquals(0, ((DemoGrandChild)obj).get());
PrivateAccessor.set(parent.c, "gcs", new DemoGrandChild[] { new DemoGrandChild(4), new DemoGrandChild(6) });
obj = PrivateAccessor.<String>invokeStatic(OmniAccessor.class, "getByPath", parent, "c{DemoChild}/gcs{DemoGrandChild[]}", "c/gcs");
assertTrue(obj instanceof DemoGrandChild[]);
assertEquals(2, ((DemoGrandChild[])obj).length);
obj = PrivateAccessor.<String>invokeStatic(OmniAccessor.class, "getByPath", parent, "c{DemoChild}/gcs{DemoGrandChild[]}", "c/gcs[1]");
assertTrue(obj instanceof DemoGrandChild);
assertEquals(6, ((DemoGrandChild)obj).get());
parent.cs = new DemoChild[] { null, prepareChildObject() };
obj = PrivateAccessor.<String>invokeStatic(OmniAccessor.class, "getByPath", parent, "cs{DemoChild[]}/gcs{DemoGrandChild[]}/i{int}", "c[1]/gcs[1]/i");
assertEquals(3, obj);
}
@Test
void should_set_by_path_segment() {
DemoParent parent = prepareDemoObject();
DemoParent parent = prepareParentObject();
DemoChild child = prepareChildObject();
PrivateAccessor.<String>invokeStatic(OmniAccessor.class, "setByPathSegment", parent.c, "gc{DemoGrandChild}", "gc", new DemoGrandChild(2));
assertEquals(2, PrivateAccessor.<Integer>get(parent.c.gc, "i"));
assertEquals(2, parent.c.gc.get());
PrivateAccessor.<String>invokeStatic(OmniAccessor.class, "setByPathSegment", parent, "cs{DemoChild[]}", "cs[2]", child);
assertNull(parent.cs[0]);
assertNull(parent.cs[1]);
assertEquals(5, parent.cs[2].gc.get());
PrivateAccessor.<String>invokeStatic(OmniAccessor.class, "setByPathSegment", parent, "cs{DemoChild[]}", "cs", child);
assertEquals(5, parent.cs[0].gc.get());
assertEquals(5, parent.cs[1].gc.get());
assertEquals(5, parent.cs[2].gc.get());
}
private DemoParent prepareDemoObject() {
return OmniConstructor.newInstance(DemoParent.class);
private DemoParent prepareParentObject() {
DemoParent parent = OmniConstructor.newInstance(DemoParent.class);
parent.cs = new DemoChild[3];
return parent;
}
private DemoChild prepareChildObject() {
DemoChild child = OmniConstructor.newInstance(DemoChild.class);
PrivateAccessor.set(child, "gcs", new DemoGrandChild[] { null, new DemoGrandChild(3) });
child.gc.set(5);
return child;
}
}