ccsession_test.py 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537
  1. # Copyright (C) 2010 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. # $Id$
  16. #
  17. # Tests for the ConfigData and MultiConfigData classes
  18. #
  19. import unittest
  20. import os
  21. from isc.config.ccsession import *
  22. from isc.config.config_data import BIND10_CONFIG_DATA_VERSION
  23. from unittest_fakesession import FakeModuleCCSession
  24. class TestHelperFunctions(unittest.TestCase):
  25. def test_parse_answer(self):
  26. self.assertRaises(ModuleCCSessionError, parse_answer, 1)
  27. self.assertRaises(ModuleCCSessionError, parse_answer, { 'just a dict': 1 })
  28. self.assertRaises(ModuleCCSessionError, parse_answer, { 'result': 1 })
  29. self.assertRaises(ModuleCCSessionError, parse_answer, { 'result': [] })
  30. self.assertRaises(ModuleCCSessionError, parse_answer, { 'result': [ 'not_an_rcode' ] })
  31. self.assertRaises(ModuleCCSessionError, parse_answer, { 'result': [ 1, 2 ] })
  32. rcode, val = parse_answer({ 'result': [ 0 ] })
  33. self.assertEqual(0, rcode)
  34. self.assertEqual(None, val)
  35. rcode, val = parse_answer({ 'result': [ 0, "something" ] })
  36. self.assertEqual(0, rcode)
  37. self.assertEqual("something", val)
  38. rcode, val = parse_answer({ 'result': [ 1, "some error" ] })
  39. self.assertEqual(1, rcode)
  40. self.assertEqual("some error", val)
  41. def test_create_answer(self):
  42. self.assertRaises(ModuleCCSessionError, create_answer, 'not_an_int')
  43. self.assertRaises(ModuleCCSessionError, create_answer, 1, 2)
  44. self.assertRaises(ModuleCCSessionError, create_answer, 1)
  45. self.assertEqual({ 'result': [ 0 ] }, create_answer(0))
  46. self.assertEqual({ 'result': [ 1, 'something bad' ] }, create_answer(1, 'something bad'))
  47. self.assertEqual({ 'result': [ 0, 'something good' ] }, create_answer(0, 'something good'))
  48. self.assertEqual({ 'result': [ 0, ['some', 'list' ] ] }, create_answer(0, ['some', 'list']))
  49. self.assertEqual({ 'result': [ 0, {'some': 'map' } ] }, create_answer(0, {'some': 'map'}))
  50. def test_parse_command(self):
  51. cmd, arg = parse_command(1)
  52. self.assertEqual(None, cmd)
  53. self.assertEqual(None, arg)
  54. cmd, arg = parse_command({})
  55. self.assertEqual(None, cmd)
  56. self.assertEqual(None, arg)
  57. cmd, arg = parse_command({ 'not a command': 1})
  58. self.assertEqual(None, cmd)
  59. self.assertEqual(None, arg)
  60. cmd, arg = parse_command({ 'command': 1})
  61. self.assertEqual(None, cmd)
  62. self.assertEqual(None, arg)
  63. cmd, arg = parse_command({ 'command': []})
  64. self.assertEqual(None, cmd)
  65. self.assertEqual(None, arg)
  66. cmd, arg = parse_command({ 'command': [ 1 ]})
  67. self.assertEqual(None, cmd)
  68. self.assertEqual(None, arg)
  69. cmd, arg = parse_command({ 'command': [ 'command' ]})
  70. self.assertEqual('command', cmd)
  71. self.assertEqual(None, arg)
  72. cmd, arg = parse_command({ 'command': [ 'command', 1 ]})
  73. self.assertEqual('command', cmd)
  74. self.assertEqual(1, arg)
  75. cmd, arg = parse_command({ 'command': [ 'command', ['some', 'argument', 'list'] ]})
  76. self.assertEqual('command', cmd)
  77. self.assertEqual(['some', 'argument', 'list'], arg)
  78. def test_create_command(self):
  79. self.assertRaises(ModuleCCSessionError, create_command, 1)
  80. self.assertEqual({'command': [ 'my_command' ]}, create_command('my_command'))
  81. self.assertEqual({'command': [ 'my_command', 1 ]}, create_command('my_command', 1))
  82. self.assertEqual({'command': [ 'my_command', [ 'some', 'list' ] ]}, create_command('my_command', [ 'some', 'list' ]))
  83. self.assertEqual({'command': [ 'my_command', { 'some': 'map' } ]}, create_command('my_command', { 'some': 'map' }))
  84. class TestModuleCCSession(unittest.TestCase):
  85. def setUp(self):
  86. if 'CONFIG_TESTDATA_PATH' in os.environ:
  87. self.data_path = os.environ['CONFIG_TESTDATA_PATH']
  88. else:
  89. self.data_path = "../../../testdata"
  90. def spec_file(self, file):
  91. return self.data_path + os.sep + file
  92. def create_session(self, spec_file_name, config_handler = None, command_handler = None, cc_session = None):
  93. return ModuleCCSession(self.spec_file(spec_file_name), config_handler, command_handler, cc_session)
  94. def test_init(self):
  95. fake_session = FakeModuleCCSession()
  96. mccs = self.create_session("spec1.spec", None, None, fake_session)
  97. self.assertEqual(isc.config.module_spec_from_file(self.spec_file("spec1.spec"))._module_spec, mccs.specification._module_spec)
  98. self.assertEqual(None, mccs._config_handler)
  99. self.assertEqual(None, mccs._command_handler)
  100. def test_start1(self):
  101. fake_session = FakeModuleCCSession()
  102. self.assertFalse("Spec1" in fake_session.subscriptions)
  103. mccs = self.create_session("spec1.spec", None, None, fake_session)
  104. self.assertTrue("Spec1" in fake_session.subscriptions)
  105. self.assertEqual(len(fake_session.message_queue), 0)
  106. self.assertRaises(ModuleCCSessionError, mccs.start)
  107. self.assertEqual(len(fake_session.message_queue), 2)
  108. self.assertEqual({'command': ['module_spec', {'module_name': 'Spec1'}]},
  109. fake_session.get_message('ConfigManager', None))
  110. self.assertEqual({'command': ['get_config', {'module_name': 'Spec1'}]},
  111. fake_session.get_message('ConfigManager', None))
  112. self.assertEqual(len(fake_session.message_queue), 0)
  113. fake_session.group_sendmsg({'result': [ 0 ]}, "Spec1")
  114. fake_session.group_sendmsg({'result': [ 0 ]}, "Spec1")
  115. mccs.start()
  116. self.assertEqual(len(fake_session.message_queue), 2)
  117. self.assertEqual({'command': ['module_spec', {'module_name': 'Spec1'}]},
  118. fake_session.get_message('ConfigManager', None))
  119. self.assertEqual({'command': ['get_config', {'module_name': 'Spec1'}]},
  120. fake_session.get_message('ConfigManager', None))
  121. mccs = None
  122. self.assertFalse("Spec1" in fake_session.subscriptions)
  123. def test_start2(self):
  124. fake_session = FakeModuleCCSession()
  125. mccs = self.create_session("spec2.spec", None, None, fake_session)
  126. self.assertEqual(len(fake_session.message_queue), 0)
  127. self.assertRaises(ModuleCCSessionError, mccs.start)
  128. self.assertEqual(len(fake_session.message_queue), 2)
  129. self.assertEqual({'command': ['module_spec', mccs.specification._module_spec]},
  130. fake_session.get_message('ConfigManager', None))
  131. self.assertEqual({'command': ['get_config', {'module_name': 'Spec2'}]},
  132. fake_session.get_message('ConfigManager', None))
  133. self.assertEqual(len(fake_session.message_queue), 0)
  134. fake_session.group_sendmsg({'result': [ 0 ]}, "Spec2")
  135. fake_session.group_sendmsg({'result': [ 0, {} ]}, "Spec2")
  136. mccs.start()
  137. self.assertEqual(len(fake_session.message_queue), 2)
  138. self.assertEqual({'command': ['module_spec', mccs.specification._module_spec]},
  139. fake_session.get_message('ConfigManager', None))
  140. self.assertEqual({'command': ['get_config', {'module_name': 'Spec2'}]},
  141. fake_session.get_message('ConfigManager', None))
  142. def test_start3(self):
  143. fake_session = FakeModuleCCSession()
  144. mccs = self.create_session("spec2.spec", None, None, fake_session)
  145. mccs.set_config_handler(self.my_config_handler_ok)
  146. self.assertEqual(len(fake_session.message_queue), 0)
  147. self.assertRaises(ModuleCCSessionError, mccs.start)
  148. self.assertEqual(len(fake_session.message_queue), 2)
  149. self.assertEqual({'command': ['module_spec', mccs.specification._module_spec]},
  150. fake_session.get_message('ConfigManager', None))
  151. self.assertEqual({'command': ['get_config', {'module_name': 'Spec2'}]},
  152. fake_session.get_message('ConfigManager', None))
  153. self.assertEqual(len(fake_session.message_queue), 0)
  154. fake_session.group_sendmsg({'result': [ 0 ]}, "Spec2")
  155. fake_session.group_sendmsg({'result': [ 0, {} ]}, "Spec2")
  156. mccs.start()
  157. self.assertEqual(len(fake_session.message_queue), 2)
  158. self.assertEqual({'command': ['module_spec', mccs.specification._module_spec]},
  159. fake_session.get_message('ConfigManager', None))
  160. self.assertEqual({'command': ['get_config', {'module_name': 'Spec2'}]},
  161. fake_session.get_message('ConfigManager', None))
  162. def test_start4(self):
  163. fake_session = FakeModuleCCSession()
  164. mccs = self.create_session("spec2.spec", None, None, fake_session)
  165. mccs.set_config_handler(self.my_config_handler_ok)
  166. self.assertEqual(len(fake_session.message_queue), 0)
  167. self.assertRaises(ModuleCCSessionError, mccs.start)
  168. self.assertEqual(len(fake_session.message_queue), 2)
  169. self.assertEqual({'command': ['module_spec', mccs.specification._module_spec]},
  170. fake_session.get_message('ConfigManager', None))
  171. self.assertEqual({'command': ['get_config', {'module_name': 'Spec2'}]},
  172. fake_session.get_message('ConfigManager', None))
  173. self.assertEqual(len(fake_session.message_queue), 0)
  174. fake_session.group_sendmsg({'result': [ 0 ]}, "Spec2")
  175. fake_session.group_sendmsg({'result': [ 1, "just an error" ]}, "Spec2")
  176. mccs.start()
  177. self.assertEqual(len(fake_session.message_queue), 2)
  178. self.assertEqual({'command': ['module_spec', mccs.specification._module_spec]},
  179. fake_session.get_message('ConfigManager', None))
  180. self.assertEqual({'command': ['get_config', {'module_name': 'Spec2'}]},
  181. fake_session.get_message('ConfigManager', None))
  182. def test_get_socket(self):
  183. fake_session = FakeModuleCCSession()
  184. mccs = self.create_session("spec1.spec", None, None, fake_session)
  185. self.assertNotEqual(None, mccs.get_socket())
  186. def test_get_session(self):
  187. fake_session = FakeModuleCCSession()
  188. mccs = self.create_session("spec1.spec", None, None, fake_session)
  189. self.assertEqual(fake_session, mccs._session)
  190. def test_close(self):
  191. fake_session = FakeModuleCCSession()
  192. mccs = self.create_session("spec1.spec", None, None, fake_session)
  193. mccs.close()
  194. self.assertEqual("closed", fake_session._socket)
  195. def my_config_handler_ok(self, new_config):
  196. return isc.config.ccsession.create_answer(0)
  197. def my_config_handler_err(self, new_config):
  198. return isc.config.ccsession.create_answer(1, "just an error")
  199. def my_config_handler_exc(self, new_config):
  200. raise Exception("just an exception")
  201. def my_command_handler_ok(self, command, args):
  202. return isc.config.ccsession.create_answer(0)
  203. def my_command_handler_no_answer(self, command, args):
  204. pass
  205. def test_check_command1(self):
  206. fake_session = FakeModuleCCSession()
  207. mccs = self.create_session("spec1.spec", None, None, fake_session)
  208. mccs.check_command()
  209. self.assertEqual(len(fake_session.message_queue), 0)
  210. fake_session.group_sendmsg({'result': [ 0 ]}, "Spec1")
  211. mccs.check_command()
  212. self.assertEqual(len(fake_session.message_queue), 0)
  213. cmd = isc.config.ccsession.create_command(isc.config.ccsession.COMMAND_CONFIG_UPDATE, { 'Spec1': 'a' })
  214. fake_session.group_sendmsg(cmd, 'Spec1')
  215. mccs.check_command()
  216. self.assertEqual(len(fake_session.message_queue), 1)
  217. self.assertEqual({'result': [2, 'Spec1 has no config handler']},
  218. fake_session.get_message('Spec1', None))
  219. def test_check_command2(self):
  220. fake_session = FakeModuleCCSession()
  221. mccs = self.create_session("spec1.spec", None, None, fake_session)
  222. mccs.set_config_handler(self.my_config_handler_ok)
  223. self.assertEqual(len(fake_session.message_queue), 0)
  224. cmd = isc.config.ccsession.create_command(isc.config.ccsession.COMMAND_CONFIG_UPDATE, { 'Spec1': 'a' })
  225. fake_session.group_sendmsg(cmd, 'Spec1')
  226. self.assertEqual(len(fake_session.message_queue), 1)
  227. mccs.check_command()
  228. self.assertEqual(len(fake_session.message_queue), 1)
  229. self.assertEqual({'result': [1, 'No config_data specification']},
  230. fake_session.get_message('Spec1', None))
  231. def test_check_command3(self):
  232. fake_session = FakeModuleCCSession()
  233. mccs = self.create_session("spec2.spec", None, None, fake_session)
  234. mccs.set_config_handler(self.my_config_handler_ok)
  235. self.assertEqual(len(fake_session.message_queue), 0)
  236. cmd = isc.config.ccsession.create_command(isc.config.ccsession.COMMAND_CONFIG_UPDATE, { 'Spec2': { 'item1': 2 }})
  237. fake_session.group_sendmsg(cmd, 'Spec2')
  238. self.assertEqual(len(fake_session.message_queue), 1)
  239. mccs.check_command()
  240. self.assertEqual(len(fake_session.message_queue), 1)
  241. self.assertEqual({'result': [0]},
  242. fake_session.get_message('Spec2', None))
  243. def test_check_command4(self):
  244. fake_session = FakeModuleCCSession()
  245. mccs = self.create_session("spec2.spec", None, None, fake_session)
  246. mccs.set_config_handler(self.my_config_handler_err)
  247. self.assertEqual(len(fake_session.message_queue), 0)
  248. cmd = isc.config.ccsession.create_command(isc.config.ccsession.COMMAND_CONFIG_UPDATE, { 'Spec2': { 'item1': 'aaa' }})
  249. fake_session.group_sendmsg(cmd, 'Spec2')
  250. self.assertEqual(len(fake_session.message_queue), 1)
  251. mccs.check_command()
  252. self.assertEqual(len(fake_session.message_queue), 1)
  253. self.assertEqual({'result': [1, 'just an error']},
  254. fake_session.get_message('Spec2', None))
  255. def test_check_command5(self):
  256. fake_session = FakeModuleCCSession()
  257. mccs = self.create_session("spec2.spec", None, None, fake_session)
  258. mccs.set_config_handler(self.my_config_handler_exc)
  259. self.assertEqual(len(fake_session.message_queue), 0)
  260. cmd = isc.config.ccsession.create_command(isc.config.ccsession.COMMAND_CONFIG_UPDATE, { 'Spec2': { 'item1': 'aaa' }})
  261. fake_session.group_sendmsg(cmd, 'Spec2')
  262. self.assertEqual(len(fake_session.message_queue), 1)
  263. mccs.check_command()
  264. self.assertEqual(len(fake_session.message_queue), 1)
  265. self.assertEqual({'result': [1, 'just an exception']},
  266. fake_session.get_message('Spec2', None))
  267. def test_check_command6(self):
  268. fake_session = FakeModuleCCSession()
  269. mccs = self.create_session("spec2.spec", None, None, fake_session)
  270. self.assertEqual(len(fake_session.message_queue), 0)
  271. cmd = isc.config.ccsession.create_command("print_message", "just a message")
  272. fake_session.group_sendmsg(cmd, 'Spec2')
  273. self.assertEqual(len(fake_session.message_queue), 1)
  274. mccs.check_command()
  275. self.assertEqual(len(fake_session.message_queue), 1)
  276. self.assertEqual({'result': [2, 'Spec2 has no command handler']},
  277. fake_session.get_message('Spec2', None))
  278. def test_check_command7(self):
  279. fake_session = FakeModuleCCSession()
  280. mccs = self.create_session("spec2.spec", None, None, fake_session)
  281. mccs.set_command_handler(self.my_command_handler_ok)
  282. self.assertEqual(len(fake_session.message_queue), 0)
  283. cmd = isc.config.ccsession.create_command("print_message", "just a message")
  284. fake_session.group_sendmsg(cmd, 'Spec2')
  285. self.assertEqual(len(fake_session.message_queue), 1)
  286. mccs.check_command()
  287. self.assertEqual(len(fake_session.message_queue), 1)
  288. self.assertEqual({'result': [0]},
  289. fake_session.get_message('Spec2', None))
  290. def test_check_command8(self):
  291. fake_session = FakeModuleCCSession()
  292. mccs = self.create_session("spec2.spec", None, None, fake_session)
  293. mccs.set_command_handler(self.my_command_handler_no_answer)
  294. self.assertEqual(len(fake_session.message_queue), 0)
  295. cmd = isc.config.ccsession.create_command("print_message", "just a message")
  296. fake_session.group_sendmsg(cmd, 'Spec2')
  297. self.assertEqual(len(fake_session.message_queue), 1)
  298. mccs.check_command()
  299. self.assertEqual(len(fake_session.message_queue), 0)
  300. def test_check_command_without_recvmsg1(self):
  301. "copied from test_check_command2"
  302. fake_session = FakeModuleCCSession()
  303. mccs = self.create_session("spec1.spec", None, None, fake_session)
  304. mccs.set_config_handler(self.my_config_handler_ok)
  305. self.assertEqual(len(fake_session.message_queue), 0)
  306. cmd = isc.config.ccsession.create_command(isc.config.ccsession.COMMAND_CONFIG_UPDATE, { 'Spec1': 'abcd' })
  307. env = { 'group': 'Spec1', 'from':None };
  308. mccs.check_command_without_recvmsg(cmd, env)
  309. self.assertEqual(len(fake_session.message_queue), 1)
  310. self.assertEqual({'result': [1, 'No config_data specification']},
  311. fake_session.get_message('Spec1', None))
  312. def test_check_command_without_recvmsg2(self):
  313. "copied from test_check_command3"
  314. fake_session = FakeModuleCCSession()
  315. mccs = self.create_session("spec2.spec", None, None, fake_session)
  316. mccs.set_config_handler(self.my_config_handler_ok)
  317. self.assertEqual(len(fake_session.message_queue), 0)
  318. cmd = isc.config.ccsession.create_command(isc.config.ccsession.COMMAND_CONFIG_UPDATE, { 'Spec2': { 'item1': 2 }})
  319. self.assertEqual(len(fake_session.message_queue), 0)
  320. env = { 'group':'Spec2', 'from':None }
  321. mccs.check_command_without_recvmsg(cmd, env)
  322. self.assertEqual(len(fake_session.message_queue), 1)
  323. self.assertEqual({'result': [0]},
  324. fake_session.get_message('Spec2', None))
  325. def test_check_command_without_recvmsg3(self):
  326. "copied from test_check_command7"
  327. fake_session = FakeModuleCCSession()
  328. mccs = self.create_session("spec2.spec", None, None, fake_session)
  329. mccs.set_command_handler(self.my_command_handler_ok)
  330. self.assertEqual(len(fake_session.message_queue), 0)
  331. cmd = isc.config.ccsession.create_command("print_message", "just a message")
  332. env = { 'group':'Spec2', 'from':None }
  333. self.assertEqual(len(fake_session.message_queue), 0)
  334. mccs.check_command_without_recvmsg(cmd, env)
  335. self.assertEqual(len(fake_session.message_queue), 1)
  336. self.assertEqual({'result': [0]},
  337. fake_session.get_message('Spec2', None))
  338. def test_remote_module(self):
  339. fake_session = FakeModuleCCSession()
  340. mccs = self.create_session("spec1.spec", None, None, fake_session)
  341. mccs.remove_remote_config("Spec2")
  342. self.assertRaises(ModuleCCSessionError, mccs.get_remote_config_value, "Spec2", "item1")
  343. self.assertFalse("Spec2" in fake_session.subscriptions)
  344. rmodname = mccs.add_remote_config(self.spec_file("spec2.spec"))
  345. self.assertTrue("Spec2" in fake_session.subscriptions)
  346. self.assertEqual("Spec2", rmodname)
  347. self.assertRaises(isc.cc.data.DataNotFoundError, mccs.get_remote_config_value, rmodname, "asdf")
  348. value, default = mccs.get_remote_config_value(rmodname, "item1")
  349. self.assertEqual(1, value)
  350. self.assertEqual(True, default)
  351. mccs.remove_remote_config(rmodname)
  352. self.assertFalse("Spec2" in fake_session.subscriptions)
  353. self.assertRaises(ModuleCCSessionError, mccs.get_remote_config_value, "Spec2", "item1")
  354. # test if unsubscription is alse sent when object is deleted
  355. rmodname = mccs.add_remote_config(self.spec_file("spec2.spec"))
  356. self.assertTrue("Spec2" in fake_session.subscriptions)
  357. mccs = None
  358. self.assertFalse("Spec2" in fake_session.subscriptions)
  359. def test_ignore_command_remote_module(self):
  360. # Create a Spec1 module and subscribe to remote config for Spec2
  361. fake_session = FakeModuleCCSession()
  362. mccs = self.create_session("spec1.spec", None, None, fake_session)
  363. mccs.set_command_handler(self.my_command_handler_ok)
  364. rmodname = mccs.add_remote_config(self.spec_file("spec2.spec"))
  365. # remove the 'get config' from the queue
  366. self.assertEqual(len(fake_session.message_queue), 1)
  367. fake_session.get_message("ConfigManager")
  368. # check if the command for the module itself is received
  369. cmd = isc.config.ccsession.create_command("just_some_command", { 'foo': 'a' })
  370. fake_session.group_sendmsg(cmd, 'Spec1')
  371. self.assertEqual(len(fake_session.message_queue), 1)
  372. mccs.check_command()
  373. self.assertEqual(len(fake_session.message_queue), 1)
  374. self.assertEqual({'result': [ 0 ]},
  375. fake_session.get_message('Spec1', None))
  376. # check if the command for the other module is ignored
  377. cmd = isc.config.ccsession.create_command("just_some_command", { 'foo': 'a' })
  378. fake_session.group_sendmsg(cmd, 'Spec2')
  379. self.assertEqual(len(fake_session.message_queue), 1)
  380. mccs.check_command()
  381. self.assertEqual(len(fake_session.message_queue), 0)
  382. class fakeUIConn():
  383. def __init__(self):
  384. self.get_answers = {}
  385. self.post_answers = {}
  386. def set_get_answer(self, name, answer):
  387. self.get_answers[name] = answer
  388. def set_post_answer(self, name, answer):
  389. self.post_answers[name] = answer
  390. def send_GET(self, name, arg = None):
  391. if name in self.get_answers:
  392. return self.get_answers[name]
  393. else:
  394. return {}
  395. def send_POST(self, name, arg = None):
  396. if name in self.post_answers:
  397. return self.post_answers[name]
  398. else:
  399. return None
  400. class TestUIModuleCCSession(unittest.TestCase):
  401. def setUp(self):
  402. if 'CONFIG_TESTDATA_PATH' in os.environ:
  403. self.data_path = os.environ['CONFIG_TESTDATA_PATH']
  404. else:
  405. self.data_path = "../../../testdata"
  406. def spec_file(self, file):
  407. return self.data_path + os.sep + file
  408. def create_uccs2(self, fake_conn):
  409. module_spec = isc.config.module_spec_from_file(self.spec_file("spec2.spec"))
  410. fake_conn.set_get_answer('/module_spec', { module_spec.get_module_name(): module_spec.get_full_spec()})
  411. fake_conn.set_get_answer('/config_data', { 'version': BIND10_CONFIG_DATA_VERSION })
  412. return UIModuleCCSession(fake_conn)
  413. def test_init(self):
  414. fake_conn = fakeUIConn()
  415. fake_conn.set_get_answer('/module_spec', {})
  416. fake_conn.set_get_answer('/config_data', { 'version': BIND10_CONFIG_DATA_VERSION })
  417. uccs = UIModuleCCSession(fake_conn)
  418. self.assertEqual({}, uccs._specifications)
  419. self.assertEqual({ 'version': BIND10_CONFIG_DATA_VERSION}, uccs._current_config)
  420. module_spec = isc.config.module_spec_from_file(self.spec_file("spec2.spec"))
  421. fake_conn.set_get_answer('/module_spec', { module_spec.get_module_name(): module_spec.get_full_spec()})
  422. fake_conn.set_get_answer('/config_data', { 'version': BIND10_CONFIG_DATA_VERSION })
  423. uccs = UIModuleCCSession(fake_conn)
  424. self.assertEqual(module_spec._module_spec, uccs._specifications['Spec2']._module_spec)
  425. fake_conn.set_get_answer('/config_data', { 'version': 123123 })
  426. self.assertRaises(ModuleCCSessionError, UIModuleCCSession, fake_conn)
  427. def test_add_remove_value(self):
  428. fake_conn = fakeUIConn()
  429. uccs = self.create_uccs2(fake_conn)
  430. self.assertRaises(isc.cc.data.DataNotFoundError, uccs.add_value, 1, "a")
  431. self.assertRaises(isc.cc.data.DataNotFoundError, uccs.add_value, "no_such_item", "a")
  432. self.assertRaises(isc.cc.data.DataNotFoundError, uccs.add_value, "Spec2/item1", "a")
  433. self.assertRaises(isc.cc.data.DataNotFoundError, uccs.remove_value, 1, "a")
  434. self.assertRaises(isc.cc.data.DataNotFoundError, uccs.remove_value, "no_such_item", "a")
  435. self.assertRaises(isc.cc.data.DataNotFoundError, uccs.remove_value, "Spec2/item1", "a")
  436. self.assertEqual({}, uccs._local_changes)
  437. uccs.add_value("Spec2/item5", "foo")
  438. self.assertEqual({'Spec2': {'item5': ['a', 'b', 'foo']}}, uccs._local_changes)
  439. uccs.remove_value("Spec2/item5", "foo")
  440. self.assertEqual({'Spec2': {'item5': ['a', 'b']}}, uccs._local_changes)
  441. uccs._local_changes = {'Spec2': {'item5': []}}
  442. uccs.remove_value("Spec2/item5", "foo")
  443. uccs.add_value("Spec2/item5", "foo")
  444. self.assertEqual({'Spec2': {'item5': ['foo']}}, uccs._local_changes)
  445. uccs.add_value("Spec2/item5", "foo")
  446. self.assertEqual({'Spec2': {'item5': ['foo']}}, uccs._local_changes)
  447. def test_commit(self):
  448. fake_conn = fakeUIConn()
  449. uccs = self.create_uccs2(fake_conn)
  450. uccs.commit()
  451. uccs._local_changes = {'Spec2': {'item5': [ 'a' ]}}
  452. uccs.commit()
  453. if __name__ == '__main__':
  454. unittest.main()