test_settings_loader.py 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223
  1. import os
  2. import inspect
  3. import platform
  4. import shutil
  5. import unittest
  6. from kalliope.core.ConfigurationManager import SettingLoader
  7. from kalliope.core.Models import Singleton
  8. from kalliope.core.Models import Resources
  9. from kalliope.core.Models.Player import Player
  10. from kalliope.core.Models.RestAPI import RestAPI
  11. from kalliope.core.Models.Settings import Settings
  12. from kalliope.core.Models.Stt import Stt
  13. from kalliope.core.Models.Trigger import Trigger
  14. from kalliope.core.Models.Tts import Tts
  15. class TestSettingLoader(unittest.TestCase):
  16. def setUp(self):
  17. # get current script directory path. We are in /an/unknown/path/kalliope/core/tests
  18. cur_script_directory = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
  19. # get parent dir. Now we are in /an/unknown/path/kalliope
  20. root_dir = os.path.normpath(cur_script_directory + os.sep + os.pardir)
  21. self.settings_file_to_test = root_dir + os.sep + "Tests/settings/settings_test.yml"
  22. self.settings_dict = {
  23. 'default_synapse': 'Default-synapse',
  24. 'rest_api':
  25. {'allowed_cors_origin': False,
  26. 'active': True,
  27. 'login': 'admin',
  28. 'password_protected': True,
  29. 'password': 'secret', 'port': 5000},
  30. 'default_trigger': 'snowboy',
  31. 'default_player': 'mplayer',
  32. 'play_on_ready_notification': 'never',
  33. 'triggers': [{'snowboy': {'pmdl_file': 'trigger/snowboy/resources/kalliope-FR-6samples.pmdl'}}],
  34. 'players': [{'mplayer': {}}, {'pyalsaaudio': {"device": "default"}}],
  35. 'speech_to_text': [{'google': {'language': 'fr-FR'}}],
  36. 'on_ready_answers': ['Kalliope is ready'],
  37. 'cache_path': '/tmp/kalliope_tts_cache',
  38. 'random_wake_up_answers': ['Oui monsieur?'],
  39. 'on_ready_sounds': ['sounds/ding.wav', 'sounds/dong.wav'],
  40. 'resource_directory': {
  41. 'stt': '/tmp/kalliope/tests/kalliope_resources_dir/stt',
  42. 'tts': '/tmp/kalliope/tests/kalliope_resources_dir/tts',
  43. 'neuron': '/tmp/kalliope/tests/kalliope_resources_dir/neurons',
  44. 'trigger': '/tmp/kalliope/tests/kalliope_resources_dir/trigger'},
  45. 'default_text_to_speech': 'pico2wave',
  46. 'default_speech_to_text': 'google',
  47. 'random_wake_up_sounds': ['sounds/ding.wav', 'sounds/dong.wav'],
  48. 'text_to_speech': [
  49. {'pico2wave': {'cache': True, 'language': 'fr-FR'}},
  50. {'voxygen': {'voice': 'Agnes', 'cache': True}}
  51. ],
  52. 'var_files': ["../Tests/settings/variables.yml"]
  53. }
  54. # Init the folders, otherwise it raises an exceptions
  55. os.makedirs("/tmp/kalliope/tests/kalliope_resources_dir/neurons")
  56. os.makedirs("/tmp/kalliope/tests/kalliope_resources_dir/stt")
  57. os.makedirs("/tmp/kalliope/tests/kalliope_resources_dir/tts")
  58. os.makedirs("/tmp/kalliope/tests/kalliope_resources_dir/trigger")
  59. def tearDown(self):
  60. # Cleanup
  61. shutil.rmtree('/tmp/kalliope/tests/kalliope_resources_dir')
  62. Singleton._instances = {}
  63. def test_singleton(self):
  64. s1 = SettingLoader(file_path=self.settings_file_to_test)
  65. s2 = SettingLoader(file_path=self.settings_file_to_test)
  66. self.assertTrue(s1.settings is s2.settings)
  67. def test_get_yaml_config(self):
  68. sl = SettingLoader(file_path=self.settings_file_to_test)
  69. self.assertEqual(sl.yaml_config, self.settings_dict)
  70. def test_get_settings(self):
  71. settings_object = Settings()
  72. settings_object.default_tts_name = "pico2wave"
  73. settings_object.default_stt_name = "google"
  74. settings_object.default_trigger_name = "snowboy"
  75. settings_object.default_player_name = "mplayer"
  76. tts1 = Tts(name="pico2wave", parameters={'cache': True, 'language': 'fr-FR'})
  77. tts2 = Tts(name="voxygen", parameters={'voice': 'Agnes', 'cache': True})
  78. settings_object.ttss = [tts1, tts2]
  79. stt = Stt(name="google", parameters={'language': 'fr-FR'})
  80. settings_object.stts = [stt]
  81. settings_object.random_wake_up_answers = ['Oui monsieur?']
  82. settings_object.random_wake_up_sounds = ['sounds/ding.wav', 'sounds/dong.wav']
  83. settings_object.play_on_ready_notification = "never"
  84. settings_object.on_ready_answers = ['Kalliope is ready']
  85. settings_object.on_ready_sounds = ['sounds/ding.wav', 'sounds/dong.wav']
  86. trigger1 = Trigger(name="snowboy",
  87. parameters={'pmdl_file': 'trigger/snowboy/resources/kalliope-FR-6samples.pmdl'})
  88. settings_object.triggers = [trigger1]
  89. player1 = Player(name="mplayer", parameters={})
  90. player2 = Player(name="pyalsaaudio", parameters={"device": "default"})
  91. settings_object.players = [player1, player2]
  92. settings_object.rest_api = RestAPI(password_protected=True, active=True,
  93. login="admin", password="secret", port=5000,
  94. allowed_cors_origin=False)
  95. settings_object.cache_path = '/tmp/kalliope_tts_cache'
  96. settings_object.default_synapse = 'Default-synapse'
  97. resources = Resources(neuron_folder="/tmp/kalliope/tests/kalliope_resources_dir/neurons",
  98. stt_folder="/tmp/kalliope/tests/kalliope_resources_dir/stt",
  99. tts_folder="/tmp/kalliope/tests/kalliope_resources_dir/tts",
  100. trigger_folder="/tmp/kalliope/tests/kalliope_resources_dir/trigger")
  101. settings_object.resources = resources
  102. settings_object.variables = {
  103. "author": "Lamonf",
  104. "test_number": 60,
  105. "test": "kalliope"
  106. }
  107. settings_object.machine = platform.machine()
  108. sl = SettingLoader(file_path=self.settings_file_to_test)
  109. self.assertEqual(settings_object, sl.settings)
  110. def test_get_default_speech_to_text(self):
  111. expected_default_speech_to_text = "google"
  112. sl = SettingLoader(file_path=self.settings_file_to_test)
  113. self.assertEqual(expected_default_speech_to_text, sl._get_default_speech_to_text(self.settings_dict))
  114. def test_get_default_text_to_speech(self):
  115. expected_default_text_to_speech = "pico2wave"
  116. sl = SettingLoader(file_path=self.settings_file_to_test)
  117. self.assertEqual(expected_default_text_to_speech, sl._get_default_text_to_speech(self.settings_dict))
  118. def test_get_default_trigger(self):
  119. expected_default_trigger = "snowboy"
  120. sl = SettingLoader(file_path=self.settings_file_to_test)
  121. self.assertEqual(expected_default_trigger, sl._get_default_trigger(self.settings_dict))
  122. def test_get_stts(self):
  123. stt = Stt(name="google", parameters={'language': 'fr-FR'})
  124. sl = SettingLoader(file_path=self.settings_file_to_test)
  125. self.assertEqual([stt], sl._get_stts(self.settings_dict))
  126. def test_get_ttss(self):
  127. tts1 = Tts(name="pico2wave", parameters={'cache': True, 'language': 'fr-FR'})
  128. tts2 = Tts(name="voxygen", parameters={'voice': 'Agnes', 'cache': True})
  129. sl = SettingLoader(file_path=self.settings_file_to_test)
  130. self.assertEqual([tts1, tts2], sl._get_ttss(self.settings_dict))
  131. def test_get_triggers(self):
  132. trigger1 = Trigger(name="snowboy",
  133. parameters={'pmdl_file': 'trigger/snowboy/resources/kalliope-FR-6samples.pmdl'})
  134. sl = SettingLoader(file_path=self.settings_file_to_test)
  135. self.assertEqual([trigger1], sl._get_triggers(self.settings_dict))
  136. def test_get_players(self):
  137. player1 = Player(name="mplayer",
  138. parameters={})
  139. player2 = Player(name="pyalsaaudio",
  140. parameters={'device': 'default'})
  141. sl = SettingLoader(file_path=self.settings_file_to_test)
  142. self.assertEqual([player1, player2], sl._get_players(self.settings_dict))
  143. def test_get_random_wake_up_answers(self):
  144. expected_random_wake_up_answers = ['Oui monsieur?']
  145. sl = SettingLoader(file_path=self.settings_file_to_test)
  146. self.assertEqual(expected_random_wake_up_answers, sl._get_random_wake_up_answers(self.settings_dict))
  147. def test_get_on_ready_answers(self):
  148. expected_on_ready_answers = ['Kalliope is ready']
  149. sl = SettingLoader(file_path=self.settings_file_to_test)
  150. self.assertEqual(expected_on_ready_answers, sl._get_on_ready_answers(self.settings_dict))
  151. def test_get_on_ready_sounds(self):
  152. expected_on_ready_sounds = ['sounds/ding.wav', 'sounds/dong.wav']
  153. sl = SettingLoader(file_path=self.settings_file_to_test)
  154. self.assertEqual(expected_on_ready_sounds, sl._get_on_ready_sounds(self.settings_dict))
  155. def test_get_rest_api(self):
  156. expected_rest_api = RestAPI(password_protected=True, active=True,
  157. login="admin", password="secret", port=5000,
  158. allowed_cors_origin=False)
  159. sl = SettingLoader(file_path=self.settings_file_to_test)
  160. self.assertEqual(expected_rest_api, sl._get_rest_api(self.settings_dict))
  161. def test_get_cache_path(self):
  162. expected_cache_path = '/tmp/kalliope_tts_cache'
  163. sl = SettingLoader(file_path=self.settings_file_to_test)
  164. self.assertEqual(expected_cache_path, sl._get_cache_path(self.settings_dict))
  165. def test_get_default_synapse(self):
  166. expected_default_synapse = 'Default-synapse'
  167. sl = SettingLoader(file_path=self.settings_file_to_test)
  168. self.assertEqual(expected_default_synapse, sl._get_default_synapse(self.settings_dict))
  169. def test_get_resources(self):
  170. resources = Resources(neuron_folder="/tmp/kalliope/tests/kalliope_resources_dir/neurons",
  171. stt_folder="/tmp/kalliope/tests/kalliope_resources_dir/stt",
  172. tts_folder="/tmp/kalliope/tests/kalliope_resources_dir/tts",
  173. trigger_folder="/tmp/kalliope/tests/kalliope_resources_dir/trigger")
  174. expected_resource = resources
  175. sl = SettingLoader(file_path=self.settings_file_to_test)
  176. self.assertEqual(expected_resource, sl._get_resources(self.settings_dict))
  177. def test_get_variables(self):
  178. expected_result = {
  179. "author": "Lamonf",
  180. "test_number": 60,
  181. "test": "kalliope"
  182. }
  183. sl = SettingLoader(file_path=self.settings_file_to_test)
  184. self.assertEqual(expected_result,
  185. sl._get_variables(self.settings_dict))
  186. if __name__ == '__main__':
  187. unittest.main()