增加动态参数的实现

Author: LinuxSuRen

src/main/java/org/suren/autotest/web/framework/data/DataSource.java

@@ -1,6 +1,19 @@
-/**
- * http://surenpi.com
+/*
+ * Copyright 2002-2007 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
  */
+
 package org.suren.autotest.web.framework.data;
 
 import org.suren.autotest.web.framework.page.Page;
@@ -10,7 +23,7 @@
  * @author suren
  * @date Jul 17, 2016 8:45:03 AM
  */
-public interface DataSource
+public interface DataSource extends DynamicDataSource
 {
 	/**
 	 * 从数据资源接口中加载数据到Page类中

src/main/java/org/suren/autotest/web/framework/data/ExcelDataSource.java

@@ -10,6 +10,7 @@
 import java.util.Iterator;
 import java.util.LinkedList;
 import java.util.List;
+import java.util.Map;
 import java.util.Set;
 
 import org.apache.poi.ss.usermodel.Cell;
@@ -19,6 +20,8 @@
 import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.config.ConfigurableBeanFactory;
+import org.springframework.context.annotation.Scope;
 import org.springframework.stereotype.Component;
 import org.suren.autotest.web.framework.page.Page;
 
@@ -28,7 +31,8 @@
  * @date Jul 17, 2016 8:56:31 AM
  */
 @Component("excel_data_source")
-public class ExcelDataSource implements DataSource
+@Scope(value = ConfigurableBeanFactory.SCOPE_PROTOTYPE)
+public class ExcelDataSource implements DataSource, DynamicDataSource
 {
 	private static final Logger LOGGER = LoggerFactory.getLogger(ExcelDataSource.class);
 
@@ -118,4 +122,25 @@ public boolean loadData(DataResource resource, int row, Page page)
 		return false;
 	}
 
+	@Override
+	public void setGlobalMap(Map<String, Object> globalMap)
+	{
+		// TODO Auto-generated method stub
+		
+	}
+
+	@Override
+	public Map<String, Object> getGlobalMap()
+	{
+		// TODO Auto-generated method stub
+		return null;
+	}
+
+	@Override
+	public String getName()
+	{
+		// TODO Auto-generated method stub
+		return null;
+	}
+
 }

src/main/java/org/suren/autotest/web/framework/data/XmlDataSource.java

@@ -1,14 +1,29 @@
-/**
- * http://surenpi.com
+/*
+ * Copyright 2002-2007 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
  */
+
 package org.suren.autotest.web.framework.data;
 
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.UnsupportedEncodingException;
 import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
 import java.net.URL;
+import java.net.URLDecoder;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -26,6 +41,8 @@
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.config.ConfigurableBeanFactory;
+import org.springframework.context.annotation.Scope;
 import org.springframework.stereotype.Component;
 import org.suren.autotest.web.framework.core.ui.AbstractElement;
 import org.suren.autotest.web.framework.core.ui.Button;
@@ -42,11 +59,15 @@
  * @date Jul 17, 2016 8:56:51 AM
  */
 @Component("xml_data_source")
-public class XmlDataSource implements DataSource
+@Scope(value = ConfigurableBeanFactory.SCOPE_PROTOTYPE)
+public class XmlDataSource implements DataSource, DynamicDataSource
 {
 	private static final Logger LOGGER = LoggerFactory.getLogger(XmlDataSource.class);
 
+	public static final String NS_URI = "http://datasource.surenpi.com";
+	
 	private Page page;
+	private URL url;
 	private Map<String, Object> globalMap = new HashMap<String, Object>();
 
 	@Autowired
@@ -87,7 +108,7 @@ private void parse(Document doc, int row)
 	{
 		String pageClass = page.getClass().getName();
 		SimpleNamespaceContext simpleNamespaceContext = new SimpleNamespaceContext();
-		simpleNamespaceContext.addNamespace("ns", "http://datasource.surenpi.com");
+		simpleNamespaceContext.addNamespace("ns", NS_URI);
 
 		XPath xpath = new DefaultXPath("/ns:dataSources");
 		xpath.setNamespaceContext(simpleNamespaceContext);
@@ -139,6 +160,7 @@ public void visit(Element node)
 				DynamicData dynamicData = getDynamicDataByType(type);
 				if(dynamicData != null)
 				{
+					dynamicData.setData(globalMap);
 					value = dynamicData.getValue(value);
 				}
 				else
@@ -259,10 +281,12 @@ private DynamicData getDynamicDataByType(String type)
 	public boolean loadData(DataResource resource, int row, Page page)
 	{
 		this.page = page;
-		URL url = null;
-		try {
+		try
+		{
 			url = resource.getUrl();
-		} catch (IOException e) {
+		}
+		catch (IOException e)
+		{
 			LOGGER.error(e.getMessage(), e);
 		}
 
@@ -288,6 +312,7 @@ public boolean loadData(DataResource resource, int row, Page page)
 	/**
 	 * @return the globalMap
 	 */
+	@Override
 	public Map<String, Object> getGlobalMap()
 	{
 		return globalMap;
@@ -296,9 +321,27 @@ public Map<String, Object> getGlobalMap()
 	/**
 	 * @param globalMap the globalMap to set
 	 */
+	@Override
 	public void setGlobalMap(Map<String, Object> globalMap)
 	{
 		this.globalMap = globalMap;
 	}
 
+	@Override
+	public String getName()
+	{
+		try
+		{
+			File file = new File(URLDecoder.decode(this.url.getFile(), "utf-8"));
+			
+			return file.getName();
+		}
+		catch (UnsupportedEncodingException e)
+		{
+			e.printStackTrace();
+		}
+		
+		return this.url.getPath();
+	}
+
 }