robotframework 图片校验

问题由来

由于某些时候需要校验图片是否为一致，比如一些重要的图标，接口返回的图片地址是否符合要求等

在网络上找了一圈，发现觉大多数都要用到额外的类库，感觉这样比较麻烦不太适合自己，想着原来用md5来判断文件是否一致的情况，那图片也可以用类似的方法，于是想到用base64编码图片来进行比较

具体代码如下：

#encoding=utf8"""通过base64编码的字符来比较两个图片是否一致     作者：Thomas    日期：2015/4/8"""import requestsimport base64from robot.utils.asserts import fail_unlessdef get_url_photo_str(url):    '''参数为图片的网络地址    use example:    ${data}=  | Get Url Photo Str | ${url} |    '''    #yield base64.b64encode(requests.get(url).content)    return base64.b64encode(requests.get(url).content)def get_path_photo_str(path):    '''参数为图片的本地地址    use example:    ${data}=  | Get Path Photo Str | ${path} |    '''    with open(path, "rb") as image_file:        encode_str = base64.b64encode(image_file.read())    #yield encode_str    return encode_strdef path_url_check(path,url):    '''参数为图片的本地地址和网络地址，然后将两者的base64编码进行比较，如果一致就表示两个图片一样    use example:    | Path Url Check | ${path} | ${url} |    '''    fail_unless(_path_url_check(path,url), "local picture is different from url picture ")def url_url_check(url1,url2):    '''参数为两个图片的网络地址，然后将两者的base64编码进行比较，如果一致就表示两个图片一样    use example:    | Url Url Check | ${url1} | ${url2} |    '''    fail_unless(_url_url_check(url1,url2),'the two url picture is different')def _url_url_check(url1,url2):    '''网络图片比较    '''    return True if get_url_photo_str(url1)==get_url_photo_str(url2) \        else Falsedef _path_url_check(path,url):    '''本地图片和网络图片比较    '''    return True if get_path_photo_str(path)==get_url_photo_str(url) \        else False

测试代码：

#!/usr/bin/env python#encoding=utf-8import unittestimport online_photo_checkclass UserInfoTest(unittest.TestCase):    def test_url_url_check(self):        url1 = 'https://assets-cdn.github.com/images/modules/open_graph/github-mark.png'        self.assertTrue(online_photo_check._url_url_check(url1, url1))        url2 = 'https://assets-cdn.github.com/images/modules/open_graph/github-octocat.png'        self.assertFalse(online_photo_check._url_url_check(url1, url2))    def test_path_url_check(self):        path = "c:\\github-octocat.png"        url1 = 'https://assets-cdn.github.com/images/modules/open_graph/github-mark.png'        self.assertFalse(online_photo_check._path_url_check(path, url1))        url2 = 'https://assets-cdn.github.com/images/modules/open_graph/github-octocat.png'        self.assertTrue(online_photo_check._path_url_check(path, url2))