前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >python接口测试之token&session的处理(十四)

python接口测试之token&session的处理(十四)

作者头像
无涯WuYa
发布2018-10-25 16:02:08
1.3K0
发布2018-10-25 16:02:08
举报
文章被收录于专栏:Python自动化测试

python接口测试之token&session处理(十二)中详细了介绍了使用postman工具如何来获取token,以及对token变量的调用和处理,同时在python接口测试之token&session处理(十三)中介绍了在jmeter的测试工具中,使用jmeter然后处理token和session,cookie,以及对token的调用,那么本节课将继续沿着上面的知识点进行,不过本节课将是以代码的方式进行,而不再是工具的形式。

在前面我们介绍了使用postman,jmeter如何获取token以及传递token,下面我们就来使用python语言来进行实现,在这里我们使用第三方的库requests,需要单独的安装下,安装的命令是:

pip install -U requests

见安装的截图:

安装成功后,如果可以在正常的导入,说明安装OK,见截图:

安装好requests好后,我们就可以开始了,关于requests有不清楚的,可以看Python接口测试之requests(七),这里我们直接来进行实战,我们再来进行回顾下我们需要做的事情,我们需要做的就是登录成功后,获取token,后面的接口都需要这个参数,所以我们需要把获取的token当作参数来进行传递,在这里我们来思考一个问题,在python的函数中,为什么会有返回值?返回值可以干什么?python中,函数的返回值是为了给另外一个函数传递参数,传递参数可以理解为是通过自动将对象赋值给本地变量名来实现。OK,下面我们就来使用实现这个过程,我们首先使用代码来实现登录的接口并且获取返回的json字符串,见实现的代码:

代码语言:javascript
复制
#!/usr/bin/env python 
#-*-coding:utf-8-*-

import requests

def getToken():
    r=requests.post(
        url='https://ecapi.parkingwang.com/v4/login',
        json={"username":"autoapi",
            "password":"8d969eef6ecad3c29a3a629280e686cf0c3f5d5a86aff3ca12020c923adc6c92",
            "role":2},
        headers={'Content-Type':'application/json;charset=UTF-8',
                    'Parkingwang-Client-Source':'ParkingWangAPIClientWeb'},timeout=5)
    return r.json()

print getToken()

见执行函数getToken后获取的json字符串:

代码语言:javascript
复制
C:\Python27\python.exe D:/git/Python/UnitCI/blog/weke.py
{u'status': 0, u'msg': u'', u'data': 
{u'token': u'ma7uW5KiqKpBYPQ6kBI1514038085951', u'id': 600192, u'name': u'\u81ea\u52a8\u5316\u6d4b\u8bd5'}}

Process finished with exit code 0

我们可以看到,获取的内容是json字符串,经过反序列化后,数据类型是字典(dict),按字典的方式来获取token就很简单的,见获取token的代码:

代码语言:javascript
复制
#!/usr/bin/env python 
#-*-coding:utf-8-*-

import requests

def getToken():
    r=requests.post(
        url='https://ecapi.parkingwang.com/v4/login',
        json={"username":"autoapi","password":"8d969eef6ecad3c29a3a629280e686cf0c3f5d5a86aff3ca12020c923adc6c92","role":2},
        headers={'Content-Type':'application/json;charset=UTF-8',
                    'Parkingwang-Client-Source':'ParkingWangAPIClientWeb'},timeout=5)
    return r.json()['data']['token']

见执行后,获取的token字符串,见截图:

下面来实现infoGet接口的测试,依据前面的案例我们知道,该接口的参数就是token,而且必须是和登录成功后返回的token是一致,否则就会提示502的错误,token无效,我们来实现infoGet接口测试,见实现代码:

代码语言:javascript
复制
#!/usr/bin/env python 
#-*-coding:utf-8-*-

import requests

def getToken():
    r=requests.post(
        url='https://ecapi.parkingwang.com/v4/login',
        json={"username":"autoapi","password":"8d969eef6ecad3c29a3a629280e686cf0c3f5d5a86aff3ca12020c923adc6c92","role":2},
        headers={'Content-Type':'application/json;charset=UTF-8','Parkingwang-Client-Source':'ParkingWangAPIClientWeb'},timeout=5)
    return r.json()['data']['token']


def infoGet():
    r=requests.post(
        url='https://ecapi.parkingwang.com/v4/infoGet',
        json={"token":getToken()},
        headers={'Content-Type':'application/json;charset=UTF-8','Parkingwang-Client-Source':'ParkingWangAPIClientWeb'},timeout=5)
    print r.json()


infoGet()

见执行infoGet函数后,返回的json字符串内容:

代码语言:javascript
复制
C:\Python27\python.exe D:/git/Python/UnitCI/blog/weke.py
{u'status': 0, u'msg': u'', u'data': {u'username': u'autoapi', u'online_agreement': 0, u'name': u'\u81ea\u52a8\u5316\u6d4b\u8bd5', u'valid_period': 0, u'id': 600192, u'phone': u'', u'pigeon_id': u'maLdrGaCEMAoS9kYW9t1514038531444', u'address': u'', u'supports': u'0', u'email': u'', u'park_code': 6666666666L}}

我们接着加一个接口,该接口为isSoonExpire,见测试该接口实现的代码:

代码语言:javascript
复制
#!/usr/bin/env python 
#-*-coding:utf-8-*-

import requests

def getToken():
    r=requests.post(
        url='https://ecapi.parkingwang.com/v4/login',
        json={"username":"autoapi","password":"8d969eef6ecad3c29a3a629280e686cf0c3f5d5a86aff3ca12020c923adc6c92","role":2},
        headers={'Content-Type':'application/json;charset=UTF-8',
                 'Parkingwang-Client-Source':'ParkingWangAPIClientWeb'},timeout=5)
    return r.json()['data']['token']


def infoGet():
    r=requests.post(
        url='https://ecapi.parkingwang.com/v4/infoGet',
        json={"token":getToken()},
        headers={'Content-Type':'application/json;charset=UTF-8',
                 'Parkingwang-Client-Source':'ParkingWangAPIClientWeb'},timeout=5)
    print r.json()


def isSoonExpire():
    r=requests.post(
        url='https://ecapi.parkingwang.com/v4/isSoonExpire',
        json={"token":getToken()},
        headers={'Content-Type':'application/json;charset=UTF-8',
                 'Parkingwang-Client-Source':'ParkingWangAPIClientWeb'},timeout=5)
    print r.json()

下面我们对代码进行重构下,引入unittest,见重构后的代码:

代码语言:javascript
复制
#!/usr/bin/env python 
#-*-coding:utf-8-*-

import requests
import  unittest

class InterfaceTest(unittest.TestCase):
    def setUp(self):
        self.url='https://ecapi.parkingwang.com/v4/'
        self.headers={'Content-Type':'application/json;charset=UTF-8',
                      'Parkingwang-Client-Source':'ParkingWangAPIClientWeb'}
        self.timeout=5

    def tearDown(self):
        pass

    def getToken(self):
        r = requests.post(
            url=self.url+'login',
            json={"username": "autoapi", 
                  "password": "8d969eef6ecad3c29a3a629280e686cf0c3f5d5a86aff3ca12020c923adc6c92",
                  "role": 2},
            headers=self.headers, timeout=self.timeout)
        return r.json()['data']['token']

    def test_infoGet(self):
        '''验证:测试infoGet接口是否正确'''
        r = requests.post(
            url=self.url+'infoGet',json={"token": self.getToken()},
           headers=self.headers, timeout=self.timeout)
        print r.json()

    def test_isSoonExpire(self):
        '''验证:测试isSoonExpire接口是否正确'''
        r = requests.post(
            url='https://ecapi.parkingwang.com/v4/isSoonExpire',
           json={"token":self.getToken()},headers=self.headers, timeout=5)
        print r.json()

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

如上的接口测试用例缺少断言,在这里特别强调,没有断言的自动化测试用例是无效的,是没有任何的价值的。至于具体断言什么,得格局具体的接口测试的业务来进行,我们对如上的接口添加断言,见增加断言后的接口测试代码:

代码语言:javascript
复制
#!/usr/bin/env python 
#-*-coding:utf-8-*-

import requests
import  unittest
import  time as t

class InterfaceTest(unittest.TestCase):
    def setUp(self):
        self.url='https://ecapi.parkingwang.com/v4/'
        self.headers={'Content-Type':'application/json;charset=UTF-8',
                      'Parkingwang-Client-Source':'ParkingWangAPIClientWeb'}
        self.timeout=5

    def tearDown(self):
        t.sleep(1)

    def getToken(self):
        r = requests.post(
            url=self.url+'login',
            json={"username": "autoapi", 
                  "password": "8d969eef6ecad3c29a3a629280e686cf0c3f5d5a86aff3ca12020c923adc6c92",
                  "role": 2},
            headers=self.headers, timeout=self.timeout)
        return r.json()['data']['token']

    def test_infoGet(self):
        '''验证:测试infoGet接口是否正确'''
        r = requests.post(
            url=self.url+'infoGet',json={"token": self.getToken()},
           headers=self.headers, timeout=self.timeout)
        self.assertEqual(r.json()['status'],0)
        self.assertEqual(r.json()['data']['username'],'autoapi')

    def test_isSoonExpire(self):
        '''验证:测试isSoonExpire接口是否正确'''
        r = requests.post(
            url='https://ecapi.parkingwang.com/v4/isSoonExpire',
           json={"token":self.getToken()},headers=self.headers, timeout=5)
        self.assertEqual(r.json()['status'],0)
        self.assertEqual(r.json()['data']['expire'],False)

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

执行上面的测试用例后,我们发现二个测试用例都调用了getToken()方法,而getToken()方法我们知道是登录的接口,登录成功后获取token,言外之意也就是说登录了二次,缺点很明显,如果在N个测试用例,参数都需要token,都会调用getToken(),那么也会出现登录N次,很实现这个方法不是我们想要的,我们要实现的是不管有多少个接口测试用例,登录只能是一次,如果登录多次,会出现token无效502的问题,或者引起其他的问题,在这里我们把登录成功后的token存储在文件中,然后从文件中读取,这样就可以登录一次,见实现的代码:

代码语言:javascript
复制
#!/usr/bin/env python 
#-*-coding:utf-8-*-

import requests
import  unittest
import  time as t
import  os

def getHeaders():
    '''获取headers'''
    return {'Content-Type':'application/json;charset=UTF-8',
            'Parkingwang-Client-Source':'ParkingWangAPIClientWeb'}

def login():
    '''把token写入到文件中'''
    r = requests.post(
        url=self.url + 'login',
        json={"username": "autoapi", 
              "password": "8d969eef6ecad3c29a3a629280e686cf0c3f5d5a86aff3ca12020c923adc6c92",
              "role": 2},
        headers=getHeaders(), timeout=5)
    with open(base_dir(), 'w') as f:
        f.write(r.json()['data']['token'])


def base_dir():
    '''获取当前文件的目录'''
    return os.path.join(os.path.dirname(__file__), 'token.md')

def getToken():
    '''读取存储在文件中的token'''
    with open(base_dir(),'r') as f:
        return f.read()

class InterfaceTest(unittest.TestCase):
    def setUp(self):
        self.url='https://ecapi.parkingwang.com/v4/'

    def tearDown(self):
        t.sleep(1)

    def test_infoGet(self):
        '''验证:测试infoGet接口是否正确'''
        r = requests.post(url=self.url+'infoGet',
                          json={"token": getToken()},headers=getHeaders(), timeout=5)
        self.assertEqual(r.json()['status'],0)
        self.assertEqual(r.json()['data']['username'],'autoapi')

    def test_isSoonExpire(self):
        '''验证:测试isSoonExpire接口是否正确'''
        r = requests.post(url=self.url+'isSoonExpire',
                          json={"token":getToken()},headers=getHeaders(), timeout=5)
        self.assertEqual(r.json()['status'],0)
        self.assertEqual(r.json()['data']['expire'],False)

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

代码经过重构后,不管有多少个测试用例,我们的登录只会进行一次。接下来我们会写关于session这部分,python是如何解决的。

python处理session

下面我们使用python来处理session的部分,下面我们来看一个例子,登录成功后,并没有返回token,只是返回了请求成功的code,而是存储session中,那么我们应该如何来获取session了,我们编写该登录的接口用例,看返回的json字符串是什么,见实现的代码以及输出的截图:

在截图中我们看到,输出的json字符串只有code是200的,再没有什么,但是后面的请求都需要带上session,我们看下一个接口请求的内容,见截图:

OK,那么就说明我们登录后必须得获取session,见获取session的代码:

通过cookies.get_dict()就可以获取session,见如上截图中获取的session内容,下面的就很简单,我们在下一个接口用例中带上session,见实现的码:

执行如上的代码后,我们发现代码错误,错误的原因是test_access_datas()函数没有带上登录成功后的session,所以导致错误,不带session,一般直接就跳转到了login的页面,让重新登录,那么如何解决了,我们只需要在test_access_datas()函数中,在接口请求中带上session就可以了,见修改后执行的代码:

如此,总结了使用postman,jmeter,python语言来处理token和session的问题,在这里一次性的进行总结,如您对我写的内容感兴趣,请分享到朋友圈让更多的人来关注,谢谢。也可关注本人的微信公众号,本人会持续进行更新,谢谢!

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2018-02-04,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 Python自动化测试 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • python处理session
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档