Selenium 处理 modal 对话框（转）

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

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

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

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

Java代码
1.public void clickAndSelectModalDialog(String locator){
2. clickForModalDialog(locator);
3. selenium.selectWindow(“name=modal”);
4.}
5.
6.private void clickForModalDialog(String locator){
7. String overrideShowModalDialogJs=="if(selenium.browserbot.getCurrentWindow().showModalDialog){";
8.
9.overrideShowModalDialogJs += "selenium.browserbot.getCurrentWindow().showModalDialog = function( sURL, vArguments, sFeatures)";
10.
11.overrideShowModalDialogJs +="selenium.browserbot.getCurrentWindow().open(sURL, 'modal', sFeatures);";
12.
13.overrideShowModalDialogJs += "};}";
14.//showModalDialog方法进行覆盖
15.selenium.getEval(overrideShowModalDialogJs);
16.selenium.click(locator);
17.selenium.openWindow(“”,”modal”);
18.selenium.waitForPopUp(“modal”,”15000”);
19.
20.}
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代码
1.public void acceptModalValue(String locator,String[] values){
2. String overrideShowModalDialogJs=="if(selenium.browserbot.getCurrentWindow().showModalDialog){";
3.
4.overrideShowModalDialogJs += "selenium.browserbot.getCurrentWindow().showModalDialog = function( sURL, vArguments, sFeatures)";
5.
6.overrideShowModalDialogJs +="{ "+generateModalDialogReturnObject(values)+”return temp”;
7.
8.overrideShowModalDialogJs += "};}";
9.//showModalDialog方法进行覆盖
10.selenium.getEval(overrideShowModalDialogJs);
11.selenium.click(locator);
12.}
13.
14.private void generateModalDialogReturnObject (String[] values){
15. StringBuffer returnObject=new StringBuffer();
16. returnObject.append(“var temp=new Array();”);
17. for(int i=0;i18. returnObject.append(“temp[”+i+”]=’”+values[i]+”’;”);
19. }
20. return returnObject.toString();
21.}
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 returnObject.append(“temp[”+i+”]=’”+values[i]+”’;”);
}
return returnObject.toString();
}


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

Java代码
1.//点击后弹出部门选择框的图片
2.String depChooseLocator=”…/img”;
3.//点击并选中该弹出网页
4.clickAndSelectModalDialog(depChooseLocator);
5.//执行部门树的操作
6.….
7.//获取该对话框要返回给父页面的值
8.String name=selenium.getEval(“window.seltree.GetAllCheckText();”);
9.String id=selenium.getEval(“window.seltree.GetAllNodeId();”);
10.//关闭部门选择对话框
11.closeModalDialog();
12.//返回父页面
13.Selenium.selectWindow(“name=main95598”);
14.//组装返回值
15.String returnValues=new String[2];
16.returnValues[0]=name;
17.returnValues[1]=id;
18.//父页面获取对话框返回值
19.acceptModalValue(depChooseLocator, returnValues);
//点击后弹出部门选择框的图片
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);


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