Selenium上机实验

package lab2;

import java.io.*;
import java.util.*;
import java.util.Map.Entry;
import java.util.concurrent.TimeUnit;

import org.openqa.selenium.*;
import org.openqa.selenium.chrome.*;

import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.*;
import org.apache.poi.hssf.usermodel.*;


public class Main {
    public static void main(String[] args) throws Exception {
        //浏览器
        System.setProperty("webdriver.chrome.driver","C:\\Program Files (x86)\\Google\\Chrome\\Application\\chromedriver.exe"); //chromedriver服务地址
        WebDriver driver =new ChromeDriver(); 
        driver.get("https://psych.liebes.top/st"); //打开指定的网站
        //文档
        InputStream is = new FileInputStream("C:\\Users\\input.xlsx");
        Workbook wb = new XSSFWorkbook(is);
        List<Map<String,String>> list = new ArrayList<Map<String,String>>();        
        Sheet sheet = wb.getSheetAt(0); //获取第一个sheet
        Row row = sheet.getRow(0); //获取第一行
        
        String name = null;
        String path = null;
        Map<String,String> map = new LinkedHashMap<String,String>();
        for(int i = 0; i < 96; i++) {
            row = sheet.getRow(i);
            if(row != null){
                name = (String) getCellFormatValue(row.getCell(0));
                path = (String) getCellFormatValue(row.getCell(1));
                map.put(name, path);        
            }else{
                break;
            }
            list.add(map);
        }
        
        for (Entry<String, String> entry : map.entrySet()) {
            String username = entry.getKey();
            String link = entry.getValue();
            driver.findElement(By.id("username")).sendKeys(username); 
            driver.findElement(By.id("password")).sendKeys(username.substring(4,10)); 
            driver.findElement(By.id("submitButton")).click();
            
            String address = driver.findElement(By.xpath("//p")).getText();
            if(address.equals(link)) {
                System.out.println("true");
            } else {
                System.out.println("false");
            }
            
            try {
                driver.manage().timeouts().implicitlyWait(3, TimeUnit.SECONDS);      
                driver.get("https://psych.liebes.top/st");
            } catch (Exception e) {
                e.printStackTrace();
            }    
        }
        
        driver.close(); //退出浏览器
    }
    
    public static Object getCellFormatValue(Cell cell){
        Object cellValue = null;
        if(cell!=null){
            //判断cell类型
            switch(cell.getCellType()){
            case Cell.CELL_TYPE_NUMERIC:{
                cellValue = String.valueOf(cell.getNumericCellValue());
                break;
            }
            case Cell.CELL_TYPE_FORMULA:{
                //判断cell是否为日期格式
                if(DateUtil.isCellDateFormatted(cell)){
                    //转换为日期格式YYYY-mm-dd
                    cellValue = cell.getDateCellValue();
                }else{
                    //数字
                    cellValue = String.valueOf(cell.getNumericCellValue());
                }
                break;
            }
            case Cell.CELL_TYPE_STRING:{
                cellValue = cell.getRichStringCellValue().getString();
                break;
            }
            default:
                cellValue = "";
            }
        }else{
            cellValue = "";
        }
        return cellValue;
    }
}