实现sessionStorage注解解析

Author: LinuxSuRen

src/main/java/org/suren/autotest/web/framework/settings/AutoModuleProxy.java

@@ -21,6 +21,9 @@
 import net.sf.cglib.proxy.Enhancer;
 import net.sf.cglib.proxy.MethodInterceptor;
 import net.sf.cglib.proxy.MethodProxy;
+import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.html5.SessionStorage;
+import org.openqa.selenium.html5.WebStorage;
 import org.suren.autotest.web.framework.annotation.AutoExpect;
 import org.suren.autotest.web.framework.annotation.AutoModule;
 import org.suren.autotest.web.framework.annotation.AutoSessionStorage;
@@ -30,10 +33,13 @@
 import org.suren.autotest.web.framework.report.RecordReportWriter;
 import org.suren.autotest.web.framework.report.record.ExceptionRecord;
 import org.suren.autotest.web.framework.report.record.NormalRecord;
+import org.suren.autotest.web.framework.util.PathUtil;
 
+import java.io.*;
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
 import java.util.List;
+import java.util.Properties;
 
 /**
  * 模块代理类
@@ -82,8 +88,10 @@ public Object intercept(Object obj, Method method, Object[] args,
 
         try
         {
+            SessionStorageConfig sessionStorageConfig = new SessionStorageConfig();
             if(autoSessionStorage != null)
             {
+                sessionStorageConfig.setAutoLoad(true);
                 Class<? extends Page> accountClz = autoSessionStorage.accountPage();
                 String accountNameField = autoSessionStorage.accountName();
 
@@ -96,10 +104,12 @@ public Object intercept(Object obj, Method method, Object[] args,
                 if(value instanceof Text)
                 {
                     String accountNameValue = ((Text) value).getValue();
-
-                    System.out.println("accountNameValue:" + accountNameValue);
 
-                    loadSessionStorage(accountNameValue);
+                    if(loadSessionStorage(accountNameValue))
+                    {
+                        sessionStorageConfig.setAccount(accountNameValue);
+                        sessionStorageConfig.setSkipLogin(true);
+                    }
                 }
                 else
                 {
@@ -108,7 +118,19 @@ public Object intercept(Object obj, Method method, Object[] args,
                 }
             }
 
-            result = methodProxy.invokeSuper(obj, args);
+            if(sessionStorageConfig.isSkipLogin())
+            {
+                result = Void.TYPE;
+            }
+            else
+            {
+                result = methodProxy.invokeSuper(obj, args);
+
+                if(sessionStorageConfig.isAutoLoad())
+                {
+                    saveSessionStorage(sessionStorageConfig.getAccount());
+                }
+            }
 
             normalRecord.setEndTime(System.currentTimeMillis());
 
@@ -137,8 +159,58 @@ public Object intercept(Object obj, Method method, Object[] args,
         return result;
     }
 
-    private void loadSessionStorage(String accountNameValue)
+    /**
+     * 保存sessionStorage信息
+     * @param account
+     */
+    private void saveSessionStorage(String account)
+    {
+        WebDriver driver = util.getEngine().getDriver();
+        if(driver instanceof WebStorage)
+        {
+            WebStorage webStorage = (WebStorage) driver;
+            SessionStorage sessionStorage = webStorage.getSessionStorage();
+
+            Properties pro = new Properties();
+            for(String key : sessionStorage.keySet())
+            {
+                pro.setProperty(key, sessionStorage.getItem(key));
+            }
+
+            PathUtil.proStore(pro, "sessionStorage." + account);
+        }
+    }
+
+    /**
+     * 加载sessionStorage信息
+     * @param accountNameValue
+     * @return
+     */
+    private boolean loadSessionStorage(String accountNameValue)
     {
+        WebDriver webDriver = util.getEngine().getDriver();
+        if(webDriver instanceof WebStorage)
+        {
+            WebStorage webStorage = (WebStorage) webDriver;
+            SessionStorage sessionStorage = webStorage.getSessionStorage();
+
+            Properties pro = new Properties();
+            if(PathUtil.proLoad(pro, "sessionStorage." + accountNameValue))
+            {
+                if(pro.isEmpty())
+                {
+                    return false;
+                }
+
+                pro.stringPropertyNames().parallelStream().forEach((key) -> {
+                    sessionStorage.setItem(key, pro.getProperty(key));
+                });
+
+                return true;
+            }
+        }
+
+        return false;
     }
 
     /**

src/main/java/org/suren/autotest/web/framework/settings/SessionStorageConfig.java

@@ -0,0 +1,62 @@
+/*
+ *
+ *  * 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.settings;
+
+/**
+ * @author suren
+ */
+public class SessionStorageConfig
+{
+    /** 跳过登录 */
+    private boolean skipLogin;
+    /** 自动加载 */
+    private boolean autoLoad;
+    /** 账户名 */
+    private String account;
+
+    public boolean isSkipLogin()
+    {
+        return skipLogin;
+    }
+
+    public void setSkipLogin(boolean skipLogin)
+    {
+        this.skipLogin = skipLogin;
+    }
+
+    public boolean isAutoLoad()
+    {
+        return autoLoad;
+    }
+
+    public void setAutoLoad(boolean autoLoad)
+    {
+        this.autoLoad = autoLoad;
+    }
+
+    public String getAccount()
+    {
+        return account;
+    }
+
+    public void setAccount(String account)
+    {
+        this.account = account;
+    }
+}

src/main/java/org/suren/autotest/web/framework/util/PathUtil.java

@@ -3,12 +3,8 @@
  */
 package org.suren.autotest.web.framework.util;
 
-import java.io.File;
-import java.io.FileNotFoundException;
-import java.io.FileOutputStream;
-import java.io.IOException;
-import java.io.InputStream;
-import java.io.OutputStream;
+import java.io.*;
+import java.util.Properties;
 
 import org.apache.commons.io.IOUtils;
 
@@ -17,8 +13,11 @@
  * @author suren
  * @date 2016年12月27日 上午8:12:18
  */
-public class PathUtil
+public abstract class PathUtil
 {
+	/** properties文件后缀 */
+	public static final String PRO_SUFFIX = ".properties";
+
 	/**
 	 * @return 用于保存框架缓存数据的根目录，如果不存在会自动创建
 	 */
@@ -33,6 +32,65 @@ public static File getRootDir()
 
 		return rootFile;
 	}
+
+    /**
+     * @see #proStore(Properties, String, String)
+     * @param pro
+     * @param fileName
+     */
+    public static void proStore(Properties pro, String fileName)
+    {
+        proStore(pro, fileName, "auto generate by phoenix framework, do not modify it.");
+    }
+
+	/**
+	 * 保存Properties文件到框架根目录中
+	 * @param pro 不能为空
+	 * @param fileName 无需带后缀，例如：demo的话，会保存到demo.properties中
+     * @param comment 注释
+     */
+	public static void proStore(Properties pro, String fileName, String comment)
+	{
+		try(OutputStream out = new FileOutputStream(
+				new File(PathUtil.getRootDir(), fileName + PRO_SUFFIX)))
+		{
+			pro.store(out, comment);
+		}
+		catch (FileNotFoundException e)
+		{
+			e.printStackTrace();
+		}
+		catch (IOException e)
+		{
+			e.printStackTrace();
+		}
+	}
+
+	/**
+     * @see #proStore(Properties, String, String)
+     * @param pro 不能为空
+	 * @param fileName
+     * @return 读取成功返回true，否则false
+     */
+	public static boolean proLoad(Properties pro, String fileName)
+	{
+		try(InputStream in = new FileInputStream(new File(PathUtil.getRootDir(), fileName + PRO_SUFFIX)))
+		{
+			pro.load(in);
+
+            return true;
+		}
+		catch (FileNotFoundException e)
+		{
+			e.printStackTrace();
+		}
+		catch (IOException e)
+		{
+			e.printStackTrace();
+		}
+
+        return false;
+	}
 
 	/**
 	 * 如果存在，则跳过，不复制