如何获取注册过程图片上的验证码?并实现自动化测试。

一、如果是生产环境,手动测试
二、
如果是测试环境,4种方法
1、利用python图形识别库pytesseract里面的函数Image to string,然后封装在python根目录下的site-packages里面,重载robotframework,例如:

import pytesseract
from PIL import Image 
def GetImageORC(self, filepath):
"""获取图片中的字符,并返回字符串,例:| Get Image ORC | ${file} |"""
self.filepath = filepath
img = Image.open(self.filepath)
orctext = pytesseract.image_to_string(img, lang='eng+chi_sim+osd')
return orctext

2、设置万能码、验证码白名单或者要开发屏蔽掉
去掉验证码的主要是安全问题,为了应对在线系统的安全性威胁,可以在修改程序时不取消验证码,而是程序中留一个“后门”—设置一个“万能验证码”,只要用户输入这个“万能验证码”,程序就认为验证通过,否则按照原先的验证方式进行验证。
  这是最简单的方法,对于开发人员来说,只是把验证码的相关代码注释掉即可,如果是在测试环境,这样做可省去了测试人员不少麻烦,如果自动化脚本是要在正式环境跑,这样就给系统带来了一定的风险。
 

#coding=utf-8
import random

#生成0到10之间的随机数
#d = random.uniform(0,10)
#print d

#生成一个1000到9999之间的随机整数
d = random.randint(1000,9999)
print u"生成的随机数:%d " %d

i = input(u"请输入随机数:")
print i

if i == d:
    print u"登录成功!!"
elif i == 1111:
    print u"登录成功!!"
else:
    print u"请重新输入验证码!"

运行结果:

>>> ================================ RESTART ================================
>>> 
生成的随机数:3764 
请输入随机数:1111
1111
登录成功!!
>>> ================================ RESTART ================================
>>> 
生成的随机数:3763 
请输入随机数:3763
3763
登录成功!!
>>> ================================ RESTART ================================
>>> 
生成的随机数:1928 
请输入随机数:1354646
1354646
请重新输入验证码!

3、想到用base64编码图片来进行比较
具体代码如下:

#encoding=utf8
"""通过base64编码的字符来比较两个图片是否一致
     作者:Thomas
    日期:2015/4/8
"""

import requests
import base64
from robot.utils.asserts import fail_unless

def 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_str

def 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 False

def _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-8

import unittest
import online_photo_check

class 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))

4、记录cookie
(适用于UI自动化测试,且目前在大部应用的用户名密码不记录在cookie 或 进行加密处理。)
通过向浏览器中添加cookie 可以绕过登录的验证码,这是比较有意思的一种解决方案。我们可以在用户登录之前,通过add_cookie()方法将用户名密码写入浏览器cookie ,再次访问系统登录链接将自动登录。例如下面的方式:

....
#访问xxxx网站
driver.get("http://www.xxxx.cn/")
#将用户名密码写入浏览器cookie
driver.add_cookie({'name':'Login_UserNumber', 'value':'username'})
driver.add_cookie({'name':'Login_Passwd', 'value':'password'})
#再次访问xxxx网站,将会自动登录
driver.get("http://www.xxxx.cn/")
time.sleep(3)
....
driver.quit()

使用cookie进行登录最大的难点是如何获得用户名密码的name ,如果找到不到name 的名字,就没办法向value 中输用户名、密码信息。
我建议是可以通过get_cookies()方法来获取登录的所有的cookie信息,从而进行找到用户名、密码的name 对象的名字;当然,最简单的方法还是询问前端开发人员。

总结:
最简单安全,行之有效的方式就是设置万能码,稍微和开发沟通一下就OK了。如果乐于“闷头苦干自力更生”的话也可研究验证码识别技术。

你可能感兴趣的:(Testing)