test_init.py 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  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.MainController.__init__') as mock_maincontroller:
  33. mock_maincontroller.return_value = None
  34. main()
  35. mock_maincontroller.assert_called()
  36. # test start gui
  37. sys.argv = ['kalliope.py', 'gui']
  38. with mock.patch('kalliope.core.ShellGui.__init__') as mock_shell_gui:
  39. mock_shell_gui.return_value = None
  40. main()
  41. mock_shell_gui.assert_called()
  42. # test run_synapse
  43. sys.argv = ['kalliope.py', 'start', '--run-synapse', 'synapse_name']
  44. with mock.patch('kalliope.core.SynapseLauncher.start_synapse_by_name') as mock_synapse_launcher:
  45. mock_synapse_launcher.return_value = None
  46. main()
  47. mock_synapse_launcher.assert_called()
  48. # test run order
  49. sys.argv = ['kalliope.py', 'start', '--run-order', 'my order']
  50. with mock.patch('kalliope.core.SynapseLauncher.run_matching_synapse_from_order') as mock_synapse_launcher:
  51. mock_synapse_launcher.return_value = None
  52. main()
  53. mock_synapse_launcher.assert_called()
  54. # action doesn't exist
  55. sys.argv = ['kalliope.py', 'non_existing_action']
  56. with self.assertRaises(SystemExit):
  57. main()
  58. # install
  59. sys.argv = ['kalliope.py', 'install', '--git-url', 'https://my_url']
  60. with mock.patch('kalliope.core.ResourcesManager.install') as mock_resource_manager:
  61. mock_resource_manager.return_value = None
  62. main()
  63. mock_resource_manager.assert_called()
  64. # install, no URL
  65. sys.argv = ['kalliope.py', 'install']
  66. with self.assertRaises(SystemExit):
  67. main()
  68. sys.argv = ['kalliope.py', 'install', '--git-url']
  69. with self.assertRaises(SystemExit):
  70. main()
  71. # uninstall
  72. sys.argv = ['kalliope.py', 'uninstall', '--neuron-name', 'neuron_name']
  73. with mock.patch('kalliope.core.ResourcesManager.uninstall') as mock_resource_manager:
  74. mock_resource_manager.return_value = None
  75. main()
  76. mock_resource_manager.assert_called()
  77. sys.argv = ['kalliope.py', 'uninstall']
  78. with self.assertRaises(SystemExit):
  79. main()
  80. if __name__ == '__main__':
  81. unittest.main()
  82. # suite = unittest.TestSuite()
  83. # suite.addTest(TestInit("test_main"))
  84. # runner = unittest.TextTestRunner()
  85. # runner.run(suite)