test_upStream.py 2.88 KB
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import unittest
import json
import urllib3
import time
from commons.api.login import login
from commons.api.upStream import upStream
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
ll=login()
uu=upStream()


class test_upStream1(unittest.TestCase):
    '''上游企业列表查询接口'''
    @classmethod
    def setUpClass(cls):
        #通过登录接口获取session
#         time.sleep(1)
        cls.se=ll.get_session("user02",proxies={'http': 'http://localhost:8888'},allow_redirects=False)

    def test_01_upStream1(self):
        '''成功查询上下游企业,无过滤条件'''
        #请求数据不需要修改,直接POST请求
        re=self.se.post(url=uu.url,headers=uu.header,json=uu.body,proxies={'http': 'http://localhost:8888', 'https': 'http://localhost:8888'})
        print(re.json())
        #响应断言
        self.assertEqual(re.status_code, 200)
        self.assertTrue("total" in re.json().keys())
        self.assertTrue("上游" in str(re.json()).replace(" ",""))
        
    def test_02_upStream1(self):
        '''成功查询上下游企业,加入企业/个人名称过滤条件'''
        #请求数据需要修改
        body=uu.body.copy()
        body["likeName"]="111"
        #POST请求
        re=self.se.post(url=uu.url,headers=uu.header,json=body,proxies={'http': 'http://localhost:8888', 'https': 'http://localhost:8888'})
        print(re.json())
        #响应断言
        self.assertEqual(re.status_code, 200)
        self.assertTrue("legalPerson':'哈哈哈'" in str(re.json()).replace(" ",""))
           
    @classmethod
    def tearDownClass(cls):
        cls.se.close()


class test_upStream2(unittest.TestCase):
    '''上游企业列表查询接口2'''
    def test_01_upStream2(self):
        '''成功查询上下游企业,加入企业/个人名称过滤条件'''
        #请求数据需要修改
        body=uu.body.copy()
        body["likeName"]="111"
        #POST请求
        re=uu.post("user01",url=uu.url,headers=uu.header,json=body)
        print(re.json())
        #响应断言
        self.assertEqual(re.status_code, 200)
        self.assertTrue("legalPerson':'哈哈哈'" in str(re.json()).replace(" ",""))

    def test_02_upStream2(self):
        '''成功查询上下游企业,加入业户名称名称过滤条件'''
        body=uu.body.copy()
        body["likeUserName"]="地利人和"
        #POST请求
        re=uu.post("user01",url=uu.url,headers=uu.header,json=body)
        print(re.json())
        #响应断言
        self.assertEqual(re.status_code, 200)
        self.assertTrue("userNames':'地利人和'" in str(re.json()).replace(" ",""))


if __name__ == "__main__":
    unittest.main(verbosity=2)

#     current_suite = unittest.TestSuite()
#     current_suite.addTest(test_upStream1("test_01_upStream1"))
#     runner = unittest.TextTestRunner()
#     runner.run(current_suite)