Merge "Unit tests for autoscaling resources. 100% coverage!"

2 files changed, 171 insertions, 6 deletions

diff --git a/heat/engine/autoscaling.py b/heat/engine/autoscaling.py
index 4fe4ee2ef..93982c219 100644
--- a/heat/engine/autoscaling.py
+++ b/heat/engine/autoscaling.py
@@ -13,12 +13,6 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
-import eventlet
-import json
-import os
-
-from heat.common import exception
-from heat.db import api as db_api
 from heat.engine import instance
 from heat.engine.resources import Resource
 
diff --git a/heat/tests/test_autoscaling.py b/heat/tests/test_autoscaling.py
new file mode 100644
index 000000000..32156c0d6
--- /dev/null
+++ b/heat/tests/test_autoscaling.py
@@ -0,0 +1,171 @@
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+
+import sys
+import os
+
+import nose
+import unittest
+import mox
+import json
+
+from nose.plugins.attrib import attr
+
+from heat.engine import autoscaling as asc
+from heat.engine import loadbalancer
+from heat.engine import parser
+
+
+@attr(tag=['unit', 'resource'])
+@attr(speed='fast')
+class AutoScalingTest(unittest.TestCase):
+    def setUp(self):
+        self.m = mox.Mox()
+        self.m.StubOutWithMock(loadbalancer.LoadBalancer, 'reload')
+
+    def tearDown(self):
+        self.m.UnsetStubs()
+        print "AutoScalingTest teardown complete"
+
+    def load_template(self):
+        self.path = os.path.dirname(os.path.realpath(__file__)).\
+            replace('heat/tests', 'templates')
+        f = open("%s/AutoScalingMultiAZSample.template" % self.path)
+        t = json.loads(f.read())
+        f.close()
+        return t
+
+    def parse_stack(self, t):
+        class DummyContext():
+            tenant = 'test_tenant'
+            username = 'test_username'
+            password = 'password'
+            auth_url = 'http://localhost:5000/v2.0'
+        t['Parameters']['KeyName']['Value'] = 'test'
+        stack = parser.Stack(DummyContext(), 'test_stack', parser.Template(t),
+                             stack_id=-1)
+
+        return stack
+
+    def create_scaling_group(self, t, stack, resource_name):
+        resource = asc.AutoScalingGroup(resource_name,
+                                      t['Resources'][resource_name],
+                                      stack)
+        self.assertEqual(None, resource.validate())
+        self.assertEqual(None, resource.create())
+        self.assertEqual(asc.AutoScalingGroup.CREATE_COMPLETE, resource.state)
+        return resource
+
+    def create_scaling_policy(self, t, stack, resource_name):
+        resource = asc.ScalingPolicy(resource_name,
+                                      t['Resources'][resource_name],
+                                      stack)
+        self.assertEqual(None, resource.validate())
+        self.assertEqual(None, resource.create())
+        self.assertEqual(asc.ScalingPolicy.CREATE_COMPLETE,
+                         resource.state)
+        return resource
+
+    def test_scaling_group(self):
+
+        t = self.load_template()
+        stack = self.parse_stack(t)
+
+        # start with min then delete
+        resource = self.create_scaling_group(t, stack, 'WebServerGroup')
+
+        self.assertEqual('WebServerGroup', resource.FnGetRefId())
+        self.assertEqual('WebServerGroup-0', resource.instance_id)
+        self.assertEqual(asc.AutoScalingGroup.UPDATE_REPLACE,
+                  resource.handle_update())
+
+        resource.delete()
+
+        # start with 3
+        properties = t['Resources']['WebServerGroup']['Properties']
+        properties['DesiredCapacity'] = '3'
+        resource = self.create_scaling_group(t, stack, 'WebServerGroup')
+        self.assertEqual('WebServerGroup-0,WebServerGroup-1,WebServerGroup-2',
+                         resource.instance_id)
+
+        # reduce to 1
+        resource.adjust(-2)
+        self.assertEqual('WebServerGroup-0', resource.instance_id)
+
+        # raise to 3
+        resource.adjust(2)
+        self.assertEqual('WebServerGroup-0,WebServerGroup-1,WebServerGroup-2',
+                         resource.instance_id)
+
+        # set to 2
+        resource.adjust(2, 'ExactCapacity')
+        self.assertEqual('WebServerGroup-0,WebServerGroup-1',
+                         resource.instance_id)
+
+        # raise above the max
+        resource.adjust(2)
+        self.assertEqual('WebServerGroup-0,WebServerGroup-1',
+                         resource.instance_id)
+
+        # lower below the min
+        resource.adjust(-2)
+        self.assertEqual('WebServerGroup-0,WebServerGroup-1',
+                         resource.instance_id)
+
+        # no change
+        resource.adjust(0)
+        self.assertEqual('WebServerGroup-0,WebServerGroup-1',
+                         resource.instance_id)
+
+        # reduce by 50%
+        resource.adjust(-50, 'PercentChangeInCapacity')
+        self.assertEqual('WebServerGroup-0',
+                 resource.instance_id)
+
+        # raise by 200%
+        resource.adjust(200, 'PercentChangeInCapacity')
+        self.assertEqual('WebServerGroup-0,WebServerGroup-1,WebServerGroup-2',
+                 resource.instance_id)
+
+        resource.delete()
+
+    def test_scaling_policy(self):
+        t = self.load_template()
+        stack = self.parse_stack(t)
+
+        # start with min then delete
+        resource = self.create_scaling_group(t, stack, 'WebServerGroup')
+        stack.resources['WebServerGroup'] = resource
+
+        self.assertEqual('WebServerGroup-0', resource.instance_id)
+
+        up_policy = self.create_scaling_policy(t, stack,
+                                               'WebServerScaleUpPolicy')
+        up_policy.alarm()
+        self.assertEqual('WebServerGroup-0,WebServerGroup-1',
+                 resource.instance_id)
+
+        down_policy = self.create_scaling_policy(t, stack,
+                                                 'WebServerScaleDownPolicy')
+        down_policy.alarm()
+        self.assertEqual('WebServerGroup-0',
+                 resource.instance_id)
+
+        resource.delete()
+
+    # allows testing of the test directly, shown below
+    if __name__ == '__main__':
+        sys.argv.append(__file__)
+        nose.main()