2017
Mar
06

如何测试 Java Private Method / Property

Tags:
繁中版

我们要使用 reflect 来测试 Java 的 private method 跟 private variable。

并且用到两个 package: java.lang.reflect.Field 跟 java.lang.reflect.Method。

Private method

下面是一个简单的范例， Class Food 有个 private Method 叫 getLink。

library has private method 
  1. package com.puritys;
  2.  
  3. public class Food {
  4.  
  5.     private String getLink(final int type) {
  6.         switch(type) {
  7.             case 1:
  8.                 return "http://xxx.xx.xx/xxx.html";
  9.             case 2:
  10.                 return "/xxx.html";
  11.         }
  12.         return "none";
  13.     }
  14. }

接著我要写一个 Unit test ， 为了测试 private method ，我们必需先用 getDeclaredMethod 拿到 Java method ，再修改它的 accessible = true，方式如下：

Test private method 
  1. package com.puritys;                                                                           
  2.  
  3. import static org.junit.Assert.assertEquals;
  4. import java.lang.reflect.Method;
  5. import org.junit.*;
  6. import com.puritys.Food;
  7.  
  8. public class FoodTest {
  9.     private Food tester;
  10.  
  11.     @Before
  12.     public void setUp() throws Exception {
  13.         this.tester = new Food();
  14.     }
  15.  
  16.  
  17.     @Test
  18.     public void testApp() throws Exception {
  19.         Method method = this.tester.getClass().getDeclaredMethod("getLink", int.class);
  20.         method.setAccessible(true);
  21.         Object[] parameters = new Object[1];
  22.         parameters[0] = 1;
  23.         String ret = (String) method.invoke(this.tester, parameters);
  24.         assertEquals("The url should be http://", "http://xxx.xx.xx/xxx.html", ret);
  25.  
  26.         parameters[0] = 2;
  27.         ret = (String) method.invoke(this.tester, parameters);
  28.         assertEquals("The url should be xx", "/xxx.html", ret);
  29.  
  30.     }
  31. }
getDeclaredMethod("getLink", int.class)
  • 第二个参数是指 getLink 接受的参数型态 ，getLink 有几个参数，这里就要写多少个。
m.setAccessible(true);
  • 将 private method 改成 accessible。
Object[] parameters = new Object[1];
parameters[0] = 2;
  • 不管 method 需要传几个参数，我们只要宣告一个 Array Object ，set 所有的参数进 Object[]

Private property

下面是一个简单的范例， Class Food 有个 private Property 叫 name 。

private property 
  1. public class Food {
  2.     private String name = "Apple";
  3.  
  4.     public String getName() {
  5.         return this.name;
  6.     }   
  7.  
  8. }

测试 private method 的方式跟测 private property 很像，我们先用 getDeclaredField 拿到 Java property ，再用 setAccessible 修改 access 权限。

test private property 
  1. package com.puritys;                                                                           
  2.  
  3. import static org.junit.Assert.assertEquals;
  4. import java.lang.reflect.Field;
  5. import org.junit.*;
  6. import com.puritys.Food;
  7.  
  8. public class FoodTest {
  9.     @Test
  10.     public void testGetName() { 
  11.         String name = "";                                                                      
  12.         try { 
  13.             Field f = this.tester.getClass().getDeclaredField("name");
  14.             f.setAccessible(true);
  15.             name = (String) f.get(this.tester);
  16.         } catch (Exception e) { 
  17.             System.out.println("has exception, error msg = " + e.getMessage());
  18.         } 
  19.         assertEquals("The name should be Apple", name, "Apple");
  20.     }
  21. }

This post was last modified by puritys


回應 (Leave a comment)

Java 相關

類別 Java

Copyright © 2014 PURITYS.COM. All rights reserved.