增加cookie

Author: LinuxSuRen

CHANGELOG

@@ -1,4 +1,5 @@
 # 1.0.2-20170325
 * Selenium的依赖版本升级到3.0
 * jre的依赖版本升级到1.8
-* 增加系统参数配置文件system.data.properties
+* 增加系统参数配置文件system.data.properties
+* 增加cookie的保存和加载

src/main/java/org/suren/autotest/web/framework/selenium/SeleniumEngine.java

@@ -29,6 +29,8 @@
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.ObjectInputStream;
+import java.io.ObjectOutputStream;
 import java.io.OutputStream;
 import java.io.UnsupportedEncodingException;
 import java.net.MalformedURLException;
@@ -40,15 +42,18 @@
 import java.util.Iterator;
 import java.util.Map;
 import java.util.Properties;
+import java.util.Set;
 import java.util.concurrent.TimeUnit;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipInputStream;
 
+import org.openqa.selenium.Cookie;
 import org.openqa.selenium.Dimension;
 import org.openqa.selenium.JavascriptExecutor;
 import org.openqa.selenium.Proxy;
 import org.openqa.selenium.UnsupportedCommandException;
 import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.WebDriver.Options;
 import org.openqa.selenium.WebDriver.Window;
 import org.openqa.selenium.chrome.ChromeDriver;
 import org.openqa.selenium.chrome.ChromeOptions;
@@ -195,6 +200,36 @@ else if(DRIVER_OPERA.equals(curDriverStr))
 			}
 		}
 
+		// 处理cookie
+		Options manage = driver.manage();
+		boolean cookieLoad = Boolean.parseBoolean(enginePro.getProperty("cookie.load", "false"));
+		File root = PathUtil.getRootDir();
+		File cookieFile = new File(root, enginePro.getProperty("cookie.save.path", "phoenix.autotest.cookie"));
+		
+		if(cookieLoad)
+		{
+			try(ObjectInputStream input = new ObjectInputStream(new FileInputStream(cookieFile)))
+			{
+				Object cookiesObj = input.readObject();
+				if(cookiesObj != null && cookiesObj instanceof Set<?>)
+				{
+					Set<Cookie> cookies =  (Set<Cookie>) cookiesObj;
+					for(Cookie cookie : cookies)
+					{
+						manage.addCookie(cookie);
+					}
+				}
+			}
+			catch (IOException e)
+			{
+				e.printStackTrace();
+			}
+			catch (ClassNotFoundException e)
+			{
+				e.printStackTrace();
+			}
+		}
+		
 		if(maximize)
 		{
 			window.maximize();
@@ -651,6 +686,24 @@ public void close()
 	{
 		if(driver != null)
 		{
+			Options manage = driver.manage();
+			boolean cookieSave = Boolean.parseBoolean(enginePro.getProperty("cookie.save", "false"));
+			File root = PathUtil.getRootDir();
+			File cookieFile = new File(root, enginePro.getProperty("cookie.save.path", "phoenix.autotest.cookie"));
+			
+			if(cookieSave)
+			{
+				Set<Cookie> cookies = manage.getCookies();
+				try(ObjectOutputStream output = new ObjectOutputStream(new FileOutputStream(cookieFile)))
+				{
+					output.writeObject(cookies);
+				}
+				catch (IOException e)
+				{
+					e.printStackTrace();
+				}
+			}
+			
 			driver.quit();
 		}
 	}

src/main/resources/engine.properties

@@ -38,8 +38,13 @@ android.platformVersion=4.3
 #android.appPackage=
 #android.appActivity=
 
+##default invoker package
 invoker.package=org.suren.autotest.web.framework.invoker
 
+cookie.save=false
+cookie.save.path=phoenix.autotest.cookie
+cookie.load=false
+
 upload.file.random.name=suren.png
 
 ##data source stuff