/test/programytest/config/brain/test_services.py

https://github.com/keiffster/program-y · Python · 103 lines · 80 code · 23 blank · 0 comment · 0 complexity · f889eff48a711cd586e2657eb05b03d1 MD5 · raw file 

    

  1. import unittest
    2. 

  2. 

  3. from programy.clients.events.console.config import ConsoleConfiguration
    4. 

  4. from programy.config.brain.services import BrainServicesConfiguration
    5. 

  5. from programy.config.file.yaml_file import YamlConfigurationFile
    6. 

  6. 

  7. 

  8. class BrainServicesConfigurationTests(unittest.TestCase):
    9. 

  9. 

  10.     def test_with_data(self):
    11. 

  11.         yaml = YamlConfigurationFile()
    12. 

  12.         self.assertIsNotNone(yaml)
    13. 

  13.         yaml.load_from_text("""
    14. 

  14.         brain:
    15. 

  15.             services:
    16. 

  16.                 REST:
    17. 

  17.                     classname: programy.services.rest.GenericRESTService
    18. 

  18.                     method: GET
    19. 

  19.                     host: 0.0.0.0
    20. 

  20.                 Pannous:
    21. 

  21.                     classname: programy.services.pannous.PannousService
    22. 

  22.                     url: http://weannie.pannous.com/api
    23. 

  23.                 Pandora:
    24. 

  24.                     classname: programy.services.pandora.PandoraService
    25. 

  25.                     url: http://www.pandorabots.com/pandora/talk-xml
    26. 

  26.                 Wikipedia:
    27. 

  27.                     classname: programy.services.wikipediaservice.WikipediaService
    28. 

  28.         """, ConsoleConfiguration(), ".")
    29. 

    30. 

  30.         brain_config = yaml.get_section("brain")
    31. 

  31. 

  32.         services_config = BrainServicesConfiguration()
    33. 

  33.         services_config.load_config_section(yaml, brain_config, ".")
    34. 

  34. 

  35.         self.assertTrue(services_config.exists("REST"))
    36. 

  36.         self.assertTrue(services_config.exists("Pannous"))
    37. 

  37.         self.assertTrue(services_config.exists("Pandora"))
    38. 

  38.         self.assertTrue(services_config.exists("Wikipedia"))
    39. 

  39.         self.assertFalse(services_config.exists("Other"))
    40. 

  40. 

  41.         self.assertIsNotNone(services_config.service("REST"))
    42. 

  42.         self.assertIsNone(services_config.service("REST2"))
    43. 

  43. 

  44.     def test_without_data(self):
    45. 

  45.         yaml = YamlConfigurationFile()
    46. 

  46.         self.assertIsNotNone(yaml)
    47. 

  47.         yaml.load_from_text("""
    48. 

  48.         brain:
    49. 

  49.             services:
    50. 

  50.         """, ConsoleConfiguration(), ".")
    51. 

    52. 

  52.         brain_config = yaml.get_section("brain")
    53. 

  53. 

  54.         services_config = BrainServicesConfiguration()
    55. 

  55.         services_config.load_config_section(yaml, brain_config, ".")
    56. 

  56. 

  57.         self.assertFalse(services_config.exists("REST"))
    58. 

  58.         self.assertFalse(services_config.exists("Pannous"))
    59. 

  59.         self.assertFalse(services_config.exists("Pandora"))
    60. 

  60.         self.assertFalse(services_config.exists("Wikipedia"))
    61. 

  61.         self.assertFalse(services_config.exists("Other"))
    62. 

  62. 

  63.     def test_with_no_data(self):
    64. 

  64.         yaml = YamlConfigurationFile()
    65. 

  65.         self.assertIsNotNone(yaml)
    66. 

  66.         yaml.load_from_text("""
    67. 

  67.         brain:
    68. 

  68.         """, ConsoleConfiguration(), ".")
    69. 

    70. 

  70.         brain_config = yaml.get_section("brain")
    71. 

  71. 

  72.         services_config = BrainServicesConfiguration()
    73. 

  73.         services_config.load_config_section(yaml, brain_config, ".")
    74. 

  74. 

  75.         self.assertFalse(services_config.exists("REST"))
    76. 

  76.         self.assertFalse(services_config.exists("Pannous"))
    77. 

  77.         self.assertFalse(services_config.exists("Pandora"))
    78. 

  78.         self.assertFalse(services_config.exists("Wikipedia"))
    79. 

  79.         self.assertFalse(services_config.exists("Other"))
    80. 

  80. 

  81.     def test_defaults(self):
    82. 

  82.         services_config = BrainServicesConfiguration()
    83. 

  83.         data = {}
    84. 

  84.         services_config.to_yaml(data, True)
    85. 

  85. 

  86.         BrainServicesConfigurationTests.assert_defaults(self, data)
    87. 

  87. 

  88.     @staticmethod
    89. 

  89.     def assert_defaults(test, data):
    90. 

  90.         test.assertEquals(data['REST']['classname'], 'programy.services.rest.GenericRESTService')
    91. 

  91.         test.assertEquals(data['REST']['method'], 'GET')
    92. 

  92.         test.assertEquals(data['REST']['host'], '0.0.0.0')
    93. 

  93. 

  94.         test.assertEquals(data['Pannous']['classname'], 'programy.services.pannous.PannousService')
    95. 

  95.         test.assertEquals(data['Pannous']['url'], 'http://weannie.pannous.com/api')
    96. 

  96. 

  97.         test.assertEquals(data['Pandora']['classname'], 'programy.services.pandora.PandoraService')
    98. 

  98.         test.assertEquals(data['Pandora']['url'], 'http://www.pandorabots.com/pandora/talk-xml')
    99. 

  99. 

  100.         test.assertEquals(data['Wikipedia']['classname'], 'programy.services.wikipediaservice.WikipediaService')
    101. 

  101. 

  102.         test.assertEquals(data['DuckDuckGo']['classname'], 'programy.services.duckduckgo.DuckDuckGoService')
    103. 

  103.         test.assertEquals(data['DuckDuckGo']['url'], 'http://api.duckduckgo.com')
    104.