綠燈 | 紅燈

妙思資訊
顯示具有 mock 標籤的文章。 顯示所有文章
顯示具有 mock 標籤的文章。 顯示所有文章

2012年1月2日 星期一

你開發的程式好測試嗎(五)?

第四種情況:
Method使用Singleton物件!

所謂Singleton模式就是獨一物件模式,若一個Class 已經有一個Instance,就無法再new 出第二個Instance。例如:系統只有一個加密器叫OnlyOneCipher,因為不想讓使用者能自行new 出一個新的加密器,所以就把OnlyOneCipher設計成一個Singleton物件。程式碼如下: 
public class OnlyoneCipher {

 private static OnlyoneCipher cipher;

 private OnlyoneCipher() {}

 public static OnlyoneCipher getInstance() {

  if (cipher == null) {
   cipher = new OnlyoneCipher();
  }
  return cipher;
 }
 
 
 public String encrypt(String source){
                 
  return doEncrypt(source);
 }

}

因為是使用Private 的Constructor(),所以只有自已能夠產出OnlyoneCipher 的Instance。外部要取得OnlyoneCipher 的Instance只能經由getInstance()這個static method。


接下來我們來看其它物件會如何引用這個OnlyoneCipher。假設有一支MessageService是用來做訊息傳遞,當要傳遞加密訊息時需經由encryptMsg()來處理。程式碼如下:
public class MessageService {
  

 public String encryptMsg(String source){
    
  return OnlyoneCipher.getInstance().encrypt(source);  
  
 }

}

因為只能經由getInstance()才可取得OnlyoneCipher物件,所以在程式中很直覺的寫法就是OnlyoneCipher.getInstance(),當引用
OnlyoneCipher的地方越多,則OnlyoneCipher.getInstance()就會散到整個應用程式中。當有一天想要換更新的Cipher,這時候麻煩就來了,除非要把所有的OnlyoneCipher.getInstance()置換,否則換不掉。另外,在進行unit test 也很難測,因為OnlyoneCipher.getInstance()這種寫法等於是跟encryptMsg()綁死,抽換不掉,如果這個OnlyoneCipher又跟底層或是環境如KEY綁在一起,那就更難測了,在測試環境下,OnlyoneCipher也許會無法正常運作,而Singleton取得物件的模式,會使得測試更難進行。

當然,Singleton 還是可以測,只是寫法要變,而且要使用Mock Framework。因為Singleton 有用到Private Constructor 所以無法extend,因此就無法使用subclass Mock,要使用CGlib 動態產生Mock Class,這時就可以使用JMock、EasyMock、Mockito 等Framework 。利用Mock 來測試Singleton 的程式碼可以改寫如下:


public class MessageService {  

  private OnlyoneCipher cipher;  
  
  
 public void setCipher(OnlyoneCipher cipher) {
  this.cipher = cipher;
 }


 public String encryptMsg(String source){
    
  return cipher.encrypt(source);  
  
 }

}

使用EasyMock 對OnlyoneCipher 做Mock,利用此Mock Inject 到MessageService中,以便進行單元測試!

public class SingletonTest {
 
 private MessageService service;
 private OnlyoneCipher cipher; 
 
 @Before
 public void setup(){
  
  service=new MessageService();
        //對OnlyoneCipher 做Mock
  cipher=createMock(OnlyoneCipher.class);  
  service.setCipher(cipher);
 }
 
 
 @Test
 public void testEncrypt(){  
  
  String source="Test String";
  String encode="aaBBddaa";  
  expect(cipher.encrypt(source)).andReturn(encode);
  replay(cipher);
  String msg=service.encryptMsg(source); 
  verify();
  assertEquals(encode,msg);
 }

}


如果非必要,儘可能少用Singleton 模式,如果想要產生獨一物件,不見得要用Singleton Pattern,可以透過Spring 來提供獨一物件。雖然Spring 控管的Bean 使用者還是可自行new ,但是比起Singleton 模式好做多了。


所以,程式用了很多Singleton,將來要置換會很麻煩而且測試也不容易,難怪會有人說 "Singleton Is Evil"。

2011年12月19日 星期一

你開發的程式好測試嗎(三)?

第二種情況:
傳入到Method的參數,型態為底層的Interface。

以Spring 3.0 MVC 為例:
假設有一個RentController,該Controler 有一個 tenantPage() Method ,傳入Metod 的參數皆為底層的Interface 。程式碼如下:

@Controller
@RequestMapping("/rent")
public class RentController {
 
 @Autowired
 private IRentService service;


 @RequestMapping(value = "/tenant", method = RequestMethod.POST)
 public String tenantPage(Model model,HttpServletRequest request) {

   Integer id=Integer.valueOf(request.getParameter("id"));
   Tenant tenant=service.getTenant(id);
   model.addAttribute("tenant", tenant);
  return "tenant";
 }
 
 
 public void setService(IRentService service) {
  this.service = service;
 }

}

其中Method 參數: model 的型態為org.springframework.ui.Model;request 的型態為javax.servlet.http.HttpServletRequest。 二個參數型態都是Interface,當撰寫Unit Test時會出現一個狀況,必需要實作這些Interface才能將參數傳入。為了測試而去實作這些複雜的Interface是沒有效率的,如果下次碰到不同的Interface 又要再實作一次,那就不會有人想再去寫Unit Test 了。

@Test
 public void testRentController(){
  
  RentController rc=new RentController();
  rc.setService(new RentServiceImpl());
  //model 和 HttpServletRequest 都是Interface,new 不出來
  Model model=yourModelImpl();
  HttpServletRequest request=yourRequestImpl();
  String pageName=rc.tenantPage(model, request);
  Assert.assertEquals("tenant",pageName);
  
 }

所以當測試遇到底層的Interface 時,可以有2種作法。

第1種作法: 如果Methd中的核心邏輯不會和底層的Interface有相依性,則可將此核心邏輯抽取出來成一個Method,Unit Test就針對這個Method 來測試。例如:在 tenantPage()中,service.getTenant(id)是主要的邏輯且沒有和底層Interface有交互作用,因此可將 service.getTenant(id) 抽取出一個Method 名為getTenant(Integer id)。這時getTenant(Integer id)就很單純 ,只有一個型態為Integer 的id,相對的也比較好進行 Unit Test 。

@RequestMapping(value = "/tenant", method = RequestMethod.POST)
 public String tenantPage(Model model,HttpServletRequest request) {

   Integer id=Integer.valueOf(request.getParameter("id"));
  //Tenant tenant=service.getTenant(id);
   model.addAttribute("tenant", getTenant(id));
  return "tenant";
 }
 
 //new extract method
 public Tenant getTenant(Integer id){
  
  return service.getTenant(id);
 }

第2種作法: 如果核心邏輯無法和底層的Interface做有效切割,這時就只能使用Mock Framework。
利用Mock 來模擬底層物件,而不必自己辛苦去實作這些Interface。有關Mock的測試方法,未來會再詳細述說。以下的測試碼是使用EasyMock Framework,來模擬Model 和 HttpServletRequest。


public class RentControllerTest {

 private RentController rc;
 private RentServiceImpl mockService;

 @Before
 public void setUp() {

  rc = new RentController();
  mockService = createMock(RentServiceImpl.class);
  rc.setService(mockService);
 }

 @Test
 public void testRentController() {

  //test prepare
  String key = "id";
  String idVal = "2";
  String attName = "tenant";  
  Tenant tenant = getTenant();
  HttpServletRequest mockRequest = createMock(HttpServletRequest.class);
  Model mockModel = createMock(Model.class);
  //expect mock behavior
  expect(mockRequest.getParameter(key)).andReturn(idVal);
  expect(mockService.getTenant(Integer.valueOf(idVal))).andReturn(tenant);
  expect(mockModel.addAttribute(attName, tenant)).andReturn(mockModel);
  //replay mock behavior
  replay(mockRequest);
  replay(mockService);
  replay(mockModel);
  //action start
  String pageName = rc.tenantPage(mockModel, mockRequest);
  //verify mock behavior
  verify(mockRequest);
  verify(mockService);
  verify(mockModel);
  //assert test result
  Assert.assertEquals(attName, pageName);
 } 

 private Tenant getTenant() {
  Tenant tenant = new Tenant(Identity.GENERAL);
  tenant.setAge(31);
  tenant.setCareer("IT");
  tenant.setName("John");
  return tenant;
 }
}

使用Mock就可以把複雜的介面切開,而且也可以擺脫環境的限制。上例中的測試碼包含了許多Expect ,其目的是要檢查是否有執行到這段程式碼。所以,Mock就相當於白箱測試,白箱測試會比黑箱測試複雜,你看測試碼是不是變得比較多了呢!