webdriver测试testlink demo

JAVA写的WEBDriver测试testlink demo,

package cn.opentest.script;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Set;
import java.util.regex.Pattern;
import java.util.concurrent.TimeUnit;
import org.junit.*;
import static org.junit.Assert.*;
import static org.hamcrest.CoreMatchers.*;
import org.openqa.selenium.*;
import org.openqa.selenium.android.AndroidDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.interactions.Actions;
import org.openqa.selenium.internal.Locatable;
import org.openqa.selenium.support.ui.Select;


public class TestLink {
private static String currentHandle;
private WebDriver driver;
private String baseUrl;
private static String utilPath = "D:\\workspace\\xxxx\\src\\cn\\opentest\\utils\\";
private StringBuffer verificationErrors = new StringBuffer();
@Before
public void setUp() throws Exception {
System.setProperty("webdriver.chrome.driver", "./lib/chromedriver.exe");
//driver = new AndroidDriver();
driver = new ChromeDriver();
baseUrl = "http://localhost:8081/testlink";
driver.get(baseUrl); 
driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS);
}




@Test
public void TestLinkTest() throws Exception {
TestLinkLogin();
//TestLinkAddUser();
//TestLinkReturnHome();
TestLinkRequirementAdd();
TestLinkRequirementUpdate();
TestLinkUpLoadAttached();
//TestLinkDelAttached();
   TestLinkReturnHome();
TestLinkEditTestCases();
} 


public void TestLinkEditTestCases() throws Exception{
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.findElement(By.linkText("Edit Test Cases")).click();
Thread.sleep(2000);
TestLinkTestCaseUpdate();
Thread.sleep(1000);
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.switchTo().frame("workframe");
driver.findElement(By.name("new_testsuite")).click();
Thread.sleep(2000);
driver.findElement(By.name("container_name")).sendKeys("Test Case 1");
driver.switchTo().frame("details___Frame");
driver.switchTo().frame(driver.findElement(By.tagName("iframe")));
driver.findElement(By.xpath("/html/body")).sendKeys("This is test case 1");
Thread.sleep(1000);
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.switchTo().frame("workframe");
driver.findElement(By.name("add_testsuite_button")).click();
}

public void TestLinkDelAttached() throws Exception{
TestLinkRequirementUpdate();
//driver.findElement(By.xpath("//button[contains(@id, 'lnk_Select_')]")); ID变化xpath用法
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.switchTo().frame("workframe");
driver.findElement(By.xpath("//*[@id='ext-gen6']/div/table[2]/tbody/tr[2]/td/a[2]/img")).click();
   Alert alert = getAlert();
   alert.accept();
switchToWindow(driver, "TestLink - Google Chrome");
driver.findElement(By.xpath("//input[@value='Close']")).click();
driver.switchTo().window(currentHandle);
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.switchTo().frame("workframe");
driver.findElement(By.name("deleteSRS")).click();
   driver.findElement(By.id("ext-gen23")).click();
   Thread.sleep(1000);
}

public void TestLinkUpLoadAttached() throws Exception{
        currentHandle = driver.getWindowHandle(); 
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.switchTo().frame("workframe");
driver.findElement(By.xpath("//input[@value='Upload new file']")).click();
Thread.sleep(1000);
switchToWindow(driver, "TestLink - Google Chrome");
driver.findElement(By.name("uploadedFile")).sendKeys("C:\\a.txt");
driver.findElement(By.xpath("//input[@type='submit']")).click();
driver.close();
driver.switchTo().window(currentHandle);
Thread.sleep(1000);
}



public void TestLinkRequirementUpdate() throws Exception{
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.switchTo().frame("treeframe");
driver.findElement(By.xpath("//input[@value='Update tree']")).click();
Thread.sleep(1000);
driver.findElement(By.id("extdd-6")).click();
Thread.sleep(1000);
}


public void TestLinkTestCaseUpdate() throws Exception{
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.switchTo().frame("treeframe");
driver.findElement(By.xpath("//input[@value='Update tree']")).click();
Thread.sleep(1000);
driver.findElement(By.id("extdd-3")).click();
Thread.sleep(1000);
}

public void TestLinkRequirementAdd() throws Exception{
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.findElement(By.linkText("Requirement Specification document")).click();
driver.switchTo().frame("treeframe");
driver.findElement(By.id("extdd-3")).click();
Thread.sleep(1000);
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.switchTo().frame("workframe");
driver.findElement(By.id("new_req_spec")).click();
Thread.sleep(1000);
driver.findElement(By.id("req_spec_title")).clear();
driver.findElement(By.id("req_spec_title")).sendKeys("new requirement 1");
Thread.sleep(3000);
driver.switchTo().frame("scope___Frame");
driver.switchTo().frame(driver.findElement(By.tagName("iframe")));
driver.findElement(By.xpath("/html/body")).sendKeys("This is test case 1");
Thread.sleep(1000);
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.switchTo().frame("workframe");
//RequirementInput();
Thread.sleep(1000);
driver.findElement(By.name("countReq")).sendKeys("1");
driver.findElement(By.name("createSRS")).click();
}

public void TestLinkReturnHome() throws Exception{
driver.switchTo().defaultContent();
driver.switchTo().frame("titlebar");
driver.findElement(By.linkText("Home")).click();
Thread.sleep(1000);
}

public void TestLinkLogin() throws Exception{
assertTrue(driver.getTitle().matches("^TestLink[\\s\\S]*$"));
driver.findElement(By.name("tl_login")).clear();
driver.findElement(By.name("tl_login")).sendKeys("admin");
driver.findElement(By.name("tl_password")).clear();
driver.findElement(By.name("tl_password")).sendKeys("admin");
driver.findElement(By.name("login_submit")).submit();
Thread.sleep(4000);
assertTrue(driver.getTitle().matches("^TestLink 1.8.5[\\s\\S]*$"));
}

public void TestLinkAddUser() throws Exception{
assertTrue(driver.getTitle().matches("^TestLink 1.8.5[\\s\\S]*$"));
String username = "test8";
driver.switchTo().frame("titlebar");
driver.findElement(By.linkText("Users")).click();
driver.switchTo().defaultContent();
driver.switchTo().frame("mainframe");
driver.findElement(By.name("doCreate")).click();
driver.findElement(By.name("login")).clear();
driver.findElement(By.name("login")).sendKeys(username);
driver.findElement(By.name("firstName")).clear();
driver.findElement(By.name("firstName")).sendKeys(username);
driver.findElement(By.name("lastName")).clear();
driver.findElement(By.name("lastName")).sendKeys(username);
driver.findElement(By.name("password")).clear();
driver.findElement(By.name("password")).sendKeys(username);
driver.findElement(By.name("emailAddress")).clear();
driver.findElement(By.name("emailAddress")).sendKeys(username+"@qq.com");
new Select(driver.findElement(By.name("rights_id"))).selectByVisibleText("tester");
new Select(driver.findElement(By.name("locale"))).selectByVisibleText("English (wide/UK)");
driver.findElement(By.name("do_update")).click();
Thread.sleep(3000);
System.out.println(driver.findElement(By.linkText(username)).getText());
assertEquals(username, driver.findElement(By.linkText(username)).getText());
int totalrow = getrow(driver, "//table[@class='simple']/tbody");
int totalcol = getcol(driver, "//table[@class='simple']/tbody");
System.out.println(totalrow +"  "+ totalcol);
}

public static void Upload() throws Exception{
String filename="Upload.vbs";
Runtime.getRuntime().exec("cscript.exe \""+utilPath+filename);
}


public boolean switchToWindow(WebDriver driver,String windowTitle){  
   boolean flag = false;  
   try {  
       String currentHandle = driver.getWindowHandle();  
       Set<String> handles = driver.getWindowHandles();  
       for (String s : handles) {  
           if (s.equals(currentHandle))  
               continue;  
           else {  
               driver.switchTo().window(s);  
               if (driver.getTitle().contains(windowTitle)) {  
                   flag = true;  
                   System.out.println("Switch to window: "  
                           + windowTitle + " successfully!");  
                   break;  
               } else  
                   continue;  
           }  
       }  
   } catch (NoSuchWindowException e) {  
       System.out.println("Window: " + windowTitle + " cound not found!");  
       System.out.println(e.fillInStackTrace());
       flag = false;  
   }  
   return flag;  
}  

    public int getrow(WebDriver driver, String xpath){
        WebElement cellrow=driver.findElement(By.xpath(xpath));
        return cellrow.findElements(By.xpath("tr")).size();
        
    }
    
    public int getcol(WebDriver driver, String xpath){
        WebElement cellrow=driver.findElement(By.xpath(xpath));
        int col = cellrow.findElements(By.xpath("tr/td")).size()/(getrow(driver, xpath)-1);
        return col;        
    }
    
    
    
    private static final String TASKLIST = "tasklist"; 
    private static final String KILL = "taskkill /IM "; 
     
    public static boolean isProcessRunging(String serviceName) throws Exception { 
     
     Process p = Runtime.getRuntime().exec(TASKLIST); 
     BufferedReader reader = new BufferedReader(new InputStreamReader( 
       p.getInputStream())); 
     String line; 
     while ((line = reader.readLine()) != null) { 
     
      System.out.println(line); 
      if (line.contains(serviceName)) { 
       return true; 
      } 
     } 
     
     return false; 
     
    } 
     
    public static void killProcess(String serviceName) throws Exception { 
     
      Runtime.getRuntime().exec(KILL + serviceName); 
     
     } 


    
    //Alert alert = getAlert();
    //alert.accept();//确定
    public Alert getAlert() {
        Alert alert = null;
        for (int i = 0; i < 3; i++) {
            try {
                alert = driver.switchTo().alert();
                if (alert == null) {
                    TimeUnit.SECONDS.sleep(1);
                    continue;
                } 
                
                if(alert.getText()!=null && !"".equals(alert.getClass())){
                    break;
                }
                else{
                    TimeUnit.SECONDS.sleep(1);
                }
                
            } catch (Throwable t) {
                try {
                    TimeUnit.SECONDS.sleep(1);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }
        return alert;
    }



@After
public void tearDown() throws Exception {
driver.quit();
String verificationErrorString = verificationErrors.toString();
if (!"".equals(verificationErrorString)) {
fail(verificationErrorString);
}
}


private boolean isElementPresent(By by) {
try {
driver.findElement(by);
return true;
} catch (NoSuchElementException e) {
return false;
}
}
}