UnitTest/TestAll.py - master - githubmirror/12306

UnitTest/TestAll.py
119 строк18 KB
xianping.wen
add open str
18 дек 2019, 11:25
18 дек 2019, 11:2588a1941
# coding=utf-8
import base64
import threading
import unittest
from collections import OrderedDict

import requests

import TickerConfig
from agency.agency_tools import proxy
from config.emailConf import sendEmail
from config.serverchanConf import sendServerChan
from inter.LiftTicketInit import liftTicketInit


def _set_header_default():
    header_dict = OrderedDict()
    header_dict["Accept"] = "*/*"
    header_dict["Accept-Encoding"] = "gzip, deflate"
    header_dict["X-Requested-With"] = "superagent"

    header_dict[
        "User-Agent"] = "Mozilla/5.0 (iPhone; CPU iPhone OS 11_0 like Mac OS X) AppleWebKit/604.1.38 (KHTML, like Gecko) Version/11.0 Mobile/15A372 Safari/604.1"
    header_dict["Content-Type"] = "application/x-www-form-urlencoded; charset=UTF-8"


class testAll(unittest.TestCase):
    def testProxy(self):
        """
        测试代理是否可用
        :return:
        """
        _proxy = proxy()
        proxie = _proxy.setProxy()
        url = "http://httpbin.org/ip"
        rsp = requests.get(url, proxies=proxie, timeout=5, headers=_set_header_default()).content
        print(u"当前代理ip地址为: {}".format(rsp))

    def testEmail(self):
        """
        实测邮箱是否可用
        :return:
        """
        sendEmail(u"订票小助手测试一下")

    # def testConfig(self):
    #     """
    #     测试config是否配置正确
    #     :return:
    #     """

    def testServerChan(self):
        """
        实测server酱是否可用
        :return:
        """
        sendServerChan(u"server酱 微信通知测试一下")

    def testUserAgent(self):
        """
        测试UserAgent
        :return:
        """
        from fake_useragent import UserAgent
        for i in range(10000):
            ua = UserAgent(verify_ssl=False)
            print(ua.random)

    def testVerfyImage(self):
        """
        测试模型加载识别
        :return:
        """
        from verify.localVerifyCode import Verify
        v = Verify()
        with open('../tkcode.png', 'rb') as f:
            base64Image = base64.b64encode(f.read())
            for i in range(5):
                t = threading.Thread(target=v.verify, args=(base64Image,))
                t.start()

    def testRemoteVerfy(self):
        """
        测试打码是否可用
        :return:
        """
        import requests
        import time
        while True:
            try:
                starttime = time.time()
                rsp = requests.post(url="http://120.77.154.140:8000/verify/base64/",
                                    data={
                                        'imageFile': '/9j/4AAQSkZJRgABAgAAAQABAAD/2wBDAAgGBgcGBQgHBwcJCQgKDBQNDAsLDBkSEw8UHRofHh0aHBwgJC4nICIsIxwcKDcpLDAxNDQ0Hyc5PTgyPC4zNDL/2wBDAQkJCQwLDBgNDRgyIRwhMj … [Строка слишком длинная. Вы можете скачать файл]
                                    timeout=60,
                                    )
                print(rsp.content)
                print(f"响应时间{time.time()-starttime}m")
            except:
                pass

    def testCdn(self):
        """
        测试cdn筛选
        :return:
        """
        cdn = ["60.9.0.19", "60.9.0.20", "113.16.212.251", "36.250.248.27"]
        from inter.LiftTicketInit import liftTicketInit
        from init.select_ticket_info import select
        from config.getCookie import getDrvicesID

        s = select()
        s.httpClint.cdn = cdn[3]
        getDrvicesID(s)
        liftTicketInit(s).reqLiftTicketInit()


if __name__ == '__main__':
    unittest.main()