policy_module/osm_policy_module/tests/test_policy_config_agent.py

   1 import json
   2 import os
   3 import unittest
   4
   5 from osm_policy_module.core.agent import PolicyModuleAgent
   6
   7
   8 class PolicyAgentTest(unittest.TestCase):
   9     def setUp(self):
  10         self.agent = PolicyModuleAgent()
  11
  12     def test_get_alarm_configs(self):
  13         with open(os.path.join(os.path.dirname(__file__), './examples/configure_scaling_full_example.json')) as file:
  14             example = json.load(file)
  15             alarm_configs = self.agent._get_alarm_configs(example)
  16             # TODO Improve assertions
  17             self.assertEqual(len(alarm_configs), 2)
  18
  19
  20 if __name__ == '__main__':
  21     unittest.main()