Selenium 处理 modal 对话框（转）

原文地址：http://www.myexception.cn/open-source/969585.html

Selenium目前没有提供对IE模态对话框(即通过showModalDialog方法打开的弹出对话框)的处理。原因在于，模态对话框会将父页面的JS挂起，直至对话框处理完毕才会继续执行父页面JS。因为Selenium的底层实现是基于JS的，所以模态对话框会同时将selenium挂起，selenium无法选中模态对话框，直至超时。

但是很多系统往往有大量的模态对话框应用。这些应用分为两类：一类是选人、选业务类型等操作，这些操作的目的是填充父页面相应的字段，不会触发其他操作；另外一类比较特殊，例如工作流的派发，这类操作在模态对话框选人返回值后还会继续执行一系列的操作（即showModalDialog方法的调用被夹在一堆JS操作中间）。

对于第一种情况，可以在测试代码里建立相应的数据常量，通过直接赋值的方式避免调用showModalDialog方法；

对于第二种情况，则只能hack。

在当前的测试用例里，采用以下的hack方法：
典型的模态对话框会在点击一个页面元素后打开（例如button），在第一次点击该页面元素时，我们对showModalDialog方法进行覆盖，改打开模态对话框为window.open打开网页，并将selenium选中该弹出网页：

Java代码  
public void clickAndSelectModalDialog(String locator){   
clickForModalDialog(locator);   
selenium.selectWindow(“name=modal”);   
}   

private void  clickForModalDialog(String locator){   
String overrideShowModalDialogJs=="if(selenium.browserbot.getCurrentWindow().showModalDialog){";   
   
   overrideShowModalDialogJs += "selenium.browserbot.getCurrentWindow().showModalDialog = function( sURL, vArguments, sFeatures)";   
 
 overrideShowModalDialogJs +="selenium.browserbot.getCurrentWindow().open(sURL, 'modal', sFeatures);";   
  
 overrideShowModalDialogJs += "};}";   
   //showModalDialog方法进行覆盖   
 selenium.getEval(overrideShowModalDialogJs);   
 selenium.click(locator);   
 selenium.openWindow(“”,”modal”);   
 selenium.waitForPopUp(“modal”,”15000”);   
 
 }  
public void clickAndSelectModalDialog(String locator){
    clickForModalDialog(locator);
     selenium.selectWindow(“name=modal”);
}

private void  clickForModalDialog(String locator){
    String overrideShowModalDialogJs=="if(selenium.browserbot.getCurrentWindow().showModalDialog){";

overrideShowModalDialogJs += "selenium.browserbot.getCurrentWindow().showModalDialog = function( sURL, vArguments, sFeatures)";

overrideShowModalDialogJs +="selenium.browserbot.getCurrentWindow().open(sURL, 'modal', sFeatures);";

overrideShowModalDialogJs += "};}";
//showModalDialog方法进行覆盖
selenium.getEval(overrideShowModalDialogJs);
selenium.click(locator);
selenium.openWindow(“”,”modal”);
selenium.waitForPopUp(“modal”,”15000”);

}  


接下来就可以在弹出网页里进行操作，获取需要返回给父页面的值，这些值一般是一个数组。获取值后关闭弹出网页并返回父页面。在父页面里再次点击打开模态对话框的元素，对showModalDialog方法进行第二次覆盖，将上面获取的值直接返回。

Java代码  
 public void  acceptModalValue(String locator,String[] values){   
   String overrideShowModalDialogJs=="if(selenium.browserbot.getCurrentWindow().showModalDialog){";   
  
 overrideShowModalDialogJs += "selenium.browserbot.getCurrentWindow().showModalDialog = function( sURL, vArguments, sFeatures)";   
 
 overrideShowModalDialogJs +="{ "+generateModalDialogReturnObject(values)+”return temp”;   
 
 overrideShowModalDialogJs += "};}";   
 //showModalDialog方法进行覆盖   
 selenium.getEval(overrideShowModalDialogJs);   
 selenium.click(locator);   
 }   
  
 private void  generateModalDialogReturnObject (String[] values){   
  StringBuffer returnObject=new StringBuffer();   
  returnObject.append(“var temp=new Array();”);   
 for(int i=0;i<values.length;i++){   
     returnObject.append(“temp[”+i+”]=’”+values[i]+”’;”);   
  }   
    return returnObject.toString();   
 }  
public void  acceptModalValue(String locator,String[] values){
    String overrideShowModalDialogJs=="if(selenium.browserbot.getCurrentWindow().showModalDialog){";

overrideShowModalDialogJs += "selenium.browserbot.getCurrentWindow().showModalDialog = function( sURL, vArguments, sFeatures)";

overrideShowModalDialogJs +="{ "+generateModalDialogReturnObject(values)+”return temp”;

overrideShowModalDialogJs += "};}";
//showModalDialog方法进行覆盖
selenium.getEval(overrideShowModalDialogJs);
selenium.click(locator);
}

private void  generateModalDialogReturnObject (String[] values){
    StringBuffer returnObject=new StringBuffer();
    returnObject.append(“var temp=new Array();”);
    for(int i=0;i<values.length;i++){
        returnObject.append(“temp[”+i+”]=’”+values[i]+”’;”);
    }
    return returnObject.toString();
}  


这样就完成了整个过程。看一个示例，该示例打开一个选部门的模态对话框，从部门树里选择部门，然后返回：

Java代码  
 //点击后弹出部门选择框的图片   
 String depChooseLocator=”…/img”;   
 <SPAN style="COLOR: #000000">//点击并选中该弹出网页   
 clickAndSelectModalDialog(depChooseLocator);</SPAN>   
 //执行部门树的操作   
   
 //获取该对话框要返回给父页面的值   
 String name=selenium.getEval(“window.seltree.GetAllCheckText();”);   
 String id=selenium.getEval(“window.seltree.GetAllNodeId();”);   
 <SPAN style="COLOR: #000000">//关闭部门选择对话框   
 closeModalDialog();</SPAN>   
 //返回父页面   
 Selenium.selectWindow(“name=main95598”);   
 //组装返回值   
 String returnValues=new String[2];   
 returnValues[0]=name;   
 returnValues[1]=id;   
 <SPAN style="COLOR: #000000">//父页面获取对话框返回值   
 acceptModalValue(depChooseLocator, returnValues);</SPAN>  
//点击后弹出部门选择框的图片
String depChooseLocator=”…/img”;
//点击并选中该弹出网页
clickAndSelectModalDialog(depChooseLocator);
//执行部门树的操作
….
//获取该对话框要返回给父页面的值
String name=selenium.getEval(“window.seltree.GetAllCheckText();”);
String id=selenium.getEval(“window.seltree.GetAllNodeId();”);
//关闭部门选择对话框
closeModalDialog();
//返回父页面
Selenium.selectWindow(“name=main95598”);
//组装返回值
String returnValues=new String[2];
returnValues[0]=name;
returnValues[1]=id;
//父页面获取对话框返回值
acceptModalValue(depChooseLocator, returnValues);  


可以进一步抽象为模板回调方法，略。