移植MonkeyRunner的图片对比和获取子图功能的实现-UiAutomator/Robotium篇

根据前一篇文章《移植MonkeyRunner的图片对比和获取子图功能的实现-Appium篇》所述，因为Appium和MonkeyRunner有一个共同点--代码控制流程都是在客户端实现的。所以要把MonkeyRunner在PC端实现的图片比对和获取子图功能移植到同样是在PC端运行的Appium是很容易的事情，但是对于在服务器端运行的Robotium和UiAutomator就是另外一回事了。

因为在Android的sdk中，MonkeyRunner获取子图和图片比对需要用到的以下两个类是没有支持的，简单来说就是java.awt这个库是不支持的：

import java.awt.image.BufferedImage;

import javax.imageio.ImageIO;

但是在Android的sdk中有Bitmap这个类来帮助我们完成类似的功能，同时这个类还提供了一个sameAs的方法来比对两个Bitmap是否一致，但是遗憾的是它没有像MonkeyRunner一样提供一个百分比来指明两个图片的差异接受程度，所以为了兼容多种情况，我们需要对sameAs方法提供多个重载方法。 当然，这只是验证代码，有bug的话自己调吧。

1. 移植代码

注意一下代码只在UiAutomator上面测试通过，但是我相信Robotium是一样的，因为他们都是运行在目标安卓机器上面的，大家可以自行验证下。

package libs;



import java.io.FileInputStream;

import java.io.FileNotFoundException;

import android.graphics.Bitmap;

import android.graphics.BitmapFactory;



public class Util {

	

	public static boolean sameAs (String path1, String path2) throws FileNotFoundException {

		boolean res = false;

		FileInputStream fis1 = new FileInputStream(path1);

		Bitmap bitmap1  = BitmapFactory.decodeStream(fis1);

		

		FileInputStream fis2 = new FileInputStream(path2);

		Bitmap bitmap2  = BitmapFactory.decodeStream(fis2);

		

		res = sameAs(bitmap1,bitmap2);

	

		return res;

			

	}

	

	public static boolean sameAs (String path1, String path2,double percent) throws FileNotFoundException {

		FileInputStream fis1 = new FileInputStream(path1);

		Bitmap bitmap1  = BitmapFactory.decodeStream(fis1);

		

		FileInputStream fis2 = new FileInputStream(path2);

		Bitmap bitmap2  = BitmapFactory.decodeStream(fis2);

		

		return sameAs(bitmap1,bitmap2,percent);

			

	}

	

	public static boolean sameAs (Bitmap bitmap1, Bitmap bitmap2, double percent) {

		if(bitmap1.getHeight() != bitmap2.getHeight())

			return false;

		

		if(bitmap1.getWidth() != bitmap2.getWidth())

			return false;

		

		if(bitmap1.getConfig() != bitmap2.getConfig())

			return false;



		int width = bitmap1.getWidth();

		int height = bitmap2.getHeight();



		int numDiffPixels = 0;

	     

	     for (int y = 0; y < height; y++) {

	       for (int x = 0; x < width; x++) {

	         if (bitmap1.getPixel(x, y) != bitmap2.getPixel(x, y)) {

	           numDiffPixels++;

	         }

	       }

	     }

	     double numberPixels = height * width;

	     double diffPercent = numDiffPixels / numberPixels;

	     return percent <= 1.0D - diffPercent;

	}

	

	public static boolean sameAs (Bitmap bmp1, Bitmap bmp2) throws FileNotFoundException {

		boolean res = false;

		

		res = bmp1.sameAs(bmp2);

		

		return res;		

	}

	

	public static Bitmap getSubImage(String path,int x,int y,int width,int height) throws FileNotFoundException {

		

		FileInputStream fis = new FileInputStream(path);

		Bitmap bitmap  = BitmapFactory.decodeStream(fis);

				

		Bitmap res = Bitmap.createBitmap(bitmap, x, y, width, height);

		

		return res;

		

	}

}

 

2. 调用代码示例

以下是UiAutomator示例，Robotium的示例请大家自行实现.

package sample.demo;



import java.io.File;

import java.io.IOException;

import libs.Util;

import android.graphics.Bitmap;



import com.android.uiautomator.core.UiDevice;

import com.android.uiautomator.core.UiObject;

import com.android.uiautomator.core.UiObjectNotFoundException;

import com.android.uiautomator.core.UiSelector;

import com.android.uiautomator.testrunner.UiAutomatorTestCase;



public class CompareScreenshots extends UiAutomatorTestCase   {

	

	public void testCompareScreenshotsNSubScrenshots() throws UiObjectNotFoundException, IOException, InterruptedException {

		UiDevice device = getUiDevice();

		//device.pressHome();

		UiObject appNotes = new UiObject(new UiSelector().text("Notes"));

		appNotes.click();

		Thread.sleep(3000);

		

		String p1 = "/data/local/tmp/1.bmp";

		String p2 = "/data/local/tmp/2.bmp";

		File f1 = new File(p1);

		if(f1.exists())

			f1.delete();

		

		File f2 = new File(p2);

		if(f2.exists())

			f2.delete();

		

		device.takeScreenshot(f1);

		device.takeScreenshot(f2);

		

		Bitmap sub1 = Util.getSubImage(p1, 6, 39, 474, 38);

		Bitmap sub2 = Util.getSubImage(p2, 6, 39, 474, 38);

		

		boolean same = Util.sameAs(sub1, sub2, 1.0);

		assertTrue(same);

		

		same = Util.sameAs(p1, p2, 0.9);

		assertTrue(same); 

		



	

	}

}

作者	自主博客	微信服务号及扫描码	CSDN
天地会珠海分舵	http://techgogogo.com	服务号:TechGoGoGo扫描码:	http://blog.csdn.net/zhubaitian