章节十二、2-如何封装一个查找单个或多个元素的通用方法

从这一节开始，不讲基本的页面操作了，开始为搭建框架做准备，例如如何封装查找元素的通用方法，这个方法封装好后其它类中都可以使用封装好的这个方法来查找元素，提高代码的复用性，方便后期维护。

一、首先我们需要封装一个能够定位单个或多个元素的类

package usefulmethods;

import java.util.List;

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;

public class GenericMethods {
    
    WebDriver driver;
//    创建一个构成方法，用于初始化对象
    public GenericMethods(WebDriver driver) {
        this.driver = driver;
    }
    
//    查找单个元素
    public WebElement getElement(String type,String locator) {
        if(type.equals("id")) {
            System.out.println("用id查找元素："+locator);
            return this.driver.findElement(By.id(locator));
        }else if(type.equals("xpath")) {
            System.out.println("用xpath查找元素："+locator);
            return this.driver.findElement(By.xpath(locator));
        }else if(type.equals("name")) {
            System.out.println("用name查找元素："+locator);
            return this.driver.findElement(By.name(locator));
        }else if(type.equals("cssSelector")) {
            System.out.println("用cssSelector查找元素："+locator);
            return this.driver.findElement(By.cssSelector(locator));
        }else if(type.equals("cssSelector")) {
            System.out.println("用cssSelector查找元素："+locator);
            return this.driver.findElement(By.cssSelector(locator));
        }else if(type.equals("linkText")) {
            System.out.println("用linkText查找元素："+locator);
            return this.driver.findElement(By.linkText(locator));
        }else if(type.equals("partialLinkText")) {
            System.out.println("用partialLinkText查找元素："+locator);
            return this.driver.findElement(By.partialLinkText(locator));
        }else if(type.equals("tagName")) {
            System.out.println("用tagName查找元素："+locator);
            return this.driver.findElement(By.tagName(locator));
        }else {
            System.out.println("定位的路径不支持");
            return null;
        }
    }
    
//    查找多个元素
    public List<WebElement> getElementList(String type,String locator) {
        if(type.equals("id")) {
            System.out.println("用id查找元素："+locator);
            return this.driver.findElements(By.id(locator));
        }else if(type.equals("xpath")) {
            System.out.println("用xpath查找元素："+locator);
            return this.driver.findElements(By.xpath(locator));
        }else if(type.equals("name")) {
            System.out.println("用name查找元素："+locator);
            return this.driver.findElements(By.name(locator));
        }else if(type.equals("cssSelector")) {
            System.out.println("用cssSelector查找元素："+locator);
            return this.driver.findElements(By.cssSelector(locator));
        }else if(type.equals("cssSelector")) {
            System.out.println("用cssSelector查找元素："+locator);
            return this.driver.findElements(By.cssSelector(locator));
        }else if(type.equals("linkText")) {
            System.out.println("用linkText查找元素："+locator);
            return this.driver.findElements(By.linkText(locator));
        }else if(type.equals("partialLinkText")) {
            System.out.println("用partialLinkText查找元素："+locator);
            return this.driver.findElements(By.partialLinkText(locator));
        }else if(type.equals("tagName")) {
            System.out.println("用tagName查找元素："+locator);
            return this.driver.findElements(By.tagName(locator));
        }else {
            System.out.println("定位的路径不支持");
            return null;
        }    
    }
}

二、对图中的单个输入框进行操作，输入“测试”（图中的页面是本地的网页，小伙伴们如果需要可以加入555191854下载或者找其它的网站进行练习）

package usefulmethods;

import java.util.concurrent.TimeUnit;

import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;

class GenericMethodsDemo {
    private WebDriver driver;
    private String url;
//    创建一个GenericMethods类的引用
    private GenericMethods gm;

    @BeforeEach
    public void setUp() throws Exception {
        driver = new ChromeDriver();
//        实例化查找元素的方法
        gm = new GenericMethods(driver);
        url = "C:\\Users\\acer\\eclipse-workspace\\SeleniumPractise\\PracticePage.html";
        driver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);
        driver.manage().window().maximize();
    }

    @Test
    public void test() {
        driver.get(url);
//        通过调用GenericMethods类中的getElement方法，传入查找元素的方式和路径
        WebElement el = gm.getElement("cssSelector", "#name");
        el.sendKeys("测试");
    }
    
    @AfterEach
    public void tearDown() throws Exception {
        Thread.sleep(2000);
        driver.quit();
    }
}

运行结果为：

三、定位多个输入框

package usefulmethods;

import java.util.List;
import java.util.concurrent.TimeUnit;

import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;

class GenericMethodsDemoList {
    private WebDriver driver;
    private String url;
//    创建一个GenericMethods类的引用
    private GenericMethods gm;

    @BeforeEach
    public void setUp() throws Exception {
        driver = new ChromeDriver();
//        实例化查找元素的方法
        gm = new GenericMethods(driver);
        url = "C:\\Users\\acer\\eclipse-workspace\\SeleniumPractise\\PracticePage.html";
        driver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);
        driver.manage().window().maximize();
    }

    @Test
    public void test() {
        driver.get(url);
//        通过调用GenericMethods类中的getElementList方法，传入查找元素的方式和路径
        List<WebElement> el = gm.getElementList("xpath", "//input[@type='text']");
        int size = el.size();
        System.out.println("获取到的元素长度为："+size);
    }
    
    @AfterEach
    public void tearDown() throws Exception {
        Thread.sleep(2000);
        driver.quit();
    }
}

运行结果：

 注意：sendKeys方法无法同时对多个输入框进行操作。

如果有不明白的小伙伴可以加群“555191854”问我，群里都是软件行业的小伙伴可以相互一起学习讨论。