基于pywinauto的Windows平台上自动化测试实践（2）

一、利用Jython进行Java和Python的数据交互的局限

在上一节中，利用Jython实现了Java对Python脚本的简单调用，但是jython仅能支持标准的python代码，不支持python扩展包。因而在Java调用的过程中也就堵死了Pywinauto的利用，因此我们需要寻找另一种调用方式。

二、利用Java.lang中的Runtime类调用Python脚本

此方法实际上是利用Runtime类新开进程，来独立调用Python脚本。下面是网上找到的例子：

public String python(String pythonPath, String[] params) {
        File file = new File(pythonPath);
        if (!file.exists()){
            return "python脚本不存在！";
        }

        String[] command = Arrays.copyOf(new String[]{"python", pythonPath}, params.length + 2);
        System.arraycopy(params, 0, command, 2, params.length);

        List<String> res = new ArrayList<>();
        try {
            Process process = Runtime.getRuntime().exec(command, null, null);
            process.waitFor();

            Scanner scanner = new Scanner(process.getInputStream());
            while (scanner.hasNextLine()) {
                String line = scanner.nextLine();
                res.add(line);
            }

        } catch (IOException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        return "success";
    }

例子中，参数pythonPath就是python脚本的绝对路径，参数params是脚本的参数数组，command就是执行python命令字符串数组，格式就是python + 脚本 + 参数，构造好command后传入exec()中执行新进程，然后调用waitFor()函数等待进程结束，结束后从进程的输入流中获得脚本的输出结果存储到字符串数组中。

乍一看，上面的代码并没有问题，对于少量的输出结果执行后相当完美，但是当脚本的输出结果大小大于inputStream缓冲区大小时，程序会阻塞在waitFor()函数这里，问题就在于脚本的输出结果是在进程执行完之后才读取，一个好的解决办法就是新开一个清理线程来不断清空缓冲区，也就是输出和读取同时进行，代码如下：

public String python(String pythonPath, String[] params) {
    File file = new File(pythonPath);
    if (!file.exists()){
        return "python脚本不存在！";
    }

    String[] command = Arrays.copyOf(new String[]{"python", pythonPath}, params.length + 2);
    System.arraycopy(params, 0, command, 2, params.length);

    List<String> res = new ArrayList<>();
    try {
        Process process = Runtime.getRuntime().exec(command, null, null);
        
        ClearThread ct = new ClearThread(process);
        ct.start();
        
        process.waitFor();
        Thread.sleep(1000);
        
        ct.setEnd(true);
        res = ct.getRes();
    } catch (IOException e) {
        e.printStackTrace();
    } catch (InterruptedException e) {
        e.printStackTrace();
    }

    return "success";
}

class ClearThread extends Thread {
    Process process;
    boolean end;
    List<String> res;

    public ClearThread(Process process) {
        this.process = process;
        end = false;
        res = new ArrayList<>();
    }

    @Override
    public void run() {
        if (process == null) {
            return;
        }
        
        Scanner scanner = new Scanner(process.getInputStream());
        while (process != null && !end) {
            while (scanner.hasNextLine()) {
                String line = scanner.nextLine();
                res.add(line);
            }
        }
    }

    public void setEnd(boolean end) {
        this.end = end;
    }

    public List<String> getRes() {
        return res;
    }
}

其中，在脚本执行执行完后调用sleep()让主线程睡眠一秒，否则会导致清理线程可能会还没拿到缓冲区数据就被end标识符结束。

 

三、最终结果

下面就是整个case的具体步骤，也就是最终的完整版本，利用Selenium + Robot 实现从web调用本地应用程序，而后利用Pywinauto实现Windows程序的自动化运行，再将结果反馈到Selenium主程序，进行验证或其他操作。

@Test
public void a_deviceUserActivity() {
    boolean result = false;
    DeviceDetailPage deviceDetailPage = nvBar.openSitesManaged(test).openDevicesDetail("CH-AEM-2012-01");
    deviceDetailPage.openRemoteTakeoverPage();
    super.sleep(20);
    String[] parameter = {};
    List<String> returnData = this.python("C:\Selenium\AEMAgentFramework\aem\framework\Pywin.py", parameter);
    for(String s : returnData){
        System.out.println(s);
    }
    super.sleep(30);
    deviceDetailPage.refreshPage();
    if(returnData.size()==3){
        result = deviceDetailPage.validateRemoteTakeoverActivity(Constants.getEnvDetails().get("username"), 
                returnData.get(0).substring(3), returnData.get(1).substring(3));
    }
    if(result)
        reportPass("CEN-4732 User Acitvity - Remote Takeover function is working");
    else
        reportFailure("CEN-4732 User Acitvity - Remote Takeover is failed");

    Assert.assertTrue(result, "CEN-4732 User Acitvity - Remote Takeover is failed");
}

需要注意的是，在调用本地应用程序的过程中网站通常会利用URI Schema这种方式打开本地应用程序，然而Selenium对这种类似Alert的弹框并不支持，因此需要利用Java的Robot工具实现点击操作

Robot robot;
try {
    robot = new Robot();
    robot.keyPress(java.awt.event.KeyEvent.VK_TAB);
    robot.keyPress(java.awt.event.KeyEvent.VK_TAB);
    robot.keyPress(java.awt.event.KeyEvent.VK_ENTER);
} catch (AWTException e) {
    e.printStackTrace();
}