增加读写sessionStorage的注解

Author: LinuxSuRen

src/main/java/org/suren/autotest/web/framework/annotation/AutoSessionStorage.java

@@ -0,0 +1,43 @@
+/*
+ *
+ *  * 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.annotation;
+
+import org.suren.autotest.web.framework.page.Page;
+
+import java.lang.annotation.*;
+
+/**
+ * 用在登录方法上，第一次登录时，保存session信息；之后登录方法被调用时则会跳过
+ * @author suren
+ */
+@Target(ElementType.METHOD)
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+public @interface AutoSessionStorage
+{
+    /**
+     * @return 用户登录Page类
+     */
+    Class<? extends Page> accountPage();
+
+    /**
+     * @return Page类中代表用户名的属性名称
+     */
+    String accountName() default "userName";
+}

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

@@ -23,10 +23,15 @@
 import net.sf.cglib.proxy.MethodProxy;
 import org.suren.autotest.web.framework.annotation.AutoExpect;
 import org.suren.autotest.web.framework.annotation.AutoModule;
+import org.suren.autotest.web.framework.annotation.AutoSessionStorage;
+import org.suren.autotest.web.framework.core.AutoTestException;
+import org.suren.autotest.web.framework.core.ui.Text;
+import org.suren.autotest.web.framework.page.Page;
 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 java.lang.reflect.Field;
 import java.lang.reflect.Method;
 import java.util.List;
 
@@ -39,11 +44,13 @@ public class AutoModuleProxy implements MethodInterceptor
     private Enhancer enhancer = new Enhancer();
     private Object target;
     private List<RecordReportWriter> recordReportWriters;
+    private SettingUtil util;
 
-    public AutoModuleProxy(Object target, List<RecordReportWriter> recordReportWriters)
+    public AutoModuleProxy(Object target, List<RecordReportWriter> recordReportWriters, SettingUtil util)
     {
         this.target = target;
         this.recordReportWriters = recordReportWriters;
+        this.util = util;
     }
 
     public Object getProxy()
@@ -64,6 +71,7 @@ public Object intercept(Object obj, Method method, Object[] args,
         Class<?> superCls = obj.getClass().getSuperclass();
         AutoModule autoModule = superCls.getAnnotation(AutoModule.class);
         AutoExpect autoExpect = method.getAnnotation(AutoExpect.class);
+        AutoSessionStorage autoSessionStorage = method.getAnnotation(AutoSessionStorage.class);
 
         NormalRecord normalRecord = new NormalRecord();
         normalRecord.setBeginTime(beginTime);
@@ -74,6 +82,32 @@ public Object intercept(Object obj, Method method, Object[] args,
 
         try
         {
+            if(autoSessionStorage != null)
+            {
+                Class<? extends Page> accountClz = autoSessionStorage.accountPage();
+                String accountNameField = autoSessionStorage.accountName();
+
+                Page page = util.getPage(accountClz);
+                Field accountField = accountClz.getDeclaredField(accountNameField);
+
+                accountField.setAccessible(true);
+                Object value = accountField.get(page);
+
+                if(value instanceof Text)
+                {
+                    String accountNameValue = ((Text) value).getValue();
+
+                    System.out.println("accountNameValue:" + accountNameValue);
+                    
+                    loadSessionStorage(accountNameValue);
+                }
+                else
+                {
+                    throw new AutoTestException("Wrong account type in class: " + accountClz + ", and field : "
+                            + accountNameField + ". It should be Text type.");
+                }
+            }
+
             result = methodProxy.invokeSuper(obj, args);
 
             normalRecord.setEndTime(System.currentTimeMillis());
@@ -103,6 +137,10 @@ public Object intercept(Object obj, Method method, Object[] args,
         return result;
     }
 
+    private void loadSessionStorage(String accountNameValue)
+    {
+    }
+
     /**
      * 根据注解配置，是否要对异常进行处理
      * @param autoExpect

src/main/java/org/suren/autotest/web/framework/spring/AutoModuleScope.java

@@ -21,7 +21,9 @@
 import org.springframework.beans.factory.ObjectFactory;
 import org.springframework.beans.factory.config.Scope;
 import org.springframework.context.annotation.ScopedProxyMode;
+import org.suren.autotest.web.framework.annotation.AutoSessionStorage;
 import org.suren.autotest.web.framework.core.EngineAware;
+import org.suren.autotest.web.framework.page.Page;
 import org.suren.autotest.web.framework.report.RecordReportWriter;
 import org.suren.autotest.web.framework.selenium.WebDriverAware;
 import org.suren.autotest.web.framework.settings.AutoModuleProxy;
@@ -56,7 +58,7 @@ public Object get(String name, ObjectFactory<?> objectFactory)
         if(object == null)
         {
             object = objectFactory.getObject();
-            AutoModuleProxy proxy = new AutoModuleProxy(object, recordReportWriters);
+            AutoModuleProxy proxy = new AutoModuleProxy(object, recordReportWriters, util);
             object = proxy.getProxy();
             putAware(object);