1，安装
$ pip install -U pytest or $ easy_install -U pytest $ py.test --version |
2，基础语法
#!/usr/bin/python import pytest def func(x): return x + 1 def test_answer(): #测试用例,方法需要以test_开头 assert func(3) == 5 |
运行之
$ py.test 2.py #注意需要用py.test命令来运行之
稍微复杂一点儿的写法：
#!/usr/bin/python import pytest class TestClass: def test_one(self): x = "this" assert "h" in x def test_two(self): x = 3 assert x > 2 |
这里总结一下用法，
1，以test_开头的方法，会被py.test自动处理，无需额外调用该方法；
2，使用assert进行判断，assert后面应该是一个boolean类型的，可以是一个判断语句
，比如3 > 2，又或者是a in b等等。
3，实际应用
这里有一个实际应用，我想批量检查一批机器上的CPU，内存，和机器上的2个分区，并将CPU大于80%，内存大于95，和分区大于80%的机器找出来，如何实现呢？
假设这里已经提供好了API，可以读取到所有机器上的CPU、内存、分区信息，API地址为http://api/latestMeteris?userCode=xxx&token=xxx&host=’172.20.116.70,172.20.116.72’&service=CPU,Memory,Disk
访问API时返回一串JSON，信息如下：
{ "message":"success", "result":"success", "start":"2017-02-28 13:54:53", "data":{ "Memory":{ "172.20.116.72":{ "swap_used":["9.60%"], "datetime":["2017-02-28 13:54:41"], "merge_time":["2017-02-28 13:54:41"], "ram_used":["25.52%"] }, "172.20.116.70":{ "swap_used":["6.17%"], "datetime":["2017-02-28 13:54:41"], "merge_time":["2017-02-28 13:54:41"], "ram_used":["25.97%"] } }, "CPU":{ "172.20.116.72":{ "datetime":["2017-02-28 13:54:41"], "merge_time":["2017-02-28 13:54:41"], "cpu_prct_used":["3.00%"] }, "172.20.116.70":{ "datetime":["2017-02-28 13:54:41"], "merge_time":["2017-02-28 13:54:41"], "cpu_prct_used":["1.00%"] } }, "Disk":{ "172.20.116.72":{ "datetime":["2017-02-28 13:54:41"], "merge_time":["2017-02-28 13:54:41"], "/export":["25.06%"], "/":["21.6%"] }, "172.20.116.70":{ "datetime":["2017-02-28 13:54:41"], "merge_time":["2017-02-28 13:54:41"], "/export":["44.68%"], "/":["36.15%"] } } }, "host_size":2, "end":"2017-02-28 13:54:53" } |
pytest脚本写法如下：
#!/usr/bin/python importos importsys importjson importurllib importurllib2 importpytest iplist=["172.20.116.70","172.20.116.72"]#定义IP列表 ips=','.join(iplist) url='http://api/latestMeteris?userCode=xxx&token=xxx&host='+ips+'&service=CPU,Memory,Disk' req=urllib.urlopen(url) result=req.read()#getastringtype a=json.loads(result)#transferstringtypetodicttype @pytest.mark.parametrize('ip',iplist) deftest_cpu(ip): value=a["data"]["CPU"][ip]["cpu_prct_used"][0] assertfloat(value.strip("%"))<80 @pytest.mark.parametrize('ip',iplist) deftest_memory(ip): value=a["data"]["Memory"][ip]["ram_used"][0] assertfloat(value.strip("%"))<95 @pytest.mark.parametrize('ip',iplist) deftest_disk(ip): value_root=a["data"]["Disk"][ip]['/'][0] value_export=a["data"]["Disk"][ip]['/export'][0] assertfloat(value_root.strip("%"))<80andfloat(value_export.strip("%"))<80 |
运行脚本试一下
$py.test2.py =========================testsessionstarts========================= platformlinux2--Python2.7.4,pytest-3.0.6,py-1.4.31,pluggy-0.4.0 rootdir:/home/zhukun/0224,inifile: collected6items 2.py...... ======================6passedin0.05seconds====================== |
意为6个测试全部通过。
