學習TDD(2)--實例：ProtoStuffUtil類的測試

2019-11-11 05:11:03

字體：大中小

來源：轉載

供稿：網友

上篇講了TDD的步驟和各種好處，俗話說的好，站在岸上是學不會游泳的。所以我們還是要拿個例子出來，實踐一下TDD。

因為是第一次嘗試，我想還是選個簡單的例子，之前寫的那個PRotoStuffUtil類就很不錯。這個類主要負責對象跟byte[]之間的相互轉換。可以參考http://blog.csdn.net/mrbcy/article/details/54869113。其實這個類已經寫好了，不太符合TDD的規范。但是體驗一下還是可以的。

配套的代碼已經上傳到http://download.csdn.net/detail/mrbcy/9748501

這個類雖然簡單但是測試的流程還是很曲折

測試目標

因為這個類是負責對象跟byte[]之間的相互轉換，所以我想從兩個方面測試它：

第一個是能夠對復雜的對象進行正確的編碼解碼第二個是對復雜對象構成的List、Map進行正確的編碼解碼

測試代碼

來看對象類代碼

package tech.mrbcy.mrpc.test.domain;import java.util.ArrayList;import java.util.Collection;import java.util.HashMap;import java.util.List;import java.util.Map;import com.sun.org.apache.bcel.internal.generic.NEW;import tech.mrbcy.mrpc.test.enumm.UserType;public class User { private int userId; private String userName; private boolean lockState; private UserType userType; private List<String> addresses = new ArrayList<String>(); private Map<String, String> favoriteMap = new HashMap<String, String>(); // getters and setters public void addAddress(String address){ addresses.add(address); } public void putFavor(String key,String value){ favoriteMap.put(key, value); } @Override public String toString() { return "User [userId=" + userId + ", userName=" + userName + ", lockState=" + lockState + ", userType=" + userType + ", addresses=" + addresses + ", favoriteMap=" + favoriteMap + "]"; }}

看起來對象很復雜了，各種的數據類型和list map也都用上去了。接下來我們編寫第一個測試。看代碼。

@Test// 對復雜對象進行解碼編碼public void testObject(){ User user = new User(); user.setUserId(10086); user.setUserName("張三"); user.setLockState(true); user.setUserType(UserType.Vip_USER); user.addAddress("上海"); user.addAddress("北京"); user.putFavor("tdd", "當當網"); user.putFavor("java","Amazon"); // 保存轉換之前的toString結果 String oldString = user.toString(); // 轉換 byte[] data = ProtostuffUtil.serializer(user); User newUser = ProtostuffUtil.deserializer(data, User.class); // 保存轉換之后的toString結果 String newString = newUser.toString(); assertEquals(oldString,newString);}

執行結果是通過。

然后編寫第二個測試。

@Test// 對復雜對象的列表進行編碼解碼public void testList(){ List<User> users = new ArrayList<User>(); User user = new User(); user.setUserId(10086); user.setUserName("張三"); user.setLockState(true); user.setUserType(UserType.VIP_USER); user.addAddress("上海"); user.addAddress("北京"); user.putFavor("tdd", "當當網"); user.putFavor("java","Amazon"); users.add(user); User user2 = new User(); user2.setUserId(10086); user2.setUserName("張三"); user2.setLockState(true); user2.setUserType(UserType.VIP_USER); user2.addAddress("上海"); user2.addAddress("北京"); user2.putFavor("tdd", "當當網"); user2.putFavor("java","Amazon"); users.add(user); // 保存轉換之前的toString結果 String oldString = users.toString(); // 轉換 byte[] data = ProtostuffUtil.serializer(users); List<User> newUsers = ProtostuffUtil.deserializer(data, users.getClass()); // 保存轉換之后的toString結果 for(User u : newUsers){ System.out.println(u); } String newString = newUsers.toString(); assertEquals(oldString,newString);}

看到重復代碼出現了，暫時不管它，重要的是先讓測試通過。但是，執行結果是失敗。

報出的錯誤是并發操作List時常出現的錯誤，就是通過外部強行修改了List的內部狀態導致的。

我非常的不解，難道對象的List不能和byte[]相互轉換？把List包到對象里面試試看。

@Test// 把List包到對象里進行編碼解碼public void testUserPack(){ List<User> users = new ArrayList<User>(); User user = new User(); user.setUserId(10086); user.setUserName("張三"); user.setLockState(true); user.setUserType(UserType.VIP_USER); user.addAddress("上海"); user.addAddress("北京"); user.putFavor("tdd", "當當網"); user.putFavor("java","Amazon"); users.add(user); User user2 = new User(); user2.setUserId(10086); user2.setUserName("張三"); user2.setLockState(true); user2.setUserType(UserType.VIP_USER); user2.addAddress("上海"); user2.addAddress("北京"); user2.putFavor("tdd", "當當網"); user2.putFavor("java","Amazon"); users.add(user); UserListPack ulp = new UserListPack(); ulp.setUsers(users); // 保存轉換之前的toString結果 String oldString = ulp.toString(); // 轉換 byte[] data = ProtostuffUtil.serializer(ulp); UserListPack newUlp = ProtostuffUtil.deserializer(data, UserListPack.class); // 保存轉換之后的toString結果 String newString = newUlp.toString(); assertEquals(oldString,newString);}

UserListPack.java

public class UserListPack { List<User> users; public List<User> getUsers() { return users; } public void setUsers(List<User> users) { this.users = users; } @Override public String toString() { return "UserListPack [users=" + users + "]"; }}

執行的結果是成功。難道ProtoStuff不能轉換List Map這樣的數據？于是我又寫了一個Map的測試。

@Testpublic void testMap(){ Map<Integer, User> uMap = new HashMap<Integer, User>(); User user = new User(); user.setUserId(10086); user.setUserName("張三"); user.setLockState(true); user.setUserType(UserType.VIP_USER); user.addAddress("上海"); user.addAddress("北京"); user.putFavor("tdd", "當當網"); user.putFavor("java","Amazon"); uMap.put(1,user); User user2 = new User(); user2.setUserId(10086); user2.setUserName("張三"); user2.setLockState(true); user2.setUserType(UserType.VIP_USER); user2.addAddress("上海"); user2.addAddress("北京"); user2.putFavor("tdd", "當當網"); user2.putFavor("java","Amazon"); uMap.put(2,user); // 保存轉換之前的toString結果 String oldString = uMap.toString(); // 轉換 byte[] data = ProtostuffUtil.serializer(uMap); Map<Integer, User> newMap = ProtostuffUtil.deserializer(data, uMap.getClass()); // 保存轉換之后的toString結果 String newString = newMap.toString(); assertEquals(oldString,newString);}

結果失敗了。

注意畫紅框的那里，說明轉換出來的Map是空的。看來ProtoStuff真的不能從byte[]轉List Map了。

我又去搜了一下Java byte[] to List也沒有好的解決辦法。

——更新：Google找到了————————–

ObjectInputStream ois = new ObjectInputStream(new ByteArrayInputStream(bytes));try { @SuppressWarnings("unchecked") ArrayList<Object> list = (ArrayList<Object>) ois.readObject(); ...} finally { ois.close();}

對應的List to byte[]代碼是：

ByteArrayOutputStream bos = new ByteArrayOutputStream();ObjectOutputStream oos = null;oos = new ObjectOutputStream(bos);oos.writeObject(mArrayList);//mArrayList is the array to convertbyte[] buff = bos.toByteArray();

這個代碼我還沒試。后續看看要不要干脆拋棄ProtoStuff，用上面的方法進行Object和byte[]互轉算了。

————————————————-

事到如今我知道ProtoStuffUtil只能轉Object，所以只能先修改測試的代碼，允許它拋異常，讓測試先過掉了。

testMap的最后一句斷言修改為assertEquals("{}",newString);

testList的@Test注解修改為@Test(expected = ConcurrentModificationException.class)

這樣測試就都通過了。

測試重構

這4個測試中的重復代碼主要集中在User對象的創建及屬性賦值和轉換并比較前后字符串這兩部分。

創建User的函數，只修改id和userName就足夠了。

private User createUser(Integer userId, String userName){ User user = new User(); user.setUserId(userId); user.setUserName(userName); user.setLockState(true); user.setUserType(UserType.VIP_USER); user.addAddress("上海"); user.addAddress("北京"); user.putFavor("tdd", "當當網"); user.putFavor("java","Amazon"); return user;}

執行比較的函數

private void doCompare(Object oldObj){ // 保存轉換之前的toString結果 String oldString = oldObj.toString(); doCompare(oldObj, oldString);}private void doCompare(Object oldObj,String expectStr){ // 轉換 byte[] data = ProtostuffUtil.serializer(oldObj); Object newObj = ProtostuffUtil.deserializer(data, oldObj.getClass()); // 保存轉換之后的toString結果 String newString = newObj.toString(); assertEquals(expectStr,newString);}

重構以后的測試代碼變得非常精簡。

@Test// 對復雜對象進行解碼編碼public void testObject(){ User user = createUser(10086, "張三"); doCompare(user);}

重構后4個測試依然是通過的。