java【selenium】封装查找元素方法，查找单个元素，查找多个元素集合，查找元素是否存在

import java.util.ArrayList;
import java.util.List;

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

/*

• 查找单个元素，或者查找元素集合
  */

public class EncapsulationElement {
WebDriver driver;
//初始化构造函数时，带着WebDriver driver
public EncapsulationElement(WebDriver driver) {
this.driver=driver;

}

//封装单个元素
public WebElement gElement(String locator,String type) {
	//将传入字符串全部转换为小写
	type = type.toLowerCase();
	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("css")) {
		System.out.println("用css查找元素: " + locator);
		return this.driver.findElement(By.cssSelector(locator));
	} else if (type.equals("classname")) {
		System.out.println("用classname查找元素: " + locator);
		return this.driver.findElement(By.className(locator));
	} else if (type.equals("tagname")) {
		System.out.println("用tagname查找元素: " + locator);
		return this.driver.findElement(By.tagName(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 {
		System.out.println("你输入查找元素方式不支持，或属性已更改，请校验");
		return null;
	}
	
	
	
}

//封装集合元素

public List<WebElement> gElementsList(String locator,String type){
	//之前是返回查找元素的集合return this.driver.findElements(By.id(locator)); 直接返回一个集合

// type=type.toLowerCase();
// 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(“css”)) {
// System.out.println("用css查找元素: " + locator);
// return this.driver.findElements(By.cssSelector(locator));
// } else if (type.equals(“classname”)) {
// System.out.println("用classname查找元素: " + locator);
// return this.driver.findElements(By.className(locator));
// } else if (type.equals(“tagname”)) {
// System.out.println("用tagname查找元素: " + locator);
// return this.driver.findElements(By.tagName(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 {
// System.out.println(“定位的路径不支持”);
// return null;
// }
//代码重构
type = type.toLowerCase();
List elementList = new ArrayList();
if (type.equals(“id”)) {
elementList = this.driver.findElements(By.id(locator));
}
else if (type.equals(“name”)) {
elementList = this.driver.findElements(By.name(locator));
}
else if (type.equals(“xpath”)) {
elementList = this.driver.findElements(By.xpath(locator));
}
else if (type.equals(“css”)) {
elementList = this.driver.findElements(By.cssSelector(locator));
}
else if (type.equals(“classname”)) {
elementList = this.driver.findElements(By.className(locator));
}
else if (type.equals(“tagname”)) {
elementList = this.driver.findElements(By.tagName(locator));
}
else if (type.equals(“linktext”)) {
elementList = this.driver.findElements(By.linkText(locator));
}
else if (type.equals(“partiallinktext”)) {
elementList = this.driver.findElements(By.partialLinkText(locator));
}
else {
System.out.println(“定位的类型不支持”);
}
if (elementList.isEmpty()) {
System.out.println(“用 " + type +”: " + locator+” 没有找到元素”);

	} else {
		System.out.println("元素用  " + type +": " + locator+" 找到了");
	}
	return elementList;

	
}
//判断元素是否存在
public boolean IselementIn(String locator ,String type) {
	//判断元素是否存在在页面上
	List<WebElement> elemntList=gElementsList(locator, type);
	int size=elemntList.size();
	if (size>0) {
		System.out.println("查找元素正确");
		return true;
		
		
	}else {
		System.out.println("查找元素失败，请校验数据正确性");

		return false;
		
	}
	
	
	
	
	
}

}