如何獲取註冊過程圖片上的驗證碼?並實現自動化測試。
一、如果是生產環境,手動測試
二、
如果是測試環境,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了。如果樂於“悶頭苦幹自力更生”的話也可研究驗證碼識別技術。