test_yaml_loader.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. import unittest
  2. from core.ConfigurationManager.YAMLLoader import YAMLFileNotFound, YAMLLoader
  3. class TestYAMLLoader(unittest.TestCase):
  4. """
  5. Class to test YAMLLoader
  6. """
  7. def setUp(self):
  8. pass
  9. def test_get_config(self):
  10. valid_file_path_to_test = "core/Tests/brains/brain_test.yml"
  11. invalid_file_path = "brains/non_existing_brain.yml"
  12. expected_result = [
  13. {'signals': [{'order': 'test_order'}],
  14. 'neurons': [{'say': {'message': ['test message']}}],
  15. 'name': 'test'},
  16. {'signals': [{'order': 'test_order_2'}],
  17. 'neurons': [{'say': {'message': ['test message']}}],
  18. 'name': 'test2'},
  19. {'includes': ['included_brain_test.yml']},
  20. {'signals': [{'order': 'test_order_3'}],
  21. 'neurons': [{'say': {'message': ['test message']}}],
  22. 'name': 'test3'}
  23. ]
  24. with self.assertRaises(YAMLFileNotFound):
  25. YAMLLoader.get_config(invalid_file_path)
  26. self.assertEqual(YAMLLoader.get_config(valid_file_path_to_test), expected_result)
  27. if __name__ == '__main__':
  28. unittest.main()