1. 程式人生 > >如何獲取註冊過程圖片上的驗證碼?並實現自動化測試。

如何獲取註冊過程圖片上的驗證碼?並實現自動化測試。

一、如果是生產環境,手動測試
二、
如果是測試環境,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了。如果樂於“悶頭苦幹自力更生”的話也可研究驗證碼識別技術。