|
4 | 4 | import com.github.binarywang.wxpay.bean.profitsharing.request.ProfitSharingV3Request; |
5 | 5 | import com.github.binarywang.wxpay.exception.WxPayException; |
6 | 6 | import com.github.binarywang.wxpay.v3.SpecEncrypt; |
| 7 | +import com.google.gson.annotations.SerializedName; |
| 8 | +import lombok.Data; |
7 | 9 | import org.testng.annotations.Test; |
8 | 10 |
|
9 | 11 | import java.lang.reflect.Field; |
@@ -126,4 +128,71 @@ public void testEncryptFieldsWithDirectField() { |
126 | 128 | assertNotNull(request.getName()); |
127 | 129 | assertEquals(request.getName(), "李四"); |
128 | 130 | } |
| 131 | + |
| 132 | + /** |
| 133 | + * 测试类继承场景下的字段加密 |
| 134 | + * 验证父类中带 @SpecEncrypt 注解的字段是否能被正确找到和加密 |
| 135 | + */ |
| 136 | + @Test |
| 137 | + public void testEncryptFieldsWithInheritance() { |
| 138 | + // 定义测试用的父类和子类 |
| 139 | + @Data |
| 140 | + class ParentRequest { |
| 141 | + @SpecEncrypt |
| 142 | + @SerializedName("parent_name") |
| 143 | + private String parentName; |
| 144 | + } |
| 145 | + |
| 146 | + @Data |
| 147 | + @lombok.EqualsAndHashCode(callSuper = false) |
| 148 | + class ChildRequest extends ParentRequest { |
| 149 | + @SpecEncrypt |
| 150 | + @SerializedName("child_name") |
| 151 | + private String childName; |
| 152 | + } |
| 153 | + |
| 154 | + // 创建子类实例 |
| 155 | + ChildRequest request = new ChildRequest(); |
| 156 | + request.setParentName("父类字段"); |
| 157 | + request.setChildName("子类字段"); |
| 158 | + |
| 159 | + // 验证能够找到父类和子类的字段 |
| 160 | + System.out.println("=== 测试继承场景 ==="); |
| 161 | + System.out.println("父类字段值: " + request.getParentName()); |
| 162 | + System.out.println("子类字段值: " + request.getChildName()); |
| 163 | + |
| 164 | + // 使用 getDeclaredFields 只能找到子类字段 |
| 165 | + Field[] childFields = ChildRequest.class.getDeclaredFields(); |
| 166 | + System.out.println("使用 getDeclaredFields 找到的字段数: " + childFields.length); |
| 167 | + |
| 168 | + // 使用 getAllFields 辅助方法应该能找到父类和子类的所有字段 |
| 169 | + java.util.List<Field> allFields = getAllFields(ChildRequest.class); |
| 170 | + System.out.println("使用 getAllFields 找到的字段数: " + allFields.size()); |
| 171 | + |
| 172 | + int annotatedFieldCount = 0; |
| 173 | + for (Field field : allFields) { |
| 174 | + if (field.isAnnotationPresent(SpecEncrypt.class)) { |
| 175 | + annotatedFieldCount++; |
| 176 | + System.out.println(" -> 找到带 @SpecEncrypt 注解的字段: " + field.getName()); |
| 177 | + } |
| 178 | + } |
| 179 | + |
| 180 | + // 应该找到2个带注解的字段(parentName 和 childName) |
| 181 | + assertTrue(annotatedFieldCount >= 2, "应该能找到至少2个带 @SpecEncrypt 注解的字段"); |
| 182 | + } |
| 183 | + |
| 184 | + /** |
| 185 | + * 辅助方法:递归获取类的所有字段,包括父类中的字段 |
| 186 | + */ |
| 187 | + private java.util.List<Field> getAllFields(Class<?> clazz) { |
| 188 | + java.util.List<Field> fields = new ArrayList<>(); |
| 189 | + while (clazz != null) { |
| 190 | + Field[] declaredFields = clazz.getDeclaredFields(); |
| 191 | + for (Field field : declaredFields) { |
| 192 | + fields.add(field); |
| 193 | + } |
| 194 | + clazz = clazz.getSuperclass(); |
| 195 | + } |
| 196 | + return fields; |
| 197 | + } |
129 | 198 | } |
0 commit comments