python使用笔记29--代码驱动

import unittest
import requests
import jsonpath
import time
import nnreport

def get_value(d,key,expr=False,more=False):
    if expr:
        jp = jsonpath.jsonpath(d,key)#获取json字符串里某个key的值，返回的是[]
    else:
        jp = jsonpath.jsonpath(d, '$..%s' % key)  # 获取json字符串里某个key的值，返回的是[]

    if jp:
        if more:
            return jp
        else:
            return jp[0]


class LiteMallBase(unittest.TestCase):
    username = "admin123"
    password = username
    host = "http://127.0.0.1:8080"

    def login(self):
        '''登录'''
        url = self.host + '/admin/auth/login'
        data = {'username': self.username, 'password': self.password}
        r = requests.get(url, json=data)
        token = get_value(r.json(), 'token')
        self.assertIsNotNone(token, msg='获取不到token值')
        return token

    def create_coupon(self, name, token, money):
        '''建券'''
        url = self.host + "/admin/coupon/create"

        data = {"name": name, "desc": "优惠券111",
                "total": "999", "discount": 100, "min": money,
                "limit": 1, "type": 0, "status": 0, "goodsType": 0,
                "goodsValue": [], "timeType": 0, "days": "10",
                "startTime": None, "endTime": None}
        headers = {'X-Litemall-Admin-Token': token}
        r = requests.get(url, json=data, headers=headers)
        cid = get_value(r.json(), 'id')
        cname = get_value(r.json(), 'name')
        errno = get_value(r.json(), 'errno')
        self.assertEqual(errno, 0, msg='创建券失败%s' % errno)
        self.assertEqual(name, cname, msg='创建券名和返回的不一致')
        return cid

    def index_data(self):
        '''查询首页数据'''
        url = self.host + '/wx/home/index'
        r = requests.get(url)
        errno = get_value(r.json(), 'errno')
        self.assertEqual(errno, 0, msg="首页数据查询失败")
        return r.json()

class CouponCreate(LiteMallBase):
    def test_create_coupon(self):
        '''建券测试'''
        name = "自动化测试%s-优惠券" % int(time.time())
        token = self.login()
        cid = self.create_coupon(name,token,9999)
        index_data = self.index_data()
        coupon_ids = get_value(index_data,'$.data.couponList..id',True,True)
        self.assertIn(cid,coupon_ids,msg='创建的券ID不存在，不存在的ID是%s'%cid)


suite = unittest.makeSuite(CouponCreate)
report = nnreport.BeautifulReport(suite)
report.report(description='建券测试报告',filename='建券测试报告.html')