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()

python編輯基礎與http接口測試_10.1章節