Java单元测试初试

在项目中，要负责编写一个用于存储数据的字典，刚开始写，为了便于测试代码正确性，在编写初期，确定了一定要在写方法的同时，编写单元测试。因为在负责前面别人编写的一个项目的维护时，发现了蛮多低级的错误，并且我没有参与此项目的完成，费了很多时间去梳理程序发现bug，因此觉得自己编写时，一定要做好测试这关。因此尝试了一番Junit的使用。如下：

先编写了一个简单的数据字典类，如下。按照需求，后期还会添加从数据库表中读取来获得字典，从hbase获取字典，以及输出字典到文件，

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.PrintWriter;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Scanner;
import java.util.Map.Entry;

public class IntValueDictionary{
private HashMap<String, Integer> dic = new HashMap<String, Integer>();

public IntValueDictionary(){}
/**
* Construct dictionary from file in the disk.
* @param path The path of the source file.
* If the line doesn't  contains a ',', the value will be set as 1000.
*/
public IntValueDictionary(String path){
try {
FileReader reader = new FileReader(new File(path));
Scanner in = new Scanner(reader);
String line = null;
String str = null;
int num = 0;
boolean singleType = true;
while(in.hasNextLine()){
line = in.nextLine();
//				System.out.println(line);
if (singleType&&line.contains(",")) {
singleType = false;
}
if (singleType == true) {
str = line;
num = 1000;
} else {
String[] entry = line.split(",");
str = entry[0];
num = Integer.parseInt(entry[1]);
}
dic.put(str, num);
}
in.close();
} catch (FileNotFoundException e) {
e.printStackTrace();
}
}
/**
* Construct dictionary from HashMap.
*/
public IntValueDictionary(HashMap<String, Integer> dic){
this.dic = dic;
}
/**
* Return a dictionary.
*/
public HashMap<String, Integer> getDic(){
return dic;
}
/**
* Check whether the source string contains any word in the dictionary.
* @param str The source string.
* @return true if source string contains at least one word in the dictionary; otherwise false.
*/
public boolean isContainedIn(String str){
if(str == null)
return false;
Iterator<Entry<String, Integer>> iter = dic.entrySet().iterator();
while(iter.hasNext()){
Entry<String, Integer> tempEntry = iter.next();
if(str.contains(tempEntry.getKey()))
return true;
}
return false;
}

public boolean containsKey(String key) {
return dic.containsKey(key);
}

public int getValue(String key) {
if(dic.get(key) == null)
return 0;
return dic.get(key);
}
/**
* Add item to the dictionary.
* @param key The key.
* @param value The value.
* @return If the key exists,then return false.
*/
public boolean addItem(String key,Integer value) {
if(dic.get(key) == null){
dic.put(key, value);
return true;
}
return false;
}
/**
* Add item to the dictionary,If the key exists,then the value will be rewritten.
* @param key The key.
* @param value The value.
*/
public void add(String key,Integer value) {
dic.put(key, value);
}

}

新建一个测试类，继承testcase，重新其中的方法，添加测试用例如下：

import udb.eda.dictionary.IntValueDictionary;
import junit.framework.TestCase;

public class IntValueDictionaryTest extends TestCase{
IntValueDictionary intValueDic;
//此方法在执行每一个测试方法之前（测试用例）之前调用
@Override
protected void setUp() throws Exception
{
// TODO Auto-generated method stub
super.setUp();
intValueDic = new IntValueDictionary("E:\\myjava\\test\\123.txt");
//System.out.println("setUp()");
}
//此方法在执行每一个测试方法之后调用
@Override
protected void tearDown() throws Exception
{
// TODO Auto-generated method stub
super.tearDown();
//System.out.println("tearDown()");
}
//测试用例
public void testContainsKey()
{
assertEquals(true, intValueDic.containsKey("a"));
assertEquals(true, intValueDic.containsKey("d"));
assertEquals(true, intValueDic.containsKey("i"));
assertEquals(false, intValueDic.containsKey("y"));
System.out.println("testContainsKey()");
}
//测试用例
public void testGetValue()
{
assertEquals(1, intValueDic.getValue("a"));
assertEquals(4, intValueDic.getValue("d"));
assertEquals(9, intValueDic.getValue("i"));
assertEquals(0, intValueDic.getValue("y"));
System.out.println("testGetValue()");
}
//测试用例
public void testIsContainedIn()
{
assertEquals(true, intValueDic.isContainedIn("kkemm"));
assertEquals(true, intValueDic.isContainedIn("ihpp"));
assertEquals(true, intValueDic.isContainedIn("ii"));
assertEquals(true, intValueDic.isContainedIn("ppl"));
assertEquals(false, intValueDic.isContainedIn("z"));
System.out.println("testIsContainedIn()");
}
public void testAddItem()
{
assertEquals(true, intValueDic.addItem("xy", 22));
assertEquals(22, intValueDic.getValue("xy"));
assertEquals(false, intValueDic.addItem("k", 100));
assertEquals(11, intValueDic.getValue("k"));
System.out.println("testAddItem()");
}
public void testAdd()
{
intValueDic.add("k", 33);
assertEquals(33, intValueDic.getValue("k"));
intValueDic.add("i", 33);
assertEquals(33, intValueDic.getValue("i"));
System.out.println("testAdd()");
}
}

run中选择junit test，即可执行测试。