Webdriver 的 PageObject 改造

MilgGtery

PageObject中提供了一个@FindBy注解，也非常好用，但由于其是一次性全部初始化所有的
WebElement，对于当前还不存在于页面上的Element在初始化时就会报错，为了解决这个问
题，自然就不能用这个@FindBy注解了，而我们可以自已去写一个类似的注解来解决这个问
题，下面是思路的实现：

自定义一个@FindBy注解

1. @Retention(RetentionPolicy.RUNTIME)
   
2. @Target({ ElementType.FIELD })
   
3. public @interface FindBy {
   
4. 
   
5.     String name();
   
6. 
   
7.     int age();
   
8. 
   
9.     int lazy() default 0;
   
10. }
    
11. 定义一个TestInfo类来模拟WebElement类
    
12. 
    
13. public class TestInfo {
    
14. 
    
15.     private String name;
    
16. 
    
17.     private int age;
    
18. 
    
19.     public String getName() {
    
20.         return name;
    
21.     }
    
22. 
    
23.     public void setName(String name) {
    
24.         this.name = name;
    
25.     }
    
26. 
    
27.     public int getAge() {
    
28.         return age;
    
29.     }
    
30. 
    
31.     public void setAge(int age) {
    
32.         this.age = age;
    
33.     }
    
34. 
    
35.     @Override
    
36.     public String toString() {
    
37.         return "TestInfo [name=" + name + ", age=" + age + "]";
    
38.     }
    
39. 
    
40. }
    
41. 解析@FindBy注解
    
42. 
    
43. public class PageUtil {
    
44. 
    
45.     private static void initialLazyInfo(Object obj, int lazy) {
    
46.         try{
    
47.             Field[] fields = obj.getClass().getDeclaredFields();
    
48.             for (Field field : fields) {
    
49.                 if(field.isAnnotationPresent(FindBy.class) && field.getType().equals(TestInfo.class)){
    
50.                     FindBy findBy = field.getAnnotation(FindBy.class);
    
51.                     if(findBy.lazy()==lazy){
    
52.                         TestInfo temp = new TestInfo();
    
53.                         temp.setName(findBy.name());
    
54.                         temp.setAge(findBy.age());
    
55.                         field.setAccessible(true);
    
56.                         field.set(obj, temp);
    
57.                     }
    
58.                 }
    
59.             }
    
60.         }catch(Exception e){
    
61.             e.printStackTrace();
    
62.         }
    
63.     }
    
64. 
    
65.     public static void initialLazy(Object obj){
    
66.         PageUtil.initialLazyInfo(obj, 0);
    
67.     }
    
68. 
    
69.     public static void initialLazy(Object obj, int lazy){
    
70.         PageUtil.initialLazyInfo(obj, lazy);
    
71.     }
    
72. 
    
73. }
    
74. 使用方式
    
75. 
    
76. public class DemoPage {
    
77. 
    
78.     public DemoPage() {
    
79.         PageUtil.initialLazy(this);
    
80.     }
    
81. 
    
82.     @FindBy(name="zf1", age=1)
    
83.     private TestInfo info1;
    
84. 
    
85.     @FindBy(name="zf2", age=2, lazy=1)
    
86.     private TestInfo info2;
    
87. 
    
88.     public void test(){
    
89.         System.out.println("info1 is: " + info1);
    
90.         System.out.println("info2 is: " + info2);
    
91.         PageUtil.initialLazy(this, 1);
    
92.         System.out.println("info2 is: " + info2);
    
93.     }
    
94. 
    
95.     public static void main(String[] args) {
    
96.         DemoPage dp = new DemoPage();
    
97.         dp.test();
    
98.     }
    
99. 
    
100. }
     
101. 运行结果
     
102. 
     
103. info1 is: TestInfo [name=zf1, age=1]
     
104. info2 is: null
     
105. info2 is: TestInfo [name=zf2, age=2]

复制代码

说明

将TestInfo初始化进行了分层次的初始化
在需要用到的地方用PageUtil.initialLazy(this, int);进行初始化
以上虽然实现了分层次的初始化，但是在要用到的地方都需要调用PageUtil.initialLazy(this,
int);，还是显得有点麻烦，要解决这个问题，还得从改造WebElement类开始，然后自定义
一个@FindBy注解，并且在需要用到元素的地方再去判断并初始化。

定义@FindBy注解

1. @Retention(RetentionPolicy.RUNTIME)
   
2. @Target({ ElementType.FIELD })
   
3. public @interface FindBy {
   
4. 
   
5.     String id() default "";
   
6. 
   
7.     String name() default "";
   
8. 
   
9.     String className() default "";
   
10. 
    
11.     String css() default "";
    
12. 
    
13.     String tagName() default "";
    
14. 
    
15.     String linkText() default "";
    
16. 
    
17.     String partialLinkText() default "";
    
18. 
    
19.     String xpath() default "";
    
20. }
    
21. 将@FindBy注解中的值结赋到一个中间类中去，在Page类进行初始化的时候进行赋值
    
22. 
    
23. public class FindElement {
    
24. 
    
25.     private String id;
    
26. 
    
27.     private String name;
    
28. 
    
29.     private String className;
    
30. 
    
31.     private String css;
    
32. 
    
33.     private String tagName;
    
34. 
    
35.     private String linkText;
    
36. 
    
37.     private String partialLinkText;
    
38. 
    
39.     private String xpath;
    
40. 
    
41.     public String getId() {
    
42.         return id;
    
43.     }
    
44. 
    
45.     public void setId(String id) {
    
46.         this.id = id;
    
47.     }
    
48. 
    
49.     public String getName() {
    
50.         return name;
    
51.     }
    
52. 
    
53.     public void setName(String name) {
    
54.         this.name = name;
    
55.     }
    
56. 
    
57.     public String getClassName() {
    
58.         return className;
    
59.     }
    
60. 
    
61.     public void setClassName(String className) {
    
62.         this.className = className;
    
63.     }
    
64. 
    
65.     public String getCss() {
    
66.         return css;
    
67.     }
    
68. 
    
69.     public void setCss(String css) {
    
70.         this.css = css;
    
71.     }
    
72. 
    
73.     public String getTagName() {
    
74.         return tagName;
    
75.     }
    
76. 
    
77.     public void setTagName(String tagName) {
    
78.         this.tagName = tagName;
    
79.     }
    
80. 
    
81.     public String getLinkText() {
    
82.         return linkText;
    
83.     }
    
84. 
    
85.     public void setLinkText(String linkText) {
    
86.         this.linkText = linkText;
    
87.     }
    
88. 
    
89.     public String getPartialLinkText() {
    
90.         return partialLinkText;
    
91.     }
    
92. 
    
93.     public void setPartialLinkText(String partialLinkText) {
    
94.         this.partialLinkText = partialLinkText;
    
95.     }
    
96. 
    
97.     public String getXpath() {
    
98.         return xpath;
    
99.     }
    
100. 
     
101.     public void setXpath(String xpath) {
     
102.         this.xpath = xpath;
     
103.     }
     
104. 
     
105. }
     
106. Page类实例化时初始化@FindBy注解并赋值给FindElement类
     
107. 
     
108. public class PageUtil {
     
109. 
     
110.     public static void initialWebElement(Object obj) {
     
111.         try{
     
112.             Field[] fields = obj.getClass().getDeclaredFields();
     
113.             for (Field field : fields) {
     
114.                 if(field.isAnnotationPresent(FindBy.class) && field.getType().equals(WebElement.class)){
     
115.                     FindBy findBy = field.getAnnotation(FindBy.class);
     
116.                     FindElement findElement = new FindElement();
     
117.                     if(!"".equals(findBy.id())){
     
118.                         findElement.setId(findBy.id());
     
119.                     }else if(!"".equals(findBy.name())){
     
120.                         findElement.setName(findBy.name());
     
121.                     }else if(!"".equals(findBy.className())){
     
122.                         findElement.setClassName(findBy.className());
     
123.                     }else if(!"".equals(findBy.css())){
     
124.                         findElement.setCss(findBy.css());
     
125.                     }else if(!"".equals(findBy.tagName())){
     
126.                         findElement.setTagName(findBy.tagName());
     
127.                     }else if(!"".equals(findBy.linkText())){
     
128.                         findElement.setLinkText(findBy.linkText());
     
129.                     }else if(!"".equals(findBy.partialLinkText())){
     
130.                         findElement.setPartialLinkText(findBy.partialLinkText());
     
131.                     }else if(!"".equals(findBy.xpath())){
     
132.                         findElement.setXpath(findBy.xpath());
     
133.                     }
     
134.                     WebElementExt ext = new WebElementExt(findElement);
     
135.                     field.setAccessible(true);
     
136.                     field.set(obj, ext);
     
137.                 }
     
138.             }
     
139.         }catch(Exception e){
     
140.             e.printStackTrace();
     
141.         }
     
142.     }
     
143. 
     
144. }
     
145. 
     

复制代码

MilgGtery

改造WebElement类，利用的是装饰器模式

public class WebElementExt implements WebElement {

private static WebDriver driver = new FirefoxDriver();

private FindElement findElement;

public WebElementExt(FindElement findElement) {
this.findElement = findElement;
}

private WebElement element;

private WebElement getWebElement(){
if(element != null){
return element;
}
if(findElement.getId() != null){
element = this.waitForElement(By.id(findElement.getId()));
}else if(findElement.getName() != null){
element = this.waitForElement(By.name(findElement.getName()));
}else if(findElement.getClassName() != null){
element = this.waitForElement(By.className(findElement.getClassName()));
}else if(findElement.getCss() != null){
element = this.waitForElement(By.cssSelector(findElement.getCss()));
}else if(findElement.getTagName() != null){
element = this.waitForElement(By.tagName(findElement.getTagName()));
}else if(findElement.getLinkText() != null){
element = this.waitForElement(By.linkText(findElement.getLinkText()));
}else if(findElement.getPartialLinkText() != null){
element = this.waitForElement(By.partialLinkText(findElement.getPartialLinkText()));
}else if(findElement.getXpath() != null){
element = this.waitForElement(By.xpath(findElement.getXpath()));
}
if(this.waitElementToBeDisplayed(element)){
return element;
}
return null;
}

private WebElement waitForElement(final By by) {
WebElement element = null;
try {
element = new WebDriverWait(driver, 20).until(new ExpectedCondition<WebElement>() {
public WebElement apply(WebDriver d) {
return d.findElement(by);
}
});
} catch (Exception e) {
e.printStackTrace();
}
return element;
}

private boolean waitElementToBeDisplayed(final WebElement element) {
boolean wait = false;
if (element == null)
return wait;
try {
wait = new WebDriverWait(driver, 20).until(new ExpectedCondition<Boolean>() {
public Boolean apply(WebDriver d) {
return element.isDisplayed();
}
});
} catch (Exception e) {
e.printStackTrace();
}
return wait;
}

@Override
public void click() {
this.getWebElement();
element.click();
}

@Override
public void submit() {
this.getWebElement();
element.submit();
}

@Override
public void sendKeys(CharSequence... charSequences) {
this.getWebElement();
element.sendKeys(charSequences);
}

@Override
public void clear() {
this.getWebElement();
element.clear();
}

@Override
public String getTagName() {
this.getWebElement();
return element.getTagName();
}

@Override
public String getAttribute(String s) {
this.getWebElement();
return element.getAttribute(s);
}

@Override
public boolean isSelected() {
this.getWebElement();
return element.isSelected();
}

@Override
public boolean isEnabled() {
this.getWebElement();
return element.isEnabled();
}

@Override
public String getText() {
this.getWebElement();
return element.getText();
}

@Override
public List<WebElement> findElements(By by) {
this.getWebElement();
return element.findElements(by);
}

@Override
public WebElement findElement(By by) {
this.getWebElement();
return element.findElement(by);
}

@Override
public boolean isDisplayed() {
this.getWebElement();
return element.isDisplayed();
}

@Override
public Point getLocation() {
this.getWebElement();
return element.getLocation();
}

@Override
public Dimension getSize() {
this.getWebElement();
return element.getSize();
}

@Override
public String getCssValue(String s) {
this.getWebElement();
return element.getCssValue(s);
}
}
使用方法

public class DemoPage {

public DemoPage() {
PageUtil.initialWebElement(this);
}

@FindBy(xpath="//abc")
private WebElement element;

public void clieckButton(){
element.click();
}

}
经过上面这样改造后，即便是页面中动态加载的元素，也不用担心会报错了，也不用多次初始化了！