test_init.py 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. import unittest
  2. import logging
  3. import sys
  4. import mock
  5. from kalliope import parse_args, configure_logging, main
  6. class TestInit(unittest.TestCase):
  7. def test_parse_args(self):
  8. # start option
  9. parser = parse_args(['value'])
  10. self.assertEqual(parser.action, "value")
  11. # no option
  12. with self.assertRaises(SystemExit):
  13. parse_args([])
  14. parser = parse_args(['start', '--run-synapse', 'run_synapse'])
  15. self.assertEqual(parser.run_synapse, 'run_synapse')
  16. parser = parse_args(['start', '--run-order', 'my order'])
  17. self.assertEqual(parser.run_order, 'my order')
  18. def test_configure_logging(self):
  19. logger = logging.getLogger("kalliope")
  20. # Level 10 = DEBUG
  21. configure_logging(debug=True)
  22. self.assertEqual(logger.getEffectiveLevel(), 10)
  23. logger.propagate = False
  24. # Level 20 = INFO
  25. configure_logging(debug=False)
  26. self.assertEqual(logger.getEffectiveLevel(), 20)
  27. # disable after testing
  28. logger.disabled = True
  29. def test_main(self):
  30. # test start kalliope
  31. sys.argv = ['kalliope.py', 'start']
  32. with mock.patch('kalliope.core.SignalLauncher.SignalLauncher.launch_signal_class_by_name') \
  33. as mock_signal_launcher:
  34. mock_signal_launcher.return_value = None
  35. main()
  36. mock_signal_launcher.assert_called()
  37. # test start gui
  38. sys.argv = ['kalliope.py', 'gui']
  39. with mock.patch('kalliope.core.ShellGui.__init__') as mock_shell_gui:
  40. mock_shell_gui.return_value = None
  41. main()
  42. mock_shell_gui.assert_called()
  43. # test run_synapse
  44. sys.argv = ['kalliope.py', 'start', '--run-synapse', 'synapse_name']
  45. with mock.patch('kalliope.core.SynapseLauncher.start_synapse_by_name') as mock_synapse_launcher:
  46. mock_synapse_launcher.return_value = None
  47. main()
  48. mock_synapse_launcher.assert_called()
  49. # test run order
  50. sys.argv = ['kalliope.py', 'start', '--run-order', 'my order']
  51. with mock.patch('kalliope.core.SynapseLauncher.run_matching_synapse_from_order') as mock_synapse_launcher:
  52. mock_synapse_launcher.return_value = None
  53. main()
  54. mock_synapse_launcher.assert_called()
  55. # action doesn't exist
  56. sys.argv = ['kalliope.py', 'non_existing_action']
  57. with self.assertRaises(SystemExit):
  58. main()
  59. # install
  60. sys.argv = ['kalliope.py', 'install', '--git-url', 'https://my_url']
  61. with mock.patch('kalliope.core.ResourcesManager.install') as mock_resource_manager:
  62. mock_resource_manager.return_value = None
  63. main()
  64. mock_resource_manager.assert_called()
  65. # install, no URL
  66. sys.argv = ['kalliope.py', 'install']
  67. with self.assertRaises(SystemExit):
  68. main()
  69. sys.argv = ['kalliope.py', 'install', '--git-url']
  70. with self.assertRaises(SystemExit):
  71. main()
  72. # uninstall
  73. sys.argv = ['kalliope.py', 'uninstall', '--neuron-name', 'neuron_name']
  74. with mock.patch('kalliope.core.ResourcesManager.uninstall') as mock_resource_manager:
  75. mock_resource_manager.return_value = None
  76. main()
  77. mock_resource_manager.assert_called()
  78. sys.argv = ['kalliope.py', 'uninstall']
  79. with self.assertRaises(SystemExit):
  80. main()
  81. if __name__ == '__main__':
  82. unittest.main()
  83. # suite = unittest.TestSuite()
  84. # suite.addTest(TestInit("test_main"))
  85. # runner = unittest.TextTestRunner()
  86. # runner.run(suite)