单元测试数据初始化--数组

论坛权限方法测试，初始化数组数据。数据我们一般是存放到数据库的，但是单元测试并且是在测服务层，我们一般不连数据库的。
自己简单的模拟数据出来又不是很适合，而且重用性低。这时我们可以把数据库的数据搬过来使用，以xml文件的格式

public class UserRightServiceTest {
    final static UserRightService userRightService = new UserRightService();
    private static Mockery context = new JUnit4Mockery() {{
	setImposteriser(ClassImposteriser.INSTANCE);
    }};
    private static States state = context.states("test");
    static UserGroupRepository userGroupRepository;
    static LockService lockService;
    static UserGroup[] userGroups;

	/**
	 * 初始化数组
	 * @throws DocumentException
	 */
    @BeforeClass
    public static void setUp() throws DocumentException {
		userGroupRepository = context.mock(UserGroupRepository.class);
		lockService = context.mock(LockService.class);
		//初始化用户组数据
		userGroups = initUserGroups("bbs6_usergroup.xml", "bbs6_usergroup_item.xml");	
		userRightService.userGroupRepository = userGroupRepository;
		userRightService.lockService = lockService;
		userRightService.userGroups = userGroups;
    }
	public static UserGroup[] initUserGroups(String sourceFilePath, String sourceFileItemPath) throws DocumentException {

		XMLParser items = new XMLParser(sourceFileItemPath);
		Iterator iterItems = items.getIterator("row");
		Map<Long, Map<String, Long>> itemMap = new HashMap<Long, Map<String, Long>>();
		Element ele = null;

		while (iterItems.hasNext()) {
			ele = (Element)iterItems.next();
			long groupId = NumberUtils.toLong(ele.elementText("usergroupid"));
			if (itemMap.containsKey(groupId)) {
				itemMap.get(groupId).put(ele.elementText("name"), NumberUtils.toLong(ele.elementText("value")));
			} else {
				Map<String, Long> tempMap = new HashMap<String, Long>();
				tempMap.put(ele.elementText("name"), NumberUtils.toLong(ele.elementText("value")));
				itemMap.put(groupId, tempMap);
			}
		}

		XMLParser iData = new XMLParser(sourceFilePath);
		Iterator iter = iData.getIterator("row");
		List<UserGroup> userGroupsList = new ArrayList<UserGroup>(iData.getLength());
		
		while (iter.hasNext()) {
			UserGroup ug = new UserGroup();
			ele = (Element)iter.next();
			long groupId = NumberUtils.toLong(ele.elementText("usergroupid"));
			ug.setUserGroupId(groupId);
			ug.setType(NumberUtils.toInt(ele.elementText("type")));
			ug.setName(ele.elementText("name"));
			ug.setLimitMin(NumberUtils.toInt(ele.elementText("limitmin")));
			ug.setItems(itemMap.get(groupId));
			userGroupsList.add(ug);
		}

		return userGroupsList.toArray(new UserGroup[0]);
	}


	
}

class XMLParser {
	private Document doc = null;
	public XMLParser() {}
	public XMLParser(String sourceFilePath) throws DocumentException {
		SAXReader reader = new SAXReader();
		doc = reader.read(UserRightServiceTest.class.getClassLoader().getResourceAsStream(sourceFilePath));
	}

	public Iterator getIterator(String nodeText) {
		return doc.getRootElement().elementIterator(nodeText);
	}

	public int getLength() {
		return doc.getRootElement().elements().size();
	}
}

• XMLParser是用来读去并解析xml文件的
• initUserGroups方法是用来初始化我们需要的数据的

多个测试方法都可以重用这个数组的了