python编辑基础与http接口测试_10.1章节

 1 # getipinfo.py
 2 def getipinfo(ip):
 3     import requests
 4     url=‘http://ip.taobao.com/service/getIpInfo.php?ip=‘
 5     r=requests.get(url+ip)
 6     r1=r.json()
 7     a={}
 8     a[‘country‘]=r1[‘data‘][‘country‘]
 9     a[‘city‘]=r1[‘data‘][‘city‘]
10     a[‘region‘]=r1[‘data‘][‘region‘]
11     return a

 1 import getipinfo
 2 import unittest
 3 import time
 4 class test_getipinfo(unittest.TestCase):
 5     def test_01(self):
 6         t=getipinfo.getipinfo(‘113.13.186.134‘)
 7         self.assertEqual(t[‘country‘],‘中国‘)
 8         self.assertEqual(t[‘region‘],‘广西‘)
 9         self.assertEqual(t[‘city‘],‘柳州‘)
10     time.sleep(5)
11     def test_02(self):
12         t = getipinfo.getipinfo(‘39.108.97.124‘)
13         self.assertEqual(t[‘country‘], ‘中国‘)
14         self.assertEqual(t[‘region‘], ‘广东‘)
15         self.assertEqual(t[‘city‘], ‘深圳‘)
16     time.sleep(5)
17     def test_03(self):
18         t = getipinfo.getipinfo(‘39.108.97.xxx‘)
19         self.assertEqual(t[‘country‘], ‘‘)
20         self.assertEqual(t[‘region‘], ‘‘)
21         self.assertEqual(t[‘city‘], ‘‘)
22 if __name__==‘__main__‘:
23     unittest.main()