b10-stats-httpd_test.py 66 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336
  1. # Copyright (C) 2011-2012 Internet Systems Consortium.
  2. #
  3. # Permission to use, copy, modify, and distribute this software for any
  4. # purpose with or without fee is hereby granted, provided that the above
  5. # copyright notice and this permission notice appear in all copies.
  6. #
  7. # THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SYSTEMS CONSORTIUM
  8. # DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL
  9. # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL
  10. # INTERNET SYSTEMS CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT,
  11. # INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING
  12. # FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
  13. # NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
  14. # WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15. """
  16. In each of these tests we start several virtual components. They are
  17. not the real components, no external processes are started. They are
  18. just simple mock objects running each in its own thread and pretending
  19. to be bind10 modules. This helps testing the stats http server in a
  20. close to real environment.
  21. """
  22. import unittest
  23. import os
  24. import imp
  25. import socket
  26. import errno
  27. import select
  28. import string
  29. import time
  30. import threading
  31. import http.client
  32. import xml.etree.ElementTree
  33. import random
  34. import urllib.parse
  35. import isc
  36. import stats_httpd
  37. import stats
  38. from test_utils import BaseModules, ThreadingServerManager, MyStats,\
  39. MyStatsHttpd, SignalHandler,\
  40. send_command, send_shutdown
  41. from isc.testutils.ccsession_mock import MockModuleCCSession
  42. # set XML Namespaces for testing
  43. XMLNS_XSL = "http://www.w3.org/1999/XSL/Transform"
  44. XMLNS_XHTML = "http://www.w3.org/1999/xhtml"
  45. XMLNS_XSD = "http://www.w3.org/2001/XMLSchema"
  46. XMLNS_XSI = stats_httpd.XMLNS_XSI
  47. DUMMY_DATA = {
  48. 'Boss' : {
  49. "boot_time": "2011-03-04T11:59:06Z"
  50. },
  51. 'Auth' : {
  52. "queries.tcp": 2,
  53. "queries.udp": 3,
  54. "queries.perzone": [{
  55. "zonename": "test.example",
  56. "queries.tcp": 2,
  57. "queries.udp": 3
  58. }],
  59. "nds_queries.perzone": {
  60. "test.example": {
  61. "queries.tcp": 2,
  62. "queries.udp": 3
  63. }
  64. }
  65. },
  66. 'Stats' : {
  67. "report_time": "2011-03-04T11:59:19Z",
  68. "boot_time": "2011-03-04T11:59:06Z",
  69. "last_update_time": "2011-03-04T11:59:07Z",
  70. "lname": "4d70d40a_c@host",
  71. "timestamp": 1299239959.560846
  72. }
  73. }
  74. def get_availaddr(address='127.0.0.1', port=8001):
  75. """returns a tuple of address and port which is available to
  76. listen on the platform. The first argument is a address for
  77. search. The second argument is a port for search. If a set of
  78. address and port is failed on the search for the availability, the
  79. port number is increased and it goes on the next trial until the
  80. available set of address and port is looked up. If the port number
  81. reaches over 65535, it may stop the search and raise a
  82. OverflowError exception."""
  83. while True:
  84. for addr in socket.getaddrinfo(
  85. address, port, 0,
  86. socket.SOCK_STREAM, socket.IPPROTO_TCP):
  87. sock = socket.socket(addr[0], socket.SOCK_STREAM)
  88. try:
  89. sock.bind((address, port))
  90. return (address, port)
  91. except socket.error:
  92. continue
  93. finally:
  94. if sock: sock.close()
  95. # This address and port number are already in use.
  96. # next port number is added
  97. port = port + 1
  98. def is_ipv6_enabled(address='::1', port=8001):
  99. """checks IPv6 enabled on the platform. address for check is '::1'
  100. and port for check is random number between 8001 and
  101. 65535. Retrying is 3 times even if it fails. The built-in socket
  102. module provides a 'has_ipv6' parameter, but it's not used here
  103. because there may be a situation where the value is True on an
  104. environment where the IPv6 config is disabled."""
  105. for p in random.sample(range(port, 65535), 3):
  106. try:
  107. sock = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
  108. sock.bind((address, p))
  109. return True
  110. except socket.error:
  111. continue
  112. finally:
  113. if sock: sock.close()
  114. return False
  115. class TestItemNameList(unittest.TestCase):
  116. def test_item_name_list(self):
  117. self.assertEqual(['a'],
  118. stats_httpd.item_name_list({'a':1}, 'a'))
  119. self.assertEqual(['a','a/b'],
  120. stats_httpd.item_name_list({'a':{'b':1}}, 'a'))
  121. self.assertEqual(['a/b'],
  122. stats_httpd.item_name_list({'a':{'b':1}}, 'a/b'))
  123. self.assertEqual(['a','a/b','a/b/c'],
  124. stats_httpd.item_name_list({'a':{'b':{'c':1}}}, 'a'))
  125. self.assertEqual(['a/b','a/b/c'],
  126. stats_httpd.item_name_list({'a':{'b':{'c':1}}}, 'a/b'))
  127. self.assertEqual(['a/b/c'],
  128. stats_httpd.item_name_list({'a':{'b':{'c':1}}}, 'a/b/c'))
  129. self.assertEqual(['a[2]'],
  130. stats_httpd.item_name_list({'a':[1,2,3]}, 'a[2]'))
  131. self.assertEqual(['a', 'a[0]', 'a[1]', 'a[2]'],
  132. stats_httpd.item_name_list({'a':[1,2,3]}, 'a'))
  133. self.assertEqual(['a', 'a[0]', 'a[1]', 'a[2]'],
  134. stats_httpd.item_name_list({'a':[1,2,3]}, ''))
  135. self.assertEqual(['a', 'a/b', 'a/b[0]', 'a/b[1]', 'a/b[2]'],
  136. stats_httpd.item_name_list({'a':{'b':[1,2,3]}}, 'a'))
  137. self.assertEqual(['a', 'a/b', 'a/b[0]', 'a/b[1]', 'a/b[2]'],
  138. stats_httpd.item_name_list({'a':{'b':[1,2,3]}}, ''))
  139. self.assertEqual(['a/b', 'a/b[0]', 'a/b[1]', 'a/b[2]'],
  140. stats_httpd.item_name_list({'a':{'b':[1,2,3]}}, 'a/b'))
  141. self.assertEqual(['a', 'a/b', 'a/b[0]', 'a/b[1]', 'a/b[2]', 'a/c'],
  142. stats_httpd.item_name_list(
  143. {'a':{'b':[1,2,3], 'c':1}}, 'a'))
  144. self.assertEqual(['a/b', 'a/b[0]', 'a/b[1]', 'a/b[2]'],
  145. stats_httpd.item_name_list(
  146. {'a':{'b':[1,2,3], 'c':1}}, 'a/b'))
  147. self.assertEqual(['a/c'],
  148. stats_httpd.item_name_list(
  149. {'a':{'b':[1,2,3], 'c':1}}, 'a/c'))
  150. self.assertRaises(isc.cc.data.DataNotFoundError,
  151. stats_httpd.item_name_list, {'x':1}, 'a')
  152. self.assertEqual([],
  153. stats_httpd.item_name_list('a', ''))
  154. self.assertEqual([],
  155. stats_httpd.item_name_list('', ''))
  156. self.assertRaises(isc.cc.data.DataTypeError,
  157. stats_httpd.item_name_list, 'a', 'a')
  158. self.assertRaises(isc.cc.data.DataTypeError,
  159. stats_httpd.item_name_list, None, None)
  160. self.assertRaises(isc.cc.data.DataTypeError,
  161. stats_httpd.item_name_list, [1,2,3], 'a')
  162. self.assertRaises(isc.cc.data.DataTypeError,
  163. stats_httpd.item_name_list, [1,2,3], '')
  164. class TestHttpHandler(unittest.TestCase):
  165. """Tests for HttpHandler class"""
  166. def setUp(self):
  167. # set the signal handler for deadlock
  168. self.sig_handler = SignalHandler(self.fail)
  169. self.base = BaseModules()
  170. self.stats_server = ThreadingServerManager(MyStats)
  171. self.stats = self.stats_server.server
  172. DUMMY_DATA['Stats']['lname'] = self.stats.cc_session.lname
  173. self.stats_server.run()
  174. (self.address, self.port) = get_availaddr()
  175. self.stats_httpd_server = ThreadingServerManager(MyStatsHttpd, (self.address, self.port))
  176. self.stats_httpd = self.stats_httpd_server.server
  177. self.stats_httpd_server.run()
  178. self.client = http.client.HTTPConnection(self.address, self.port)
  179. self.client._http_vsn_str = 'HTTP/1.0\n'
  180. self.client.connect()
  181. def tearDown(self):
  182. self.client.close()
  183. self.stats_httpd_server.shutdown()
  184. self.stats_server.shutdown()
  185. self.base.shutdown()
  186. # reset the signal handler
  187. self.sig_handler.reset()
  188. def test_do_GET(self):
  189. self.assertTrue(type(self.stats_httpd.httpd) is list)
  190. self.assertEqual(len(self.stats_httpd.httpd), 1)
  191. self.assertEqual((self.address, self.port), self.stats_httpd.http_addrs[0])
  192. def check_XML_URL_PATH(mod=None, item=None):
  193. url_path = stats_httpd.XML_URL_PATH
  194. if mod is not None:
  195. url_path = url_path + '/' + mod
  196. if item is not None:
  197. url_path = url_path + '/' + item
  198. self.client.putrequest('GET', url_path)
  199. self.client.endheaders()
  200. response = self.client.getresponse()
  201. self.assertEqual(response.getheader("Content-type"), "text/xml")
  202. self.assertGreater(int(response.getheader("Content-Length")), 0)
  203. self.assertEqual(response.status, 200)
  204. xml_doctype = response.readline().decode()
  205. xsl_doctype = response.readline().decode()
  206. self.assertGreater(len(xml_doctype), 0)
  207. self.assertGreater(len(xsl_doctype), 0)
  208. root = xml.etree.ElementTree.parse(response).getroot()
  209. self.assertGreater(root.tag.find('statistics'), 0)
  210. schema_loc = '{%s}schemaLocation' % XMLNS_XSI
  211. # check the path of XSD
  212. self.assertEqual(root.attrib[schema_loc],
  213. stats_httpd.XSD_NAMESPACE + ' '
  214. + stats_httpd.XSD_URL_PATH)
  215. # check the path of XSL
  216. self.assertTrue(xsl_doctype.startswith(
  217. '<?xml-stylesheet type="text/xsl" href="' +
  218. stats_httpd.XSL_URL_PATH
  219. + '"?>'))
  220. for elem in root.find('item'):
  221. attr = elem.attrib
  222. value = isc.cc.data.find(DUMMY_DATA, attr['identifier'])
  223. if type(value) is list or type(value) is dict:
  224. self.assertFalse('value' in attr)
  225. elif type(value) is int or type(value) is float:
  226. self.assertEqual(attr['value'], str(value))
  227. else:
  228. self.assertEqual(attr['value'], value)
  229. # URL is '/bind10/statistics/xml'
  230. check_XML_URL_PATH(mod=None, item=None)
  231. for m in DUMMY_DATA:
  232. # URL is '/bind10/statistics/xml/Module'
  233. check_XML_URL_PATH(mod=m)
  234. for k in DUMMY_DATA[m].keys():
  235. # URL is '/bind10/statistics/xml/Module/Item'
  236. check_XML_URL_PATH(mod=m, item=k)
  237. def check_XSD_URL_PATH(mod=None, item=None):
  238. url_path = stats_httpd.XSD_URL_PATH
  239. if mod is not None:
  240. url_path = url_path + '/' + mod
  241. if item is not None:
  242. url_path = url_path + '/' + item
  243. self.client.putrequest('GET', url_path)
  244. self.client.endheaders()
  245. response = self.client.getresponse()
  246. self.assertEqual(response.getheader("Content-type"), "text/xml")
  247. self.assertTrue(int(response.getheader("Content-Length")) > 0)
  248. self.assertEqual(response.status, 200)
  249. root = xml.etree.ElementTree.parse(response).getroot()
  250. url_xmlschema = '{http://www.w3.org/2001/XMLSchema}'
  251. self.assertTrue(root.tag.find('schema') > 0)
  252. self.assertTrue(hasattr(root, 'attrib'))
  253. self.assertTrue('targetNamespace' in root.attrib)
  254. self.assertEqual(root.attrib['targetNamespace'],
  255. stats_httpd.XSD_NAMESPACE)
  256. if mod is None and item is None:
  257. for (mod, itm) in DUMMY_DATA.items():
  258. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'element', 'complexType', 'all', 'element' ] ])
  259. mod_elm = dict([ (elm.attrib['name'], elm) for elm in root.findall(xsdpath) ])
  260. self.assertTrue(mod in mod_elm)
  261. for (it, val) in itm.items():
  262. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'complexType', 'all', 'element' ] ])
  263. itm_elm = dict([ (elm.attrib['name'], elm) for elm in mod_elm[mod].findall(xsdpath) ])
  264. self.assertTrue(it in itm_elm)
  265. if type(val) is list:
  266. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'complexType', 'sequence', 'element' ] ])
  267. itm_elm2 = dict([ (elm.attrib['name'], elm) for elm in itm_elm[it].findall(xsdpath) ])
  268. self.assertTrue('zones' in itm_elm2)
  269. for i in val:
  270. for k in i.keys():
  271. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'complexType', 'all', 'element' ] ])
  272. self.assertTrue(
  273. k in [ elm.attrib['name'] for elm in itm_elm2['zones'].findall(xsdpath) ])
  274. elif item is None:
  275. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'element', 'complexType', 'all', 'element' ] ])
  276. mod_elm = dict([ (elm.attrib['name'], elm) for elm in root.findall(xsdpath) ])
  277. self.assertTrue(mod in mod_elm)
  278. for (it, val) in DUMMY_DATA[mod].items():
  279. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'complexType', 'all', 'element' ] ])
  280. itm_elm = dict([ (elm.attrib['name'], elm) for elm in mod_elm[mod].findall(xsdpath) ])
  281. self.assertTrue(it in itm_elm)
  282. if type(val) is list:
  283. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'complexType', 'sequence', 'element' ] ])
  284. itm_elm2 = dict([ (elm.attrib['name'], elm) for elm in itm_elm[it].findall(xsdpath) ])
  285. self.assertTrue('zones' in itm_elm2)
  286. for i in val:
  287. for k in i.keys():
  288. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'complexType', 'all', 'element' ] ])
  289. self.assertTrue(
  290. k in [ elm.attrib['name'] for elm in itm_elm2['zones'].findall(xsdpath) ])
  291. else:
  292. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'element', 'complexType', 'all', 'element' ] ])
  293. mod_elm = dict([ (elm.attrib['name'], elm) for elm in root.findall(xsdpath) ])
  294. self.assertTrue(mod in mod_elm)
  295. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'complexType', 'all', 'element' ] ])
  296. itm_elm = dict([ (elm.attrib['name'], elm) for elm in mod_elm[mod].findall(xsdpath) ])
  297. self.assertTrue(item in itm_elm)
  298. if type(DUMMY_DATA[mod][item]) is list:
  299. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'complexType', 'sequence', 'element' ] ])
  300. itm_elm2 = dict([ (elm.attrib['name'], elm) for elm in itm_elm[item].findall(xsdpath) ])
  301. self.assertTrue('zones' in itm_elm2)
  302. for i in DUMMY_DATA[mod][item]:
  303. for k in i.keys():
  304. xsdpath = '/'.join([ url_xmlschema + t for t in [ 'complexType', 'all', 'element' ] ])
  305. self.assertTrue(
  306. k in [ elm.attrib['name'] for elm in itm_elm2['zones'].findall(xsdpath) ])
  307. # URL is '/bind10/statistics/xsd'
  308. check_XSD_URL_PATH(mod=None, item=None)
  309. for m in DUMMY_DATA:
  310. # URL is '/bind10/statistics/xsd/Module'
  311. check_XSD_URL_PATH(mod=m)
  312. for k in DUMMY_DATA[m].keys():
  313. # URL is '/bind10/statistics/xsd/Module/Item'
  314. check_XSD_URL_PATH(mod=m, item=k)
  315. def check_XSL_URL_PATH(mod=None, item=None):
  316. url_path = stats_httpd.XSL_URL_PATH
  317. if mod is not None:
  318. url_path = url_path + '/' + mod
  319. if item is not None:
  320. url_path = url_path + '/' + item
  321. self.client.putrequest('GET', url_path)
  322. self.client.endheaders()
  323. response = self.client.getresponse()
  324. self.assertEqual(response.getheader("Content-type"), "text/xml")
  325. self.assertTrue(int(response.getheader("Content-Length")) > 0)
  326. self.assertEqual(response.status, 200)
  327. root = xml.etree.ElementTree.parse(response).getroot()
  328. url_trans = '{http://www.w3.org/1999/XSL/Transform}'
  329. url_xhtml = '{http://www.w3.org/1999/xhtml}'
  330. self.assertEqual(root.tag, url_trans + 'stylesheet')
  331. if item is None and mod is None:
  332. xslpath = url_trans + 'template/' + url_xhtml + 'table/' + url_trans + 'for-each'
  333. mod_fe = dict([ (x.attrib['select'], x) for x in root.findall(xslpath) ])
  334. for (mod, itms) in DUMMY_DATA.items():
  335. self.assertTrue(mod in mod_fe)
  336. for (k, v) in itms.items():
  337. if type(v) is list:
  338. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  339. + url_xhtml + 'table/' + url_trans + 'for-each'
  340. itm_fe = dict([ (x.attrib['select'], x) for x in mod_fe[mod].findall(xslpath) ])
  341. self.assertTrue(k in itm_fe)
  342. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  343. + url_xhtml + 'a'
  344. itm_a = [ x.attrib['href'] for x in itm_fe[k].findall(xslpath) ]
  345. self.assertTrue(stats_httpd.XML_URL_PATH + '/' + mod + '/' + k in itm_a)
  346. for itms in v:
  347. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  348. + url_xhtml + 'table/' + url_trans + 'for-each'
  349. itm_fe = dict([ (x.attrib['select'], x) for x in itm_fe[k].findall(xslpath) ])
  350. self.assertTrue('zones' in itm_fe)
  351. for (k, v) in itms.items():
  352. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  353. + url_xhtml + 'table/' + url_xhtml + 'tr/' \
  354. + url_xhtml + 'td/' + url_trans + 'value-of'
  355. itm_vo = [ x.attrib['select'] for x in itm_fe['zones'].findall(xslpath) ]
  356. self.assertTrue(k in itm_vo)
  357. else:
  358. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  359. + url_xhtml + 'table/' + url_xhtml + 'tr/' \
  360. + url_xhtml + 'td/' + url_trans + 'value-of'
  361. itm_vo = [ x.attrib['select'] for x in mod_fe[mod].findall(xslpath) ]
  362. self.assertTrue(k in itm_vo)
  363. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  364. + url_xhtml + 'table/' + url_xhtml + 'tr/' \
  365. + url_xhtml + 'td/' + url_xhtml + 'a'
  366. itm_a = [ x.attrib['href'] for x in mod_fe[mod].findall(xslpath) ]
  367. self.assertTrue(stats_httpd.XML_URL_PATH + '/' + mod + '/' + k in itm_a)
  368. elif item is None:
  369. xslpath = url_trans + 'template/' + url_xhtml + 'table/' + url_trans + 'for-each'
  370. mod_fe = dict([ (x.attrib['select'], x) for x in root.findall(xslpath) ])
  371. self.assertTrue(mod in mod_fe)
  372. for (k, v) in DUMMY_DATA[mod].items():
  373. if type(v) is list:
  374. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  375. + url_xhtml + 'table/' + url_trans + 'for-each'
  376. itm_fe = dict([ (x.attrib['select'], x) for x in mod_fe[mod].findall(xslpath) ])
  377. self.assertTrue(k in itm_fe)
  378. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  379. + url_xhtml + 'a'
  380. itm_a = [ x.attrib['href'] for x in itm_fe[k].findall(xslpath) ]
  381. self.assertTrue(stats_httpd.XML_URL_PATH + '/' + mod + '/' + k in itm_a)
  382. for itms in v:
  383. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  384. + url_xhtml + 'table/' + url_trans + 'for-each'
  385. itm_fe = dict([ (x.attrib['select'], x) for x in itm_fe[k].findall(xslpath) ])
  386. self.assertTrue('zones' in itm_fe)
  387. for (k, v) in itms.items():
  388. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  389. + url_xhtml + 'table/' + url_xhtml + 'tr/' \
  390. + url_xhtml + 'td/' + url_trans + 'value-of'
  391. itm_vo = [ x.attrib['select'] for x in itm_fe['zones'].findall(xslpath) ]
  392. self.assertTrue(k in itm_vo)
  393. else:
  394. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  395. + url_xhtml + 'table/' + url_xhtml + 'tr/' \
  396. + url_xhtml + 'td/' + url_trans + 'value-of'
  397. itm_vo = [ x.attrib['select'] for x in mod_fe[mod].findall(xslpath) ]
  398. self.assertTrue(k in itm_vo)
  399. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  400. + url_xhtml + 'table/' + url_xhtml + 'tr/' \
  401. + url_xhtml + 'td/' + url_xhtml + 'a'
  402. itm_a = [ x.attrib['href'] for x in mod_fe[mod].findall(xslpath) ]
  403. self.assertTrue(stats_httpd.XML_URL_PATH + '/' + mod + '/' + k in itm_a)
  404. else:
  405. xslpath = url_trans + 'template/' + url_xhtml + 'table/' + url_trans + 'for-each'
  406. mod_fe = dict([ (x.attrib['select'], x) for x in root.findall(xslpath) ])
  407. self.assertTrue(mod in mod_fe)
  408. if type(DUMMY_DATA[mod][item]) is list:
  409. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  410. + url_xhtml + 'table/' + url_trans + 'for-each'
  411. itm_fe = dict([ (x.attrib['select'], x) for x in mod_fe[mod].findall(xslpath) ])
  412. self.assertTrue(item in itm_fe)
  413. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  414. + url_xhtml + 'a'
  415. itm_a = [ x.attrib['href'] for x in itm_fe[item].findall(xslpath) ]
  416. self.assertTrue(stats_httpd.XML_URL_PATH + '/' + mod + '/' + item in itm_a)
  417. for itms in DUMMY_DATA[mod][item]:
  418. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  419. + url_xhtml + 'table/' + url_trans + 'for-each'
  420. itm_fe = dict([ (x.attrib['select'], x) for x in itm_fe[item].findall(xslpath) ])
  421. self.assertTrue('zones' in itm_fe)
  422. for (k, v) in itms.items():
  423. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  424. + url_xhtml + 'table/' + url_xhtml + 'tr/' \
  425. + url_xhtml + 'td/' + url_trans + 'value-of'
  426. itm_vo = [ x.attrib['select'] for x in itm_fe['zones'].findall(xslpath) ]
  427. self.assertTrue(k in itm_vo)
  428. else:
  429. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  430. + url_xhtml + 'table/' + url_xhtml + 'tr/' \
  431. + url_xhtml + 'td/' + url_trans + 'value-of'
  432. itm_vo = [ x.attrib['select'] for x in mod_fe[mod].findall(xslpath) ]
  433. self.assertTrue(item in itm_vo)
  434. xslpath = url_xhtml + 'tr/' + url_xhtml + 'td/' \
  435. + url_xhtml + 'table/' + url_xhtml + 'tr/' \
  436. + url_xhtml + 'td/' + url_xhtml + 'a'
  437. itm_a = [ x.attrib['href'] for x in mod_fe[mod].findall(xslpath) ]
  438. self.assertTrue(stats_httpd.XML_URL_PATH + '/' + mod + '/' + item in itm_a)
  439. # URL is '/bind10/statistics/xsl'
  440. check_XSL_URL_PATH(mod=None, item=None)
  441. for m in DUMMY_DATA:
  442. # URL is '/bind10/statistics/xsl/Module'
  443. check_XSL_URL_PATH(mod=m)
  444. for k in DUMMY_DATA[m].keys():
  445. # URL is '/bind10/statistics/xsl/Module/Item'
  446. check_XSL_URL_PATH(mod=m, item=k)
  447. # 302 redirect
  448. self.client._http_vsn_str = 'HTTP/1.1'
  449. self.client.putrequest('GET', '/')
  450. self.client.putheader('Host', self.address)
  451. self.client.endheaders()
  452. response = self.client.getresponse()
  453. self.assertEqual(response.status, 302)
  454. self.assertEqual(response.getheader('Location'),
  455. "http://%s:%d%s" % (self.address, self.port, stats_httpd.XML_URL_PATH))
  456. # 404 NotFound (random path)
  457. self.client._http_vsn_str = 'HTTP/1.0'
  458. self.client.putrequest('GET', '/path/to/foo/bar')
  459. self.client.endheaders()
  460. response = self.client.getresponse()
  461. self.assertEqual(response.status, 404)
  462. self.client._http_vsn_str = 'HTTP/1.0'
  463. self.client.putrequest('GET', '/bind10/foo')
  464. self.client.endheaders()
  465. response = self.client.getresponse()
  466. self.assertEqual(response.status, 404)
  467. self.client._http_vsn_str = 'HTTP/1.0'
  468. self.client.putrequest('GET', '/bind10/statistics/foo')
  469. self.client.endheaders()
  470. response = self.client.getresponse()
  471. self.assertEqual(response.status, 404)
  472. self.client._http_vsn_str = 'HTTP/1.0'
  473. self.client.putrequest('GET', stats_httpd.XML_URL_PATH + 'Auth') # with no slash
  474. self.client.endheaders()
  475. response = self.client.getresponse()
  476. self.assertEqual(response.status, 404)
  477. # 200 ok
  478. self.client._http_vsn_str = 'HTTP/1.0'
  479. self.client.putrequest('GET', stats_httpd.XML_URL_PATH + '/')
  480. self.client.endheaders()
  481. response = self.client.getresponse()
  482. self.assertEqual(response.status, 200)
  483. self.client._http_vsn_str = 'HTTP/1.0'
  484. self.client.putrequest('GET', stats_httpd.XML_URL_PATH + '#foo')
  485. self.client.endheaders()
  486. response = self.client.getresponse()
  487. self.assertEqual(response.status, 200)
  488. self.client._http_vsn_str = 'HTTP/1.0'
  489. self.client.putrequest('GET', stats_httpd.XML_URL_PATH + '?foo=bar')
  490. self.client.endheaders()
  491. response = self.client.getresponse()
  492. self.assertEqual(response.status, 200)
  493. # 404 NotFound (too long path)
  494. self.client._http_vsn_str = 'HTTP/1.0'
  495. self.client.putrequest('GET', stats_httpd.XML_URL_PATH + '/Boss/boot_time/a')
  496. self.client.endheaders()
  497. response = self.client.getresponse()
  498. self.assertEqual(response.status, 404)
  499. # 404 NotFound (nonexistent module name)
  500. self.client._http_vsn_str = 'HTTP/1.0'
  501. self.client.putrequest('GET', stats_httpd.XML_URL_PATH + '/Foo')
  502. self.client.endheaders()
  503. response = self.client.getresponse()
  504. self.assertEqual(response.status, 404)
  505. self.client._http_vsn_str = 'HTTP/1.0'
  506. self.client.putrequest('GET', stats_httpd.XSD_URL_PATH + '/Foo')
  507. self.client.endheaders()
  508. response = self.client.getresponse()
  509. self.assertEqual(response.status, 404)
  510. self.client._http_vsn_str = 'HTTP/1.0'
  511. self.client.putrequest('GET', stats_httpd.XSL_URL_PATH + '/Foo')
  512. self.client.endheaders()
  513. response = self.client.getresponse()
  514. self.assertEqual(response.status, 404)
  515. # 404 NotFound (nonexistent item name)
  516. self.client._http_vsn_str = 'HTTP/1.0'
  517. self.client.putrequest('GET', stats_httpd.XML_URL_PATH + '/Foo/bar')
  518. self.client.endheaders()
  519. response = self.client.getresponse()
  520. self.assertEqual(response.status, 404)
  521. self.client._http_vsn_str = 'HTTP/1.0'
  522. self.client.putrequest('GET', stats_httpd.XSD_URL_PATH + '/Foo/bar')
  523. self.client.endheaders()
  524. response = self.client.getresponse()
  525. self.assertEqual(response.status, 404)
  526. self.client._http_vsn_str = 'HTTP/1.0'
  527. self.client.putrequest('GET', stats_httpd.XSL_URL_PATH + '/Foo/bar')
  528. self.client.endheaders()
  529. response = self.client.getresponse()
  530. self.assertEqual(response.status, 404)
  531. # 404 NotFound (existent module but nonexistent item name)
  532. self.client._http_vsn_str = 'HTTP/1.0'
  533. self.client.putrequest('GET', stats_httpd.XML_URL_PATH + '/Auth/bar')
  534. self.client.endheaders()
  535. response = self.client.getresponse()
  536. self.assertEqual(response.status, 404)
  537. self.client._http_vsn_str = 'HTTP/1.0'
  538. self.client.putrequest('GET', stats_httpd.XSD_URL_PATH + '/Auth/bar')
  539. self.client.endheaders()
  540. response = self.client.getresponse()
  541. self.assertEqual(response.status, 404)
  542. self.client._http_vsn_str = 'HTTP/1.0'
  543. self.client.putrequest('GET', stats_httpd.XSL_URL_PATH + '/Auth/bar')
  544. self.client.endheaders()
  545. response = self.client.getresponse()
  546. self.assertEqual(response.status, 404)
  547. def test_do_GET_failed1(self):
  548. # checks status
  549. self.assertEqual(send_command("status", "Stats"),
  550. (0, "Stats is up. (PID " + str(os.getpid()) + ")"))
  551. # failure case(Stats is down)
  552. self.assertTrue(self.stats.running)
  553. self.assertEqual(send_shutdown("Stats"), (0, None)) # Stats is down
  554. self.assertFalse(self.stats.running)
  555. self.stats_httpd.cc_session.set_timeout(milliseconds=100)
  556. # request XML
  557. self.client.putrequest('GET', stats_httpd.XML_URL_PATH)
  558. self.client.endheaders()
  559. response = self.client.getresponse()
  560. self.assertEqual(response.status, 500)
  561. # request XSD
  562. self.client.putrequest('GET', stats_httpd.XSD_URL_PATH)
  563. self.client.endheaders()
  564. response = self.client.getresponse()
  565. self.assertEqual(response.status, 500)
  566. # request XSL
  567. self.client.putrequest('GET', stats_httpd.XSL_URL_PATH)
  568. self.client.endheaders()
  569. response = self.client.getresponse()
  570. self.assertEqual(response.status, 500)
  571. def test_do_GET_failed2(self):
  572. # failure case(Stats replies an error)
  573. self.stats.mccs.set_command_handler(
  574. lambda cmd, args: \
  575. isc.config.ccsession.create_answer(1, "specified arguments are incorrect: I have an error.")
  576. )
  577. # request XML
  578. self.client.putrequest('GET', stats_httpd.XML_URL_PATH)
  579. self.client.endheaders()
  580. response = self.client.getresponse()
  581. self.assertEqual(response.status, 404)
  582. # request XSD
  583. self.client.putrequest('GET', stats_httpd.XSD_URL_PATH)
  584. self.client.endheaders()
  585. response = self.client.getresponse()
  586. self.assertEqual(response.status, 404)
  587. # request XSL
  588. self.client.putrequest('GET', stats_httpd.XSL_URL_PATH)
  589. self.client.endheaders()
  590. response = self.client.getresponse()
  591. self.assertEqual(response.status, 404)
  592. def test_do_HEAD(self):
  593. self.client.putrequest('HEAD', stats_httpd.XML_URL_PATH)
  594. self.client.endheaders()
  595. response = self.client.getresponse()
  596. self.assertEqual(response.status, 200)
  597. self.client.putrequest('HEAD', '/path/to/foo/bar')
  598. self.client.endheaders()
  599. response = self.client.getresponse()
  600. self.assertEqual(response.status, 404)
  601. class TestHttpServerError(unittest.TestCase):
  602. """Tests for HttpServerError exception"""
  603. def test_raises(self):
  604. try:
  605. raise stats_httpd.HttpServerError('Nothing')
  606. except stats_httpd.HttpServerError as err:
  607. self.assertEqual(str(err), 'Nothing')
  608. class TestHttpServer(unittest.TestCase):
  609. """Tests for HttpServer class"""
  610. def setUp(self):
  611. # set the signal handler for deadlock
  612. self.sig_handler = SignalHandler(self.fail)
  613. self.base = BaseModules()
  614. def tearDown(self):
  615. if hasattr(self, "stats_httpd"):
  616. self.stats_httpd.stop()
  617. self.base.shutdown()
  618. # reset the signal handler
  619. self.sig_handler.reset()
  620. def test_httpserver(self):
  621. self.stats_httpd = MyStatsHttpd(get_availaddr())
  622. self.assertEqual(type(self.stats_httpd.httpd), list)
  623. self.assertEqual(len(self.stats_httpd.httpd), 1)
  624. for httpd in self.stats_httpd.httpd:
  625. self.assertTrue(isinstance(httpd, stats_httpd.HttpServer))
  626. class TestStatsHttpdError(unittest.TestCase):
  627. """Tests for StatsHttpdError exception"""
  628. def test_raises1(self):
  629. try:
  630. raise stats_httpd.StatsHttpdError('Nothing')
  631. except stats_httpd.StatsHttpdError as err:
  632. self.assertEqual(str(err), 'Nothing')
  633. def test_raises2(self):
  634. try:
  635. raise stats_httpd.StatsHttpdDataError('Nothing')
  636. except stats_httpd.StatsHttpdDataError as err:
  637. self.assertEqual(str(err), 'Nothing')
  638. class TestStatsHttpd(unittest.TestCase):
  639. """Tests for StatsHttpd class"""
  640. def setUp(self):
  641. # set the signal handler for deadlock
  642. self.sig_handler = SignalHandler(self.fail)
  643. self.base = BaseModules()
  644. self.stats_server = ThreadingServerManager(MyStats)
  645. self.stats_server.run()
  646. # checking IPv6 enabled on this platform
  647. self.ipv6_enabled = is_ipv6_enabled()
  648. def tearDown(self):
  649. if hasattr(self, "stats_httpd"):
  650. self.stats_httpd.stop()
  651. self.stats_server.shutdown()
  652. self.base.shutdown()
  653. # reset the signal handler
  654. self.sig_handler.reset()
  655. def test_init(self):
  656. server_address = get_availaddr()
  657. self.stats_httpd = MyStatsHttpd(server_address)
  658. self.assertEqual(self.stats_httpd.running, False)
  659. self.assertEqual(self.stats_httpd.poll_intval, 0.5)
  660. self.assertNotEqual(len(self.stats_httpd.httpd), 0)
  661. self.assertEqual(type(self.stats_httpd.mccs), isc.config.ModuleCCSession)
  662. self.assertEqual(type(self.stats_httpd.cc_session), isc.cc.Session)
  663. self.assertEqual(len(self.stats_httpd.config), 2)
  664. self.assertTrue('listen_on' in self.stats_httpd.config)
  665. self.assertEqual(len(self.stats_httpd.config['listen_on']), 1)
  666. self.assertTrue('address' in self.stats_httpd.config['listen_on'][0])
  667. self.assertTrue('port' in self.stats_httpd.config['listen_on'][0])
  668. self.assertTrue(server_address in set(self.stats_httpd.http_addrs))
  669. ans = send_command(
  670. isc.config.ccsession.COMMAND_GET_MODULE_SPEC,
  671. "ConfigManager", {"module_name":"StatsHttpd"})
  672. # assert StatsHttpd is added to ConfigManager
  673. self.assertNotEqual(ans, (0,{}))
  674. self.assertTrue(ans[1]['module_name'], 'StatsHttpd')
  675. def test_init_hterr(self):
  676. orig_open_httpd = stats_httpd.StatsHttpd.open_httpd
  677. def err_open_httpd(arg): raise stats_httpd.HttpServerError
  678. stats_httpd.StatsHttpd.open_httpd = err_open_httpd
  679. self.assertRaises(stats_httpd.HttpServerError, stats_httpd.StatsHttpd)
  680. ans = send_command(
  681. isc.config.ccsession.COMMAND_GET_MODULE_SPEC,
  682. "ConfigManager", {"module_name":"StatsHttpd"})
  683. # assert StatsHttpd is removed from ConfigManager
  684. self.assertEqual(ans, (0,{}))
  685. stats_httpd.StatsHttpd.open_httpd = orig_open_httpd
  686. def test_openclose_mccs(self):
  687. self.stats_httpd = MyStatsHttpd(get_availaddr())
  688. mccs = MockModuleCCSession()
  689. self.stats_httpd.mccs = mccs
  690. self.assertFalse(self.stats_httpd.mccs.stopped)
  691. self.assertFalse(self.stats_httpd.mccs.closed)
  692. self.stats_httpd.close_mccs()
  693. self.assertTrue(mccs.stopped)
  694. self.assertTrue(mccs.closed)
  695. self.assertEqual(self.stats_httpd.mccs, None)
  696. self.stats_httpd.open_mccs()
  697. self.assertIsNotNone(self.stats_httpd.mccs)
  698. self.stats_httpd.mccs = None
  699. self.assertEqual(self.stats_httpd.mccs, None)
  700. self.assertEqual(self.stats_httpd.close_mccs(), None)
  701. def test_mccs(self):
  702. self.stats_httpd = MyStatsHttpd(get_availaddr())
  703. self.assertIsNotNone(self.stats_httpd.mccs.get_socket())
  704. self.assertTrue(
  705. isinstance(self.stats_httpd.mccs.get_socket(), socket.socket))
  706. self.assertTrue(
  707. isinstance(self.stats_httpd.cc_session, isc.cc.session.Session))
  708. statistics_spec = self.stats_httpd.get_stats_spec()
  709. for mod in DUMMY_DATA:
  710. self.assertTrue(mod in statistics_spec)
  711. for cfg in statistics_spec[mod]:
  712. self.assertTrue('item_name' in cfg)
  713. self.assertTrue(cfg['item_name'] in DUMMY_DATA[mod])
  714. self.assertTrue(len(statistics_spec[mod]), len(DUMMY_DATA[mod]))
  715. self.stats_httpd.close_mccs()
  716. self.assertIsNone(self.stats_httpd.mccs)
  717. def test_httpd(self):
  718. # dual stack (addresses is ipv4 and ipv6)
  719. if self.ipv6_enabled:
  720. server_addresses = (get_availaddr('::1'), get_availaddr())
  721. self.stats_httpd = MyStatsHttpd(*server_addresses)
  722. for ht in self.stats_httpd.httpd:
  723. self.assertTrue(isinstance(ht, stats_httpd.HttpServer))
  724. self.assertTrue(ht.address_family in set([socket.AF_INET, socket.AF_INET6]))
  725. self.assertTrue(isinstance(ht.socket, socket.socket))
  726. # dual stack (address is ipv6)
  727. if self.ipv6_enabled:
  728. server_addresses = get_availaddr('::1')
  729. self.stats_httpd = MyStatsHttpd(server_addresses)
  730. for ht in self.stats_httpd.httpd:
  731. self.assertTrue(isinstance(ht, stats_httpd.HttpServer))
  732. self.assertEqual(ht.address_family, socket.AF_INET6)
  733. self.assertTrue(isinstance(ht.socket, socket.socket))
  734. # dual/single stack (address is ipv4)
  735. server_addresses = get_availaddr()
  736. self.stats_httpd = MyStatsHttpd(server_addresses)
  737. for ht in self.stats_httpd.httpd:
  738. self.assertTrue(isinstance(ht, stats_httpd.HttpServer))
  739. self.assertEqual(ht.address_family, socket.AF_INET)
  740. self.assertTrue(isinstance(ht.socket, socket.socket))
  741. # any address (IPv4)
  742. server_addresses = get_availaddr(address='0.0.0.0')
  743. self.stats_httpd = MyStatsHttpd(server_addresses)
  744. for ht in self.stats_httpd.httpd:
  745. self.assertTrue(isinstance(ht, stats_httpd.HttpServer))
  746. self.assertEqual(ht.address_family,socket.AF_INET)
  747. self.assertTrue(isinstance(ht.socket, socket.socket))
  748. # any address (IPv6)
  749. if self.ipv6_enabled:
  750. server_addresses = get_availaddr(address='::')
  751. self.stats_httpd = MyStatsHttpd(server_addresses)
  752. for ht in self.stats_httpd.httpd:
  753. self.assertTrue(isinstance(ht, stats_httpd.HttpServer))
  754. self.assertEqual(ht.address_family,socket.AF_INET6)
  755. self.assertTrue(isinstance(ht.socket, socket.socket))
  756. # existent hostname
  757. self.assertRaises(stats_httpd.HttpServerError, MyStatsHttpd,
  758. get_availaddr(address='localhost'))
  759. # nonexistent hostname
  760. self.assertRaises(stats_httpd.HttpServerError, MyStatsHttpd, ('my.host.domain', 8000))
  761. # over flow of port number
  762. self.assertRaises(stats_httpd.HttpServerError, MyStatsHttpd, ('127.0.0.1', 80000))
  763. # negative
  764. self.assertRaises(stats_httpd.HttpServerError, MyStatsHttpd, ('127.0.0.1', -8000))
  765. # alphabet
  766. self.assertRaises(stats_httpd.HttpServerError, MyStatsHttpd, ('127.0.0.1', 'ABCDE'))
  767. # Address already in use
  768. server_addresses = get_availaddr()
  769. self.stats_httpd_server = ThreadingServerManager(MyStatsHttpd, server_addresses)
  770. self.stats_httpd_server.run()
  771. self.assertRaises(stats_httpd.HttpServerError, MyStatsHttpd, server_addresses)
  772. send_shutdown("StatsHttpd")
  773. def test_running(self):
  774. self.stats_httpd_server = ThreadingServerManager(MyStatsHttpd, get_availaddr())
  775. self.stats_httpd = self.stats_httpd_server.server
  776. self.assertFalse(self.stats_httpd.running)
  777. self.stats_httpd_server.run()
  778. self.assertEqual(send_command("status", "StatsHttpd"),
  779. (0, "Stats Httpd is up. (PID " + str(os.getpid()) + ")"))
  780. self.assertTrue(self.stats_httpd.running)
  781. self.assertEqual(send_shutdown("StatsHttpd"), (0, None))
  782. self.assertFalse(self.stats_httpd.running)
  783. self.stats_httpd_server.shutdown()
  784. # failure case
  785. self.stats_httpd = MyStatsHttpd(get_availaddr())
  786. self.stats_httpd.cc_session.close()
  787. self.assertRaises(ValueError, self.stats_httpd.start)
  788. def test_failure_with_a_select_error (self):
  789. """checks select.error is raised if the exception except
  790. errno.EINTR is raised while it's selecting"""
  791. def raise_select_except(*args):
  792. raise select.error('dummy error')
  793. orig_select = stats_httpd.select.select
  794. stats_httpd.select.select = raise_select_except
  795. self.stats_httpd = MyStatsHttpd(get_availaddr())
  796. self.assertRaises(select.error, self.stats_httpd.start)
  797. stats_httpd.select.select = orig_select
  798. def test_nofailure_with_errno_EINTR(self):
  799. """checks no exception is raised if errno.EINTR is raised
  800. while it's selecting"""
  801. def raise_select_except(*args):
  802. raise select.error(errno.EINTR)
  803. orig_select = stats_httpd.select.select
  804. stats_httpd.select.select = raise_select_except
  805. self.stats_httpd_server = ThreadingServerManager(MyStatsHttpd, get_availaddr())
  806. self.stats_httpd_server.run()
  807. self.stats_httpd_server.shutdown()
  808. stats_httpd.select.select = orig_select
  809. def test_open_template(self):
  810. self.stats_httpd = MyStatsHttpd(get_availaddr())
  811. # successful conditions
  812. tmpl = self.stats_httpd.open_template(stats_httpd.XML_TEMPLATE_LOCATION)
  813. self.assertTrue(isinstance(tmpl, string.Template))
  814. opts = dict(
  815. xml_string="<dummy></dummy>",
  816. xsl_url_path="/path/to/")
  817. lines = tmpl.substitute(opts)
  818. for n in opts:
  819. self.assertTrue(lines.find(opts[n])>0)
  820. tmpl = self.stats_httpd.open_template(stats_httpd.XSD_TEMPLATE_LOCATION)
  821. self.assertTrue(isinstance(tmpl, string.Template))
  822. opts = dict(xsd_string="<dummy></dummy>")
  823. lines = tmpl.substitute(opts)
  824. for n in opts:
  825. self.assertTrue(lines.find(opts[n])>0)
  826. tmpl = self.stats_httpd.open_template(stats_httpd.XSL_TEMPLATE_LOCATION)
  827. self.assertTrue(isinstance(tmpl, string.Template))
  828. opts = dict(
  829. xsl_string="<dummy></dummy>",
  830. xsd_namespace="http://host/path/to/")
  831. lines = tmpl.substitute(opts)
  832. for n in opts:
  833. self.assertTrue(lines.find(opts[n])>0)
  834. # unsuccessful condition
  835. self.assertRaises(
  836. IOError,
  837. self.stats_httpd.open_template, '/path/to/foo/bar')
  838. def test_commands(self):
  839. self.stats_httpd = MyStatsHttpd(get_availaddr())
  840. self.assertEqual(self.stats_httpd.command_handler("status", None),
  841. isc.config.ccsession.create_answer(
  842. 0, "Stats Httpd is up. (PID " + str(os.getpid()) + ")"))
  843. self.stats_httpd.running = True
  844. self.assertEqual(self.stats_httpd.command_handler("shutdown", None),
  845. isc.config.ccsession.create_answer(0))
  846. self.assertFalse(self.stats_httpd.running)
  847. self.assertEqual(
  848. self.stats_httpd.command_handler("__UNKNOWN_COMMAND__", None),
  849. isc.config.ccsession.create_answer(
  850. 1, "Unknown command: __UNKNOWN_COMMAND__"))
  851. def test_config(self):
  852. self.stats_httpd = MyStatsHttpd(get_availaddr())
  853. self.assertEqual(
  854. self.stats_httpd.config_handler(dict(_UNKNOWN_KEY_=None)),
  855. isc.config.ccsession.create_answer(
  856. 1, "unknown item _UNKNOWN_KEY_"))
  857. addresses = get_availaddr()
  858. self.assertEqual(
  859. self.stats_httpd.config_handler(
  860. dict(listen_on=[dict(address=addresses[0],port=addresses[1])])),
  861. isc.config.ccsession.create_answer(0))
  862. self.assertTrue("listen_on" in self.stats_httpd.config)
  863. for addr in self.stats_httpd.config["listen_on"]:
  864. self.assertTrue("address" in addr)
  865. self.assertTrue("port" in addr)
  866. self.assertTrue(addr["address"] == addresses[0])
  867. self.assertTrue(addr["port"] == addresses[1])
  868. if self.ipv6_enabled:
  869. addresses = get_availaddr("::1")
  870. self.assertEqual(
  871. self.stats_httpd.config_handler(
  872. dict(listen_on=[dict(address=addresses[0],port=addresses[1])])),
  873. isc.config.ccsession.create_answer(0))
  874. self.assertTrue("listen_on" in self.stats_httpd.config)
  875. for addr in self.stats_httpd.config["listen_on"]:
  876. self.assertTrue("address" in addr)
  877. self.assertTrue("port" in addr)
  878. self.assertTrue(addr["address"] == addresses[0])
  879. self.assertTrue(addr["port"] == addresses[1])
  880. addresses = get_availaddr()
  881. self.assertEqual(
  882. self.stats_httpd.config_handler(
  883. dict(listen_on=[dict(address=addresses[0],port=addresses[1])])),
  884. isc.config.ccsession.create_answer(0))
  885. self.assertTrue("listen_on" in self.stats_httpd.config)
  886. for addr in self.stats_httpd.config["listen_on"]:
  887. self.assertTrue("address" in addr)
  888. self.assertTrue("port" in addr)
  889. self.assertTrue(addr["address"] == addresses[0])
  890. self.assertTrue(addr["port"] == addresses[1])
  891. (ret, arg) = isc.config.ccsession.parse_answer(
  892. self.stats_httpd.config_handler(
  893. dict(listen_on=[dict(address="1.2.3.4",port=543210)]))
  894. )
  895. self.assertEqual(ret, 1)
  896. def test_xml_handler(self):
  897. self.stats_httpd = MyStatsHttpd(get_availaddr())
  898. module_name = 'Dummy'
  899. stats_spec = \
  900. { module_name :
  901. [{
  902. "item_name": "foo",
  903. "item_type": "string",
  904. "item_optional": False,
  905. "item_default": "bar",
  906. "item_description": "foo is bar",
  907. "item_title": "Foo"
  908. },
  909. {
  910. "item_name": "foo2",
  911. "item_type": "list",
  912. "item_optional": False,
  913. "item_default": [
  914. {
  915. "zonename" : "test1",
  916. "queries.udp" : 1,
  917. "queries.tcp" : 2
  918. },
  919. {
  920. "zonename" : "test2",
  921. "queries.udp" : 3,
  922. "queries.tcp" : 4
  923. }
  924. ],
  925. "item_title": "Foo bar",
  926. "item_description": "Foo bar",
  927. "list_item_spec": {
  928. "item_name": "foo2-1",
  929. "item_type": "map",
  930. "item_optional": False,
  931. "item_default": {},
  932. "map_item_spec": [
  933. {
  934. "item_name": "foo2-1-1",
  935. "item_type": "string",
  936. "item_optional": False,
  937. "item_default": "",
  938. "item_title": "Foo2 1 1",
  939. "item_description": "Foo bar"
  940. },
  941. {
  942. "item_name": "foo2-1-2",
  943. "item_type": "integer",
  944. "item_optional": False,
  945. "item_default": 0,
  946. "item_title": "Foo2 1 2",
  947. "item_description": "Foo bar"
  948. },
  949. {
  950. "item_name": "foo2-1-3",
  951. "item_type": "integer",
  952. "item_optional": False,
  953. "item_default": 0,
  954. "item_title": "Foo2 1 3",
  955. "item_description": "Foo bar"
  956. }
  957. ]
  958. }
  959. }]
  960. }
  961. stats_data = \
  962. { module_name : { 'foo':'bar',
  963. 'foo2': [
  964. {
  965. "foo2-1-1" : "bar1",
  966. "foo2-1-2" : 10,
  967. "foo2-1-3" : 9
  968. },
  969. {
  970. "foo2-1-1" : "bar2",
  971. "foo2-1-2" : 8,
  972. "foo2-1-3" : 7
  973. }
  974. ] } }
  975. self.stats_httpd.get_stats_spec = lambda x,y: stats_spec
  976. self.stats_httpd.get_stats_data = lambda x,y: stats_data
  977. xml_string = self.stats_httpd.xml_handler()
  978. stats_xml = xml.etree.ElementTree.fromstring(xml_string)
  979. schema_loc = '{%s}schemaLocation' % XMLNS_XSI
  980. self.assertEqual(stats_xml.attrib[schema_loc],
  981. stats_httpd.XML_ROOT_ATTRIB['xsi:schemaLocation'])
  982. stats_data = stats_data[module_name]
  983. stats_spec = stats_spec[module_name]
  984. names = stats_httpd.item_name_list(stats_data, '')
  985. for i in range(0, len(names)):
  986. self.assertEqual('%s/%s' % (module_name, names[i]), stats_xml[i].attrib['identifier'])
  987. value = isc.cc.data.find(stats_data, names[i])
  988. if type(value) is int:
  989. value = str(value)
  990. if type(value) is dict or type(value) is list:
  991. self.assertFalse('value' in stats_xml[i].attrib)
  992. else:
  993. self.assertEqual(value, stats_xml[i].attrib['value'])
  994. self.assertEqual(module_name, stats_xml[i].attrib['owner'])
  995. self.assertEqual(urllib.parse.quote('%s/%s/%s' % (stats_httpd.XML_URL_PATH,
  996. module_name, names[i])),
  997. stats_xml[i].attrib['uri'])
  998. spec = isc.config.find_spec_part(stats_spec, names[i])
  999. self.assertEqual(spec['item_name'], stats_xml[i].attrib['name'])
  1000. self.assertEqual(spec['item_type'], stats_xml[i].attrib['type'])
  1001. self.assertEqual(spec['item_description'], stats_xml[i].attrib['description'])
  1002. self.assertEqual(spec['item_title'], stats_xml[i].attrib['title'])
  1003. self.assertEqual(str(spec['item_optional']).lower(), stats_xml[i].attrib['optional'])
  1004. default = spec['item_default']
  1005. if type(default) is int:
  1006. default = str(default)
  1007. if type(default) is dict or type(default) is list:
  1008. self.assertFalse('default' in stats_xml[i].attrib)
  1009. else:
  1010. self.assertEqual(default, stats_xml[i].attrib['default'])
  1011. self.assertFalse('item_format' in spec)
  1012. self.assertFalse('format' in stats_xml[i].attrib)
  1013. def test_xsd_handler(self):
  1014. self.stats_httpd = MyStatsHttpd(get_availaddr())
  1015. self.stats_httpd.get_stats_spec = lambda x,y: \
  1016. { "Dummy" :
  1017. [{
  1018. "item_name": "foo",
  1019. "item_type": "string",
  1020. "item_optional": False,
  1021. "item_default": "bar",
  1022. "item_description": "foo is bar",
  1023. "item_title": "Foo"
  1024. },
  1025. {
  1026. "item_name": "hoo_time",
  1027. "item_type": "string",
  1028. "item_optional": False,
  1029. "item_default": "2011-01-01T01:01:01Z",
  1030. "item_description": "hoo time",
  1031. "item_title": "Hoo Time",
  1032. "item_format": "date-time"
  1033. },
  1034. {
  1035. "item_name": "foo2",
  1036. "item_type": "list",
  1037. "item_optional": False,
  1038. "item_default": [
  1039. {
  1040. "zonename" : "test1",
  1041. "queries.udp" : 1,
  1042. "queries.tcp" : 2
  1043. },
  1044. {
  1045. "zonename" : "test2",
  1046. "queries.udp" : 3,
  1047. "queries.tcp" : 4
  1048. }
  1049. ],
  1050. "item_title": "Foo bar",
  1051. "item_description": "Foo bar",
  1052. "list_item_spec": {
  1053. "item_name": "foo2-1",
  1054. "item_type": "map",
  1055. "item_optional": False,
  1056. "item_default": {},
  1057. "map_item_spec": [
  1058. {
  1059. "item_name": "foo2-1-1",
  1060. "item_type": "string",
  1061. "item_optional": False,
  1062. "item_default": "",
  1063. "item_title": "Foo2 1 1",
  1064. "item_description": "Foo bar"
  1065. },
  1066. {
  1067. "item_name": "foo2-1-2",
  1068. "item_type": "integer",
  1069. "item_optional": False,
  1070. "item_default": 0,
  1071. "item_title": "Foo2 1 2",
  1072. "item_description": "Foo bar"
  1073. },
  1074. {
  1075. "item_name": "foo2-1-3",
  1076. "item_type": "integer",
  1077. "item_optional": False,
  1078. "item_default": 0,
  1079. "item_title": "Foo2 1 3",
  1080. "item_description": "Foo bar"
  1081. }
  1082. ]
  1083. }
  1084. }]
  1085. }
  1086. xsd_body1 = self.stats_httpd.open_template(
  1087. stats_httpd.XSD_TEMPLATE_LOCATION).substitute(
  1088. xsd_string='<schema targetNamespace="' + stats_httpd.XSD_NAMESPACE + '" xmlns="http://www.w3.org/2001/XMLSchema" xmlns:bind10="' + stats_httpd.XSD_NAMESPACE + '"><annotation><documentation>XML schema of the statistics data in BIND 10</documentation></annotation><element name="statistics"><annotation><documentation>A set of statistics data</documentation></annotation><complexType><all><element name="Dummy"><complexType><all><element maxOccurs="1" minOccurs="1" name="foo" type="string"><annotation><appinfo>Foo</appinfo><documentation>foo is bar</documentation></annotation></element><element maxOccurs="1" minOccurs="1" name="hoo_time" type="dateTime"><annotation><appinfo>Hoo Time</appinfo><documentation>hoo time</documentation></annotation></element><element maxOccurs="1" minOccurs="1" name="foo2"><complexType><sequence><element maxOccurs="unbounded" minOccurs="1" name="foo2-1"><complexType><all><element maxOccurs="1" minOccurs="1" name="foo2-1-1" type="string"><annotation><appinfo>Foo2 1 1</appinfo><documentation>Foo bar</documentation></annotation></element><element maxOccurs="1" minOccurs="1" name="foo2-1-2" type="integer"><annotation><appinfo>Foo2 1 2</appinfo><documentation>Foo bar</documentation></annotation></element><element maxOccurs="1" minOccurs="1" name="foo2-1-3" type="integer"><annotation><appinfo>Foo2 1 3</appinfo><documentation>Foo bar</documentation></annotation></element></all></complexType></element></sequence></complexType></element></all></complexType></element></all></complexType></element></schema>')
  1089. xsd_body2 = self.stats_httpd.xsd_handler()
  1090. self.assertEqual(type(xsd_body1), str)
  1091. self.assertEqual(type(xsd_body2), str)
  1092. self.assertEqual(xsd_body1, xsd_body2)
  1093. self.stats_httpd.get_stats_spec = lambda x,y: \
  1094. { "Dummy" :
  1095. [{
  1096. "item_name": "bar",
  1097. "item_type": "string",
  1098. "item_optional": False,
  1099. "item_default": "foo",
  1100. "item_description": "bar is foo",
  1101. "item_title": "bar"
  1102. },
  1103. {
  1104. "item_name": "boo_time",
  1105. "item_type": "string",
  1106. "item_optional": False,
  1107. "item_default": "2012-02-02T02:02:02Z",
  1108. "item_description": "boo time",
  1109. "item_title": "Boo Time",
  1110. "item_format": "date-time"
  1111. },
  1112. {
  1113. "item_name": "foo2",
  1114. "item_type": "list",
  1115. "item_optional": False,
  1116. "item_default": [
  1117. {
  1118. "zonename" : "test1",
  1119. "queries.udp" : 1,
  1120. "queries.tcp" : 2
  1121. },
  1122. {
  1123. "zonename" : "test2",
  1124. "queries.udp" : 3,
  1125. "queries.tcp" : 4
  1126. }
  1127. ],
  1128. "item_title": "Foo bar",
  1129. "item_description": "Foo bar",
  1130. "list_item_spec": {
  1131. "item_name": "foo2-1",
  1132. "item_type": "map",
  1133. "item_optional": False,
  1134. "item_default": {},
  1135. "map_item_spec": [
  1136. {
  1137. "item_name": "foo2-1-1",
  1138. "item_type": "string",
  1139. "item_optional": False,
  1140. "item_default": "",
  1141. "item_title": "Foo2 1 1",
  1142. "item_description": "Foo bar"
  1143. },
  1144. {
  1145. "item_name": "foo2-1-2",
  1146. "item_type": "integer",
  1147. "item_optional": False,
  1148. "item_default": 0,
  1149. "item_title": "Foo2 1 2",
  1150. "item_description": "Foo bar"
  1151. },
  1152. {
  1153. "item_name": "foo2-1-3",
  1154. "item_type": "integer",
  1155. "item_optional": False,
  1156. "item_default": 0,
  1157. "item_title": "Foo2 1 3",
  1158. "item_description": "Foo bar"
  1159. }
  1160. ]
  1161. }
  1162. }]
  1163. }
  1164. xsd_body2 = self.stats_httpd.xsd_handler()
  1165. self.assertNotEqual(xsd_body1, xsd_body2)
  1166. def test_xsl_handler(self):
  1167. self.stats_httpd = MyStatsHttpd(get_availaddr())
  1168. self.stats_httpd.get_stats_spec = lambda x,y: \
  1169. { "Dummy" :
  1170. [{
  1171. "item_name": "foo",
  1172. "item_type": "string",
  1173. "item_optional": False,
  1174. "item_default": "bar",
  1175. "item_description": "foo bar",
  1176. "item_title": "Foo"
  1177. },
  1178. {
  1179. "item_name": "foo2",
  1180. "item_type": "list",
  1181. "item_optional": False,
  1182. "item_default": [
  1183. {
  1184. "zonename" : "test1",
  1185. "queries.udp" : 1,
  1186. "queries.tcp" : 2
  1187. },
  1188. {
  1189. "zonename" : "test2",
  1190. "queries.udp" : 3,
  1191. "queries.tcp" : 4
  1192. }
  1193. ],
  1194. "item_title": "Foo bar",
  1195. "item_description": "Foo bar",
  1196. "list_item_spec": {
  1197. "item_name": "foo2-1",
  1198. "item_type": "map",
  1199. "item_optional": False,
  1200. "item_default": {},
  1201. "map_item_spec": [
  1202. {
  1203. "item_name": "foo2-1-1",
  1204. "item_type": "string",
  1205. "item_optional": False,
  1206. "item_default": "",
  1207. "item_title": "Foo2 1 1",
  1208. "item_description": "Foo bar"
  1209. },
  1210. {
  1211. "item_name": "foo2-1-2",
  1212. "item_type": "integer",
  1213. "item_optional": False,
  1214. "item_default": 0,
  1215. "item_title": "Foo2 1 2",
  1216. "item_description": "Foo bar"
  1217. },
  1218. {
  1219. "item_name": "foo2-1-3",
  1220. "item_type": "integer",
  1221. "item_optional": False,
  1222. "item_default": 0,
  1223. "item_title": "Foo2 1 3",
  1224. "item_description": "Foo bar"
  1225. }
  1226. ]
  1227. }
  1228. }]
  1229. }
  1230. xsl_body1 = self.stats_httpd.open_template(
  1231. stats_httpd.XSL_TEMPLATE_LOCATION).substitute(
  1232. xsl_string='<xsl:template match="bind10:statistics"><table><tr><th>Module Name</th><th>Module Item</th></tr><xsl:for-each select="Dummy"><tr><td><a href="' + stats_httpd.XML_URL_PATH + '/Dummy">Dummy</a></td><td><table><tr><th>Item Name</th><th>Item Value</th></tr><tr><td class="title" title="foo bar"><a href="' + stats_httpd.XML_URL_PATH + '/Dummy/foo">Foo</a></td><td><xsl:value-of select="foo" /></td></tr><xsl:for-each select="foo2"><tr><td class="title" title="Foo bar"><a href="' + stats_httpd.XML_URL_PATH + '/Dummy/foo2">Foo bar</a></td><td><table><tr><th>Item Name</th><th>Item Value</th></tr><xsl:for-each select="foo2-1"><tr><td class="title" title="">foo2-1</td><td><table><tr><th>Item Name</th><th>Item Value</th></tr><tr><td class="title" title="Foo bar">Foo2 1 1</td><td><xsl:value-of select="foo2-1-1" /></td></tr><tr><td class="title" title="Foo bar">Foo2 1 2</td><td><xsl:value-of select="foo2-1-2" /></td></tr><tr><td class="title" title="Foo bar">Foo2 1 3</td><td><xsl:value-of select="foo2-1-3" /></td></tr></table></td></tr></xsl:for-each></table></td></tr></xsl:for-each></table></td></tr></xsl:for-each></table></xsl:template>',
  1233. xsd_namespace=stats_httpd.XSD_NAMESPACE)
  1234. xsl_body2 = self.stats_httpd.xsl_handler()
  1235. self.assertEqual(type(xsl_body1), str)
  1236. self.assertEqual(type(xsl_body2), str)
  1237. self.assertEqual(xsl_body1, xsl_body2)
  1238. self.stats_httpd.get_stats_spec = lambda x,y: \
  1239. { "Dummy" :
  1240. [{
  1241. "item_name": "bar",
  1242. "item_type": "string",
  1243. "item_optional": False,
  1244. "item_default": "foo",
  1245. "item_description": "bar is foo",
  1246. "item_title": "bar"
  1247. }]
  1248. }
  1249. xsl_body2 = self.stats_httpd.xsl_handler()
  1250. self.assertNotEqual(xsl_body1, xsl_body2)
  1251. def test_for_without_B10_FROM_SOURCE(self):
  1252. # just lets it go through the code without B10_FROM_SOURCE env
  1253. # variable
  1254. if "B10_FROM_SOURCE" in os.environ:
  1255. tmppath = os.environ["B10_FROM_SOURCE"]
  1256. os.environ.pop("B10_FROM_SOURCE")
  1257. imp.reload(stats_httpd)
  1258. os.environ["B10_FROM_SOURCE"] = tmppath
  1259. imp.reload(stats_httpd)
  1260. if __name__ == "__main__":
  1261. unittest.main()