event.py 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975
  1. # ex:ts=4:sw=4:sts=4:et
  2. # -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
  3. #
  4. # BitBake Tests for the Event implementation (event.py)
  5. #
  6. # Copyright (C) 2017 Intel Corporation
  7. #
  8. # SPDX-License-Identifier: GPL-2.0-only
  9. #
  10. import unittest
  11. import bb
  12. import logging
  13. import bb.compat
  14. import bb.event
  15. import importlib
  16. import threading
  17. import time
  18. import pickle
  19. from unittest.mock import Mock
  20. from unittest.mock import call
  21. from bb.msg import BBLogFormatter
  22. class EventQueueStubBase(object):
  23. """ Base class for EventQueueStub classes """
  24. def __init__(self):
  25. self.event_calls = []
  26. return
  27. def _store_event_data_string(self, event):
  28. if isinstance(event, logging.LogRecord):
  29. formatter = BBLogFormatter("%(levelname)s: %(message)s")
  30. self.event_calls.append(formatter.format(event))
  31. else:
  32. self.event_calls.append(bb.event.getName(event))
  33. return
  34. class EventQueueStub(EventQueueStubBase):
  35. """ Class used as specification for UI event handler queue stub objects """
  36. def __init__(self):
  37. super(EventQueueStub, self).__init__()
  38. def send(self, event):
  39. super(EventQueueStub, self)._store_event_data_string(event)
  40. class PickleEventQueueStub(EventQueueStubBase):
  41. """ Class used as specification for UI event handler queue stub objects
  42. with sendpickle method """
  43. def __init__(self):
  44. super(PickleEventQueueStub, self).__init__()
  45. def sendpickle(self, pickled_event):
  46. event = pickle.loads(pickled_event)
  47. super(PickleEventQueueStub, self)._store_event_data_string(event)
  48. class UIClientStub(object):
  49. """ Class used as specification for UI event handler stub objects """
  50. def __init__(self):
  51. self.event = None
  52. class EventHandlingTest(unittest.TestCase):
  53. """ Event handling test class """
  54. def setUp(self):
  55. self._test_process = Mock()
  56. ui_client1 = UIClientStub()
  57. ui_client2 = UIClientStub()
  58. self._test_ui1 = Mock(wraps=ui_client1)
  59. self._test_ui2 = Mock(wraps=ui_client2)
  60. importlib.reload(bb.event)
  61. def _create_test_handlers(self):
  62. """ Method used to create a test handler ordered dictionary """
  63. test_handlers = bb.compat.OrderedDict()
  64. test_handlers["handler1"] = self._test_process.handler1
  65. test_handlers["handler2"] = self._test_process.handler2
  66. return test_handlers
  67. def test_class_handlers(self):
  68. """ Test set_class_handlers and get_class_handlers methods """
  69. test_handlers = self._create_test_handlers()
  70. bb.event.set_class_handlers(test_handlers)
  71. self.assertEqual(test_handlers,
  72. bb.event.get_class_handlers())
  73. def test_handlers(self):
  74. """ Test set_handlers and get_handlers """
  75. test_handlers = self._create_test_handlers()
  76. bb.event.set_handlers(test_handlers)
  77. self.assertEqual(test_handlers,
  78. bb.event.get_handlers())
  79. def test_clean_class_handlers(self):
  80. """ Test clean_class_handlers method """
  81. cleanDict = bb.compat.OrderedDict()
  82. self.assertEqual(cleanDict,
  83. bb.event.clean_class_handlers())
  84. def test_register(self):
  85. """ Test register method for class handlers """
  86. result = bb.event.register("handler", self._test_process.handler)
  87. self.assertEqual(result, bb.event.Registered)
  88. handlers_dict = bb.event.get_class_handlers()
  89. self.assertIn("handler", handlers_dict)
  90. def test_already_registered(self):
  91. """ Test detection of an already registed class handler """
  92. bb.event.register("handler", self._test_process.handler)
  93. handlers_dict = bb.event.get_class_handlers()
  94. self.assertIn("handler", handlers_dict)
  95. result = bb.event.register("handler", self._test_process.handler)
  96. self.assertEqual(result, bb.event.AlreadyRegistered)
  97. def test_register_from_string(self):
  98. """ Test register method receiving code in string """
  99. result = bb.event.register("string_handler", " return True")
  100. self.assertEqual(result, bb.event.Registered)
  101. handlers_dict = bb.event.get_class_handlers()
  102. self.assertIn("string_handler", handlers_dict)
  103. def test_register_with_mask(self):
  104. """ Test register method with event masking """
  105. mask = ["bb.event.OperationStarted",
  106. "bb.event.OperationCompleted"]
  107. result = bb.event.register("event_handler",
  108. self._test_process.event_handler,
  109. mask)
  110. self.assertEqual(result, bb.event.Registered)
  111. handlers_dict = bb.event.get_class_handlers()
  112. self.assertIn("event_handler", handlers_dict)
  113. def test_remove(self):
  114. """ Test remove method for class handlers """
  115. test_handlers = self._create_test_handlers()
  116. bb.event.set_class_handlers(test_handlers)
  117. count = len(test_handlers)
  118. bb.event.remove("handler1", None)
  119. test_handlers = bb.event.get_class_handlers()
  120. self.assertEqual(len(test_handlers), count - 1)
  121. with self.assertRaises(KeyError):
  122. bb.event.remove("handler1", None)
  123. def test_execute_handler(self):
  124. """ Test execute_handler method for class handlers """
  125. mask = ["bb.event.OperationProgress"]
  126. result = bb.event.register("event_handler",
  127. self._test_process.event_handler,
  128. mask)
  129. self.assertEqual(result, bb.event.Registered)
  130. event = bb.event.OperationProgress(current=10, total=100)
  131. bb.event.execute_handler("event_handler",
  132. self._test_process.event_handler,
  133. event,
  134. None)
  135. self._test_process.event_handler.assert_called_once_with(event)
  136. def test_fire_class_handlers(self):
  137. """ Test fire_class_handlers method """
  138. mask = ["bb.event.OperationStarted"]
  139. result = bb.event.register("event_handler1",
  140. self._test_process.event_handler1,
  141. mask)
  142. self.assertEqual(result, bb.event.Registered)
  143. result = bb.event.register("event_handler2",
  144. self._test_process.event_handler2,
  145. "*")
  146. self.assertEqual(result, bb.event.Registered)
  147. event1 = bb.event.OperationStarted()
  148. event2 = bb.event.OperationCompleted(total=123)
  149. bb.event.fire_class_handlers(event1, None)
  150. bb.event.fire_class_handlers(event2, None)
  151. bb.event.fire_class_handlers(event2, None)
  152. expected_event_handler1 = [call(event1)]
  153. expected_event_handler2 = [call(event1),
  154. call(event2),
  155. call(event2)]
  156. self.assertEqual(self._test_process.event_handler1.call_args_list,
  157. expected_event_handler1)
  158. self.assertEqual(self._test_process.event_handler2.call_args_list,
  159. expected_event_handler2)
  160. def test_class_handler_filters(self):
  161. """ Test filters for class handlers """
  162. mask = ["bb.event.OperationStarted"]
  163. result = bb.event.register("event_handler1",
  164. self._test_process.event_handler1,
  165. mask)
  166. self.assertEqual(result, bb.event.Registered)
  167. result = bb.event.register("event_handler2",
  168. self._test_process.event_handler2,
  169. "*")
  170. self.assertEqual(result, bb.event.Registered)
  171. bb.event.set_eventfilter(
  172. lambda name, handler, event, d :
  173. name == 'event_handler2' and
  174. bb.event.getName(event) == "OperationStarted")
  175. event1 = bb.event.OperationStarted()
  176. event2 = bb.event.OperationCompleted(total=123)
  177. bb.event.fire_class_handlers(event1, None)
  178. bb.event.fire_class_handlers(event2, None)
  179. bb.event.fire_class_handlers(event2, None)
  180. expected_event_handler1 = []
  181. expected_event_handler2 = [call(event1)]
  182. self.assertEqual(self._test_process.event_handler1.call_args_list,
  183. expected_event_handler1)
  184. self.assertEqual(self._test_process.event_handler2.call_args_list,
  185. expected_event_handler2)
  186. def test_change_handler_event_mapping(self):
  187. """ Test changing the event mapping for class handlers """
  188. event1 = bb.event.OperationStarted()
  189. event2 = bb.event.OperationCompleted(total=123)
  190. # register handler for all events
  191. result = bb.event.register("event_handler1",
  192. self._test_process.event_handler1,
  193. "*")
  194. self.assertEqual(result, bb.event.Registered)
  195. bb.event.fire_class_handlers(event1, None)
  196. bb.event.fire_class_handlers(event2, None)
  197. expected = [call(event1), call(event2)]
  198. self.assertEqual(self._test_process.event_handler1.call_args_list,
  199. expected)
  200. # unregister handler and register it only for OperationStarted
  201. bb.event.remove("event_handler1",
  202. self._test_process.event_handler1)
  203. mask = ["bb.event.OperationStarted"]
  204. result = bb.event.register("event_handler1",
  205. self._test_process.event_handler1,
  206. mask)
  207. self.assertEqual(result, bb.event.Registered)
  208. bb.event.fire_class_handlers(event1, None)
  209. bb.event.fire_class_handlers(event2, None)
  210. expected = [call(event1), call(event2), call(event1)]
  211. self.assertEqual(self._test_process.event_handler1.call_args_list,
  212. expected)
  213. # unregister handler and register it only for OperationCompleted
  214. bb.event.remove("event_handler1",
  215. self._test_process.event_handler1)
  216. mask = ["bb.event.OperationCompleted"]
  217. result = bb.event.register("event_handler1",
  218. self._test_process.event_handler1,
  219. mask)
  220. self.assertEqual(result, bb.event.Registered)
  221. bb.event.fire_class_handlers(event1, None)
  222. bb.event.fire_class_handlers(event2, None)
  223. expected = [call(event1), call(event2), call(event1), call(event2)]
  224. self.assertEqual(self._test_process.event_handler1.call_args_list,
  225. expected)
  226. def test_register_UIHhandler(self):
  227. """ Test register_UIHhandler method """
  228. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  229. self.assertEqual(result, 1)
  230. def test_UIHhandler_already_registered(self):
  231. """ Test registering an UIHhandler already existing """
  232. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  233. self.assertEqual(result, 1)
  234. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  235. self.assertEqual(result, 2)
  236. def test_unregister_UIHhandler(self):
  237. """ Test unregister_UIHhandler method """
  238. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  239. self.assertEqual(result, 1)
  240. result = bb.event.unregister_UIHhandler(1)
  241. self.assertIs(result, None)
  242. def test_fire_ui_handlers(self):
  243. """ Test fire_ui_handlers method """
  244. self._test_ui1.event = Mock(spec_set=EventQueueStub)
  245. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  246. self.assertEqual(result, 1)
  247. self._test_ui2.event = Mock(spec_set=PickleEventQueueStub)
  248. result = bb.event.register_UIHhandler(self._test_ui2, mainui=True)
  249. self.assertEqual(result, 2)
  250. event1 = bb.event.OperationStarted()
  251. bb.event.fire_ui_handlers(event1, None)
  252. expected = [call(event1)]
  253. self.assertEqual(self._test_ui1.event.send.call_args_list,
  254. expected)
  255. expected = [call(pickle.dumps(event1))]
  256. self.assertEqual(self._test_ui2.event.sendpickle.call_args_list,
  257. expected)
  258. def test_ui_handler_mask_filter(self):
  259. """ Test filters for UI handlers """
  260. mask = ["bb.event.OperationStarted"]
  261. debug_domains = {}
  262. self._test_ui1.event = Mock(spec_set=EventQueueStub)
  263. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  264. bb.event.set_UIHmask(result, logging.INFO, debug_domains, mask)
  265. self._test_ui2.event = Mock(spec_set=PickleEventQueueStub)
  266. result = bb.event.register_UIHhandler(self._test_ui2, mainui=True)
  267. bb.event.set_UIHmask(result, logging.INFO, debug_domains, mask)
  268. event1 = bb.event.OperationStarted()
  269. event2 = bb.event.OperationCompleted(total=1)
  270. bb.event.fire_ui_handlers(event1, None)
  271. bb.event.fire_ui_handlers(event2, None)
  272. expected = [call(event1)]
  273. self.assertEqual(self._test_ui1.event.send.call_args_list,
  274. expected)
  275. expected = [call(pickle.dumps(event1))]
  276. self.assertEqual(self._test_ui2.event.sendpickle.call_args_list,
  277. expected)
  278. def test_ui_handler_log_filter(self):
  279. """ Test log filters for UI handlers """
  280. mask = ["*"]
  281. debug_domains = {'BitBake.Foo': logging.WARNING}
  282. self._test_ui1.event = EventQueueStub()
  283. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  284. bb.event.set_UIHmask(result, logging.ERROR, debug_domains, mask)
  285. self._test_ui2.event = PickleEventQueueStub()
  286. result = bb.event.register_UIHhandler(self._test_ui2, mainui=True)
  287. bb.event.set_UIHmask(result, logging.ERROR, debug_domains, mask)
  288. event1 = bb.event.OperationStarted()
  289. bb.event.fire_ui_handlers(event1, None) # All events match
  290. event_log_handler = bb.event.LogHandler()
  291. logger = logging.getLogger("BitBake")
  292. logger.addHandler(event_log_handler)
  293. logger1 = logging.getLogger("BitBake.Foo")
  294. logger1.warning("Test warning LogRecord1") # Matches debug_domains level
  295. logger1.info("Test info LogRecord") # Filtered out
  296. logger2 = logging.getLogger("BitBake.Bar")
  297. logger2.error("Test error LogRecord") # Matches filter base level
  298. logger2.warning("Test warning LogRecord2") # Filtered out
  299. logger.removeHandler(event_log_handler)
  300. expected = ['OperationStarted',
  301. 'WARNING: Test warning LogRecord1',
  302. 'ERROR: Test error LogRecord']
  303. self.assertEqual(self._test_ui1.event.event_calls, expected)
  304. self.assertEqual(self._test_ui2.event.event_calls, expected)
  305. def test_fire(self):
  306. """ Test fire method used to trigger class and ui event handlers """
  307. mask = ["bb.event.ConfigParsed"]
  308. result = bb.event.register("event_handler1",
  309. self._test_process.event_handler1,
  310. mask)
  311. self._test_ui1.event = Mock(spec_set=EventQueueStub)
  312. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  313. self.assertEqual(result, 1)
  314. event1 = bb.event.ConfigParsed()
  315. bb.event.fire(event1, None)
  316. expected = [call(event1)]
  317. self.assertEqual(self._test_process.event_handler1.call_args_list,
  318. expected)
  319. self.assertEqual(self._test_ui1.event.send.call_args_list,
  320. expected)
  321. def test_fire_from_worker(self):
  322. """ Test fire_from_worker method """
  323. self._test_ui1.event = Mock(spec_set=EventQueueStub)
  324. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  325. self.assertEqual(result, 1)
  326. event1 = bb.event.ConfigParsed()
  327. bb.event.fire_from_worker(event1, None)
  328. expected = [call(event1)]
  329. self.assertEqual(self._test_ui1.event.send.call_args_list,
  330. expected)
  331. def test_worker_fire(self):
  332. """ Test the triggering of bb.event.worker_fire callback """
  333. bb.event.worker_fire = Mock()
  334. event = bb.event.Event()
  335. bb.event.fire(event, None)
  336. expected = [call(event, None)]
  337. self.assertEqual(bb.event.worker_fire.call_args_list, expected)
  338. def test_print_ui_queue(self):
  339. """ Test print_ui_queue method """
  340. event1 = bb.event.OperationStarted()
  341. event2 = bb.event.OperationCompleted(total=123)
  342. bb.event.fire(event1, None)
  343. bb.event.fire(event2, None)
  344. event_log_handler = bb.event.LogHandler()
  345. logger = logging.getLogger("BitBake")
  346. logger.addHandler(event_log_handler)
  347. logger.info("Test info LogRecord")
  348. logger.warning("Test warning LogRecord")
  349. with self.assertLogs("BitBake", level="INFO") as cm:
  350. bb.event.print_ui_queue()
  351. logger.removeHandler(event_log_handler)
  352. self.assertEqual(cm.output,
  353. ["INFO:BitBake:Test info LogRecord",
  354. "WARNING:BitBake:Test warning LogRecord"])
  355. def _set_threadlock_test_mockups(self):
  356. """ Create UI event handler mockups used in enable and disable
  357. threadlock tests """
  358. def ui1_event_send(event):
  359. if type(event) is bb.event.ConfigParsed:
  360. self._threadlock_test_calls.append("w1_ui1")
  361. if type(event) is bb.event.OperationStarted:
  362. self._threadlock_test_calls.append("w2_ui1")
  363. time.sleep(2)
  364. def ui2_event_send(event):
  365. if type(event) is bb.event.ConfigParsed:
  366. self._threadlock_test_calls.append("w1_ui2")
  367. if type(event) is bb.event.OperationStarted:
  368. self._threadlock_test_calls.append("w2_ui2")
  369. time.sleep(2)
  370. self._threadlock_test_calls = []
  371. self._test_ui1.event = EventQueueStub()
  372. self._test_ui1.event.send = ui1_event_send
  373. result = bb.event.register_UIHhandler(self._test_ui1, mainui=True)
  374. self.assertEqual(result, 1)
  375. self._test_ui2.event = EventQueueStub()
  376. self._test_ui2.event.send = ui2_event_send
  377. result = bb.event.register_UIHhandler(self._test_ui2, mainui=True)
  378. self.assertEqual(result, 2)
  379. def _set_and_run_threadlock_test_workers(self):
  380. """ Create and run the workers used to trigger events in enable and
  381. disable threadlock tests """
  382. worker1 = threading.Thread(target=self._thread_lock_test_worker1)
  383. worker2 = threading.Thread(target=self._thread_lock_test_worker2)
  384. worker1.start()
  385. time.sleep(1)
  386. worker2.start()
  387. worker1.join()
  388. worker2.join()
  389. def _thread_lock_test_worker1(self):
  390. """ First worker used to fire the ConfigParsed event for enable and
  391. disable threadlocks tests """
  392. bb.event.fire(bb.event.ConfigParsed(), None)
  393. def _thread_lock_test_worker2(self):
  394. """ Second worker used to fire the OperationStarted event for enable
  395. and disable threadlocks tests """
  396. bb.event.fire(bb.event.OperationStarted(), None)
  397. def test_enable_threadlock(self):
  398. """ Test enable_threadlock method """
  399. self._set_threadlock_test_mockups()
  400. bb.event.enable_threadlock()
  401. self._set_and_run_threadlock_test_workers()
  402. # Calls to UI handlers should be in order as all the registered
  403. # handlers for the event coming from the first worker should be
  404. # called before processing the event from the second worker.
  405. self.assertEqual(self._threadlock_test_calls,
  406. ["w1_ui1", "w1_ui2", "w2_ui1", "w2_ui2"])
  407. def test_disable_threadlock(self):
  408. """ Test disable_threadlock method """
  409. self._set_threadlock_test_mockups()
  410. bb.event.disable_threadlock()
  411. self._set_and_run_threadlock_test_workers()
  412. # Calls to UI handlers should be intertwined together. Thanks to the
  413. # delay in the registered handlers for the event coming from the first
  414. # worker, the event coming from the second worker starts being
  415. # processed before finishing handling the first worker event.
  416. self.assertEqual(self._threadlock_test_calls,
  417. ["w1_ui1", "w2_ui1", "w1_ui2", "w2_ui2"])
  418. class EventClassesTest(unittest.TestCase):
  419. """ Event classes test class """
  420. _worker_pid = 54321
  421. def setUp(self):
  422. bb.event.worker_pid = EventClassesTest._worker_pid
  423. def test_Event(self):
  424. """ Test the Event base class """
  425. event = bb.event.Event()
  426. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  427. def test_HeartbeatEvent(self):
  428. """ Test the HeartbeatEvent class """
  429. time = 10
  430. event = bb.event.HeartbeatEvent(time)
  431. self.assertEqual(event.time, time)
  432. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  433. def test_OperationStarted(self):
  434. """ Test OperationStarted event class """
  435. msg = "Foo Bar"
  436. event = bb.event.OperationStarted(msg)
  437. self.assertEqual(event.msg, msg)
  438. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  439. def test_OperationCompleted(self):
  440. """ Test OperationCompleted event class """
  441. msg = "Foo Bar"
  442. total = 123
  443. event = bb.event.OperationCompleted(total, msg)
  444. self.assertEqual(event.msg, msg)
  445. self.assertEqual(event.total, total)
  446. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  447. def test_OperationProgress(self):
  448. """ Test OperationProgress event class """
  449. msg = "Foo Bar"
  450. total = 123
  451. current = 111
  452. event = bb.event.OperationProgress(current, total, msg)
  453. self.assertEqual(event.msg, msg + ": %s/%s" % (current, total))
  454. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  455. def test_ConfigParsed(self):
  456. """ Test the ConfigParsed class """
  457. event = bb.event.ConfigParsed()
  458. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  459. def test_MultiConfigParsed(self):
  460. """ Test MultiConfigParsed event class """
  461. mcdata = {"foobar": "Foo Bar"}
  462. event = bb.event.MultiConfigParsed(mcdata)
  463. self.assertEqual(event.mcdata, mcdata)
  464. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  465. def test_RecipeEvent(self):
  466. """ Test RecipeEvent event base class """
  467. callback = lambda a: 2 * a
  468. event = bb.event.RecipeEvent(callback)
  469. self.assertEqual(event.fn(1), callback(1))
  470. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  471. def test_RecipePreFinalise(self):
  472. """ Test RecipePreFinalise event class """
  473. callback = lambda a: 2 * a
  474. event = bb.event.RecipePreFinalise(callback)
  475. self.assertEqual(event.fn(1), callback(1))
  476. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  477. def test_RecipeTaskPreProcess(self):
  478. """ Test RecipeTaskPreProcess event class """
  479. callback = lambda a: 2 * a
  480. tasklist = [("foobar", callback)]
  481. event = bb.event.RecipeTaskPreProcess(callback, tasklist)
  482. self.assertEqual(event.fn(1), callback(1))
  483. self.assertEqual(event.tasklist, tasklist)
  484. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  485. def test_RecipeParsed(self):
  486. """ Test RecipeParsed event base class """
  487. callback = lambda a: 2 * a
  488. event = bb.event.RecipeParsed(callback)
  489. self.assertEqual(event.fn(1), callback(1))
  490. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  491. def test_StampUpdate(self):
  492. targets = ["foo", "bar"]
  493. stampfns = [lambda:"foobar"]
  494. event = bb.event.StampUpdate(targets, stampfns)
  495. self.assertEqual(event.targets, targets)
  496. self.assertEqual(event.stampPrefix, stampfns)
  497. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  498. def test_BuildBase(self):
  499. """ Test base class for bitbake build events """
  500. name = "foo"
  501. pkgs = ["bar"]
  502. failures = 123
  503. event = bb.event.BuildBase(name, pkgs, failures)
  504. self.assertEqual(event.name, name)
  505. self.assertEqual(event.pkgs, pkgs)
  506. self.assertEqual(event.getFailures(), failures)
  507. name = event.name = "bar"
  508. pkgs = event.pkgs = ["foo"]
  509. self.assertEqual(event.name, name)
  510. self.assertEqual(event.pkgs, pkgs)
  511. self.assertEqual(event.getFailures(), failures)
  512. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  513. def test_BuildInit(self):
  514. """ Test class for bitbake build invocation events """
  515. event = bb.event.BuildInit()
  516. self.assertEqual(event.name, None)
  517. self.assertEqual(event.pkgs, [])
  518. self.assertEqual(event.getFailures(), 0)
  519. name = event.name = "bar"
  520. pkgs = event.pkgs = ["foo"]
  521. self.assertEqual(event.name, name)
  522. self.assertEqual(event.pkgs, pkgs)
  523. self.assertEqual(event.getFailures(), 0)
  524. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  525. def test_BuildStarted(self):
  526. """ Test class for build started events """
  527. name = "foo"
  528. pkgs = ["bar"]
  529. failures = 123
  530. event = bb.event.BuildStarted(name, pkgs, failures)
  531. self.assertEqual(event.name, name)
  532. self.assertEqual(event.pkgs, pkgs)
  533. self.assertEqual(event.getFailures(), failures)
  534. self.assertEqual(event.msg, "Building Started")
  535. name = event.name = "bar"
  536. pkgs = event.pkgs = ["foo"]
  537. msg = event.msg = "foobar"
  538. self.assertEqual(event.name, name)
  539. self.assertEqual(event.pkgs, pkgs)
  540. self.assertEqual(event.getFailures(), failures)
  541. self.assertEqual(event.msg, msg)
  542. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  543. def test_BuildCompleted(self):
  544. """ Test class for build completed events """
  545. total = 1000
  546. name = "foo"
  547. pkgs = ["bar"]
  548. failures = 123
  549. interrupted = 1
  550. event = bb.event.BuildCompleted(total, name, pkgs, failures,
  551. interrupted)
  552. self.assertEqual(event.name, name)
  553. self.assertEqual(event.pkgs, pkgs)
  554. self.assertEqual(event.getFailures(), failures)
  555. self.assertEqual(event.msg, "Building Failed")
  556. event2 = bb.event.BuildCompleted(total, name, pkgs)
  557. self.assertEqual(event2.name, name)
  558. self.assertEqual(event2.pkgs, pkgs)
  559. self.assertEqual(event2.getFailures(), 0)
  560. self.assertEqual(event2.msg, "Building Succeeded")
  561. self.assertEqual(event2.pid, EventClassesTest._worker_pid)
  562. def test_DiskFull(self):
  563. """ Test DiskFull event class """
  564. dev = "/dev/foo"
  565. type = "ext4"
  566. freespace = "104M"
  567. mountpoint = "/"
  568. event = bb.event.DiskFull(dev, type, freespace, mountpoint)
  569. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  570. def test_MonitorDiskEvent(self):
  571. """ Test MonitorDiskEvent class """
  572. available_bytes = 10000000
  573. free_bytes = 90000000
  574. total_bytes = 1000000000
  575. du = bb.event.DiskUsageSample(available_bytes, free_bytes,
  576. total_bytes)
  577. event = bb.event.MonitorDiskEvent(du)
  578. self.assertEqual(event.disk_usage.available_bytes, available_bytes)
  579. self.assertEqual(event.disk_usage.free_bytes, free_bytes)
  580. self.assertEqual(event.disk_usage.total_bytes, total_bytes)
  581. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  582. def test_NoProvider(self):
  583. """ Test NoProvider event class """
  584. item = "foobar"
  585. event1 = bb.event.NoProvider(item)
  586. self.assertEqual(event1.getItem(), item)
  587. self.assertEqual(event1.isRuntime(), False)
  588. self.assertEqual(str(event1), "Nothing PROVIDES 'foobar'")
  589. runtime = True
  590. dependees = ["foo", "bar"]
  591. reasons = None
  592. close_matches = ["foibar", "footbar"]
  593. event2 = bb.event.NoProvider(item, runtime, dependees, reasons,
  594. close_matches)
  595. self.assertEqual(event2.isRuntime(), True)
  596. expected = ("Nothing RPROVIDES 'foobar' (but foo, bar RDEPENDS"
  597. " on or otherwise requires it). Close matches:\n"
  598. " foibar\n"
  599. " footbar")
  600. self.assertEqual(str(event2), expected)
  601. reasons = ["Item does not exist on database"]
  602. close_matches = ["foibar", "footbar"]
  603. event3 = bb.event.NoProvider(item, runtime, dependees, reasons,
  604. close_matches)
  605. expected = ("Nothing RPROVIDES 'foobar' (but foo, bar RDEPENDS"
  606. " on or otherwise requires it)\n"
  607. "Item does not exist on database")
  608. self.assertEqual(str(event3), expected)
  609. self.assertEqual(event3.pid, EventClassesTest._worker_pid)
  610. def test_MultipleProviders(self):
  611. """ Test MultipleProviders event class """
  612. item = "foobar"
  613. candidates = ["foobarv1", "foobars"]
  614. event1 = bb.event.MultipleProviders(item, candidates)
  615. self.assertEqual(event1.isRuntime(), False)
  616. self.assertEqual(event1.getItem(), item)
  617. self.assertEqual(event1.getCandidates(), candidates)
  618. expected = ("Multiple providers are available for foobar (foobarv1,"
  619. " foobars)\n"
  620. "Consider defining a PREFERRED_PROVIDER entry to match "
  621. "foobar")
  622. self.assertEqual(str(event1), expected)
  623. runtime = True
  624. event2 = bb.event.MultipleProviders(item, candidates, runtime)
  625. self.assertEqual(event2.isRuntime(), runtime)
  626. expected = ("Multiple providers are available for runtime foobar "
  627. "(foobarv1, foobars)\n"
  628. "Consider defining a PREFERRED_RPROVIDER entry to match "
  629. "foobar")
  630. self.assertEqual(str(event2), expected)
  631. self.assertEqual(event2.pid, EventClassesTest._worker_pid)
  632. def test_ParseStarted(self):
  633. """ Test ParseStarted event class """
  634. total = 123
  635. event = bb.event.ParseStarted(total)
  636. self.assertEqual(event.msg, "Recipe parsing Started")
  637. self.assertEqual(event.total, total)
  638. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  639. def test_ParseCompleted(self):
  640. """ Test ParseCompleted event class """
  641. cached = 10
  642. parsed = 13
  643. skipped = 7
  644. virtuals = 2
  645. masked = 1
  646. errors = 0
  647. total = 23
  648. event = bb.event.ParseCompleted(cached, parsed, skipped, masked,
  649. virtuals, errors, total)
  650. self.assertEqual(event.msg, "Recipe parsing Completed")
  651. expected = [cached, parsed, skipped, virtuals, masked, errors,
  652. cached + parsed, total]
  653. actual = [event.cached, event.parsed, event.skipped, event.virtuals,
  654. event.masked, event.errors, event.sofar, event.total]
  655. self.assertEqual(str(actual), str(expected))
  656. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  657. def test_ParseProgress(self):
  658. """ Test ParseProgress event class """
  659. current = 10
  660. total = 100
  661. event = bb.event.ParseProgress(current, total)
  662. self.assertEqual(event.msg,
  663. "Recipe parsing" + ": %s/%s" % (current, total))
  664. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  665. def test_CacheLoadStarted(self):
  666. """ Test CacheLoadStarted event class """
  667. total = 123
  668. event = bb.event.CacheLoadStarted(total)
  669. self.assertEqual(event.msg, "Loading cache Started")
  670. self.assertEqual(event.total, total)
  671. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  672. def test_CacheLoadProgress(self):
  673. """ Test CacheLoadProgress event class """
  674. current = 10
  675. total = 100
  676. event = bb.event.CacheLoadProgress(current, total)
  677. self.assertEqual(event.msg,
  678. "Loading cache" + ": %s/%s" % (current, total))
  679. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  680. def test_CacheLoadCompleted(self):
  681. """ Test CacheLoadCompleted event class """
  682. total = 23
  683. num_entries = 12
  684. event = bb.event.CacheLoadCompleted(total, num_entries)
  685. self.assertEqual(event.msg, "Loading cache Completed")
  686. expected = [total, num_entries]
  687. actual = [event.total, event.num_entries]
  688. self.assertEqual(str(actual), str(expected))
  689. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  690. def test_TreeDataPreparationStarted(self):
  691. """ Test TreeDataPreparationStarted event class """
  692. event = bb.event.TreeDataPreparationStarted()
  693. self.assertEqual(event.msg, "Preparing tree data Started")
  694. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  695. def test_TreeDataPreparationProgress(self):
  696. """ Test TreeDataPreparationProgress event class """
  697. current = 10
  698. total = 100
  699. event = bb.event.TreeDataPreparationProgress(current, total)
  700. self.assertEqual(event.msg,
  701. "Preparing tree data" + ": %s/%s" % (current, total))
  702. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  703. def test_TreeDataPreparationCompleted(self):
  704. """ Test TreeDataPreparationCompleted event class """
  705. total = 23
  706. event = bb.event.TreeDataPreparationCompleted(total)
  707. self.assertEqual(event.msg, "Preparing tree data Completed")
  708. self.assertEqual(event.total, total)
  709. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  710. def test_DepTreeGenerated(self):
  711. """ Test DepTreeGenerated event class """
  712. depgraph = Mock()
  713. event = bb.event.DepTreeGenerated(depgraph)
  714. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  715. def test_TargetsTreeGenerated(self):
  716. """ Test TargetsTreeGenerated event class """
  717. model = Mock()
  718. event = bb.event.TargetsTreeGenerated(model)
  719. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  720. def test_ReachableStamps(self):
  721. """ Test ReachableStamps event class """
  722. stamps = [Mock(), Mock()]
  723. event = bb.event.ReachableStamps(stamps)
  724. self.assertEqual(event.stamps, stamps)
  725. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  726. def test_FilesMatchingFound(self):
  727. """ Test FilesMatchingFound event class """
  728. pattern = "foo.*bar"
  729. matches = ["foobar"]
  730. event = bb.event.FilesMatchingFound(pattern, matches)
  731. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  732. def test_ConfigFilesFound(self):
  733. """ Test ConfigFilesFound event class """
  734. variable = "FOO_BAR"
  735. values = ["foo", "bar"]
  736. event = bb.event.ConfigFilesFound(variable, values)
  737. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  738. def test_ConfigFilePathFound(self):
  739. """ Test ConfigFilePathFound event class """
  740. path = "/foo/bar"
  741. event = bb.event.ConfigFilePathFound(path)
  742. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  743. def test_message_classes(self):
  744. """ Test message event classes """
  745. msg = "foobar foo bar"
  746. event = bb.event.MsgBase(msg)
  747. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  748. event = bb.event.MsgDebug(msg)
  749. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  750. event = bb.event.MsgNote(msg)
  751. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  752. event = bb.event.MsgWarn(msg)
  753. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  754. event = bb.event.MsgError(msg)
  755. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  756. event = bb.event.MsgFatal(msg)
  757. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  758. event = bb.event.MsgPlain(msg)
  759. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  760. def test_LogExecTTY(self):
  761. """ Test LogExecTTY event class """
  762. msg = "foo bar"
  763. prog = "foo.sh"
  764. sleep_delay = 10
  765. retries = 3
  766. event = bb.event.LogExecTTY(msg, prog, sleep_delay, retries)
  767. self.assertEqual(event.msg, msg)
  768. self.assertEqual(event.prog, prog)
  769. self.assertEqual(event.sleep_delay, sleep_delay)
  770. self.assertEqual(event.retries, retries)
  771. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  772. def _throw_zero_division_exception(self):
  773. a = 1 / 0
  774. return
  775. def _worker_handler(self, event, d):
  776. self._returned_event = event
  777. return
  778. def test_LogHandler(self):
  779. """ Test LogHandler class """
  780. logger = logging.getLogger("TestEventClasses")
  781. logger.propagate = False
  782. handler = bb.event.LogHandler(logging.INFO)
  783. logger.addHandler(handler)
  784. bb.event.worker_fire = self._worker_handler
  785. try:
  786. self._throw_zero_division_exception()
  787. except ZeroDivisionError as ex:
  788. logger.exception(ex)
  789. event = self._returned_event
  790. try:
  791. pe = pickle.dumps(event)
  792. newevent = pickle.loads(pe)
  793. except:
  794. self.fail('Logged event is not serializable')
  795. self.assertEqual(event.taskpid, EventClassesTest._worker_pid)
  796. def test_MetadataEvent(self):
  797. """ Test MetadataEvent class """
  798. eventtype = "footype"
  799. eventdata = {"foo": "bar"}
  800. event = bb.event.MetadataEvent(eventtype, eventdata)
  801. self.assertEqual(event.type, eventtype)
  802. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  803. def test_ProcessStarted(self):
  804. """ Test ProcessStarted class """
  805. processname = "foo"
  806. total = 9783128974
  807. event = bb.event.ProcessStarted(processname, total)
  808. self.assertEqual(event.processname, processname)
  809. self.assertEqual(event.total, total)
  810. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  811. def test_ProcessProgress(self):
  812. """ Test ProcessProgress class """
  813. processname = "foo"
  814. progress = 243224
  815. event = bb.event.ProcessProgress(processname, progress)
  816. self.assertEqual(event.processname, processname)
  817. self.assertEqual(event.progress, progress)
  818. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  819. def test_ProcessFinished(self):
  820. """ Test ProcessFinished class """
  821. processname = "foo"
  822. total = 1242342344
  823. event = bb.event.ProcessFinished(processname)
  824. self.assertEqual(event.processname, processname)
  825. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  826. def test_SanityCheck(self):
  827. """ Test SanityCheck class """
  828. event1 = bb.event.SanityCheck()
  829. self.assertEqual(event1.generateevents, True)
  830. self.assertEqual(event1.pid, EventClassesTest._worker_pid)
  831. generateevents = False
  832. event2 = bb.event.SanityCheck(generateevents)
  833. self.assertEqual(event2.generateevents, generateevents)
  834. self.assertEqual(event2.pid, EventClassesTest._worker_pid)
  835. def test_SanityCheckPassed(self):
  836. """ Test SanityCheckPassed class """
  837. event = bb.event.SanityCheckPassed()
  838. self.assertEqual(event.pid, EventClassesTest._worker_pid)
  839. def test_SanityCheckFailed(self):
  840. """ Test SanityCheckFailed class """
  841. msg = "The sanity test failed."
  842. event1 = bb.event.SanityCheckFailed(msg)
  843. self.assertEqual(event1.pid, EventClassesTest._worker_pid)
  844. network_error = True
  845. event2 = bb.event.SanityCheckFailed(msg, network_error)
  846. self.assertEqual(event2.pid, EventClassesTest._worker_pid)
  847. def test_network_event_classes(self):
  848. """ Test network event classes """
  849. event1 = bb.event.NetworkTest()
  850. generateevents = False
  851. self.assertEqual(event1.pid, EventClassesTest._worker_pid)
  852. event2 = bb.event.NetworkTest(generateevents)
  853. self.assertEqual(event2.pid, EventClassesTest._worker_pid)
  854. event3 = bb.event.NetworkTestPassed()
  855. self.assertEqual(event3.pid, EventClassesTest._worker_pid)
  856. event4 = bb.event.NetworkTestFailed()
  857. self.assertEqual(event4.pid, EventClassesTest._worker_pid)
  858. def test_FindSigInfoResult(self):
  859. """ Test FindSigInfoResult event class """
  860. result = [Mock()]
  861. event = bb.event.FindSigInfoResult(result)
  862. self.assertEqual(event.result, result)
  863. self.assertEqual(event.pid, EventClassesTest._worker_pid)