heat tests : add tests for new manager show_watch method

Add unit test for new show_watch engine action

Change-Id: I00153f258687e02837b901c399df51085bb34303
Signed-off-by: Steven Hardy <shardy@redhat.com>

diff --git a/heat/tests/test_stacks.py b/heat/tests/test_stacks.py
index caa590d56834a358762297ac1e2f7e44d3b68fc5..d48e8a1d8647ab20e0c12a6c6fd93b27ef9ef528 100644 (file)
--- a/heat/tests/test_stacks.py
+++ b/heat/tests/test_stacks.py
@@ -26,6 +26,7 @@ from nose import with_setup
 
 from heat.common import context
 from heat.tests.v1_1 import fakes
+import heat.engine.api as engine_api
 from heat.engine import instance as instances
 import heat.db as db_api
 from heat.engine import parser
@@ -362,6 +363,44 @@ class stackManagerTest(unittest.TestCase):
         self.assertFalse('AWS::CloudFormation::Init' in metadata)
         self.assertEqual(metadata, test_metadata)
 
+    def test_show_watch(self):
+        # Insert two dummy watch rules into the DB
+        values = {'stack_name': u'wordpress_ha', 'state': 'NORMAL',
+                  'name': u'HttpFailureAlarm',
+                   'rule': {
+                        u'EvaluationPeriods': u'1',
+                        u'AlarmActions': [u'WebServerRestartPolicy'],
+                        u'AlarmDescription': u'Restart the WikiDatabase',
+                        u'Namespace': u'system/linux',
+                        u'Period': u'300',
+                        u'ComparisonOperator': u'GreaterThanThreshold',
+                        u'Statistic': u'SampleCount',
+                        u'Threshold': u'2',
+                        u'MetricName': u'ServiceFailure'}}
+        db_ret = db_api.watch_rule_create(self.ctx, values)
+        self.assertNotEqual(db_ret, None)
+        values['name'] = "AnotherWatch"
+        db_ret = db_api.watch_rule_create(self.ctx, values)
+        self.assertNotEqual(db_ret, None)
+
+        # watch_name=None should return both watches
+        result = self.man.show_watch(self.ctx, watch_name=None)
+        self.assertEqual(2, len(result))
+
+        # watch_name="HttpFailureAlarm" should return only one
+        result = self.man.show_watch(self.ctx, watch_name="HttpFailureAlarm")
+        self.assertEqual(1, len(result))
+
+        # watch_name="nonexistent" should raise an AttributeError
+        self.assertRaises(AttributeError,
+                          self.man.show_watch,
+                          self.ctx, watch_name="nonexistent")
+
+        # Check the response has all keys defined in the engine API
+        for key in engine_api.WATCH_KEYS:
+            self.assertTrue(key in result[0])
+
+
 # allows testing of the test directly
 if __name__ == '__main__':
     sys.argv.append(__file__)