2007-05-13

package com.softleader.unittest;
public class UserDAOImpl implements UserDAO{
	public UserDAOImpl(){
		// 連結資料庫
	}
	public void createUser(User user) {
		// insert into user ( id, name ) values ( $id, $name );
	}
	public User getUser(long id) {
		// select * from user where id = $id;
		return null;
	}
	public void removeUser(long id) {
		// delete user where id = $id
	}
	public void setUser(User user) {
		// update user set name = $name where id = $id
	}	
}

package com.softleader.unittest;
import org.apache.commons.lang.builder.ToStringBuilder;
public class User {
	private long id;
	private String name;
	public long getId() {
		return id;
	}
	public void setId(long id) {
		this.id = id;
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public String toString() {
		return ToStringBuilder.reflectionToString(this);
	}
}

public User getUser(long id) {
		User user = new User();
		user.setId(1);
		user.setName("jini");
		return user;
	}

package com.softleader.unittest;
public class UserBusinessDelegate {
	private UserDAO dao;
	public void setDAO(UserDAO dao) {
		this.dao = dao;
	}
	public void deleteUser(long id) {
		dao.removeUser(id);
	}
	public User getUser(long id) {
		return dao.getUser(id);
	}
	public void insertUser(User user) {
		dao.createUser(user);
	}
	public void updateUser(User user) {
		dao.setUser(user);
	}
}

package com.softleader.unittest;
import org.jmock.Expectations;
import org.jmock.Mockery;
import org.jmock.integration.junit4.JUnit4Mockery;
import org.junit.Test;
public class UserBussinessDelegateTest {
	// Mockery 初始化
	Mockery context = new JUnit4Mockery();
    @Test 
    public void testGetUser() {
    	// 利用 context 呼叫 UserDAO interface 產生模擬物件
    	final UserDAO dao = context.mock(UserDAO.class);
    	// 將 模擬 DAO 放到 Delegate 之中
    	UserBusinessDelegate delegate = new UserBusinessDelegate();
    	delegate.setDAO(dao);
    	// 建立一個模擬物件
    	final User testuser = new User();
    	testuser.setId(1);
    	testuser.setName("jini");
        // 設定期望的處理情況
        context.checking(new Expectations() {{
            one (dao).getUser(1);            
            will(returnValue(testuser));
        }});
    	System.out.println(delegate.getUser(1));
    }
}