heat/heat/tests/test_stacks.py

import sys
import os

import nose
import unittest
import mox
import json
import sqlalchemy
from nose.plugins.attrib import attr
from nose import with_setup

from heat.common import context
from heat.tests.v1_1 import fakes
from heat.engine import instance as instances
import heat.db as db_api
from heat.engine import parser
from heat.engine import manager
from heat.engine import auth


@attr(tag=['unit', 'resource'])
@attr(speed='fast')
class stacksTest(unittest.TestCase):
    def setUp(self):
        self.m = mox.Mox()
        self.fc = fakes.FakeClient()
        self.path = os.path.dirname(os.path.realpath(__file__)).\
            replace('heat/tests', 'templates')

    def tearDown(self):
        self.m.UnsetStubs()
        print "stackTest teardown complete"

    # We use this in a number of tests so it's factored out here.
    def start_wordpress_stack(self, stack_name):
        f = open("%s/WordPress_Single_Instance_gold.template" % self.path)
        t = json.loads(f.read())
        f.close()
        params = {}
        parameters = {}
        t['Parameters']['KeyName']['Value'] = 'test'
        stack = parser.Stack(None, stack_name, t, 0, params)
        self.m.StubOutWithMock(instances.Instance, 'nova')
        instances.Instance.nova().AndReturn(self.fc)
        instances.Instance.nova().AndReturn(self.fc)
        instances.Instance.nova().AndReturn(self.fc)
        instances.Instance.nova().AndReturn(self.fc)
        instance = stack.resources['WebServer']
        instance.itype_oflavor['m1.large'] = 'm1.large'
        instance.calculate_properties()
        server_userdata = instance._build_userdata(
                                instance.properties['UserData'])
        self.m.StubOutWithMock(self.fc.servers, 'create')
        self.fc.servers.create(image=744, flavor=3, key_name='test',
                name='WebServer', security_groups=None,
                userdata=server_userdata).\
                AndReturn(self.fc.servers.list()[-1])
        return stack

    def test_wordpress_single_instance_stack_create(self):
        stack = self.start_wordpress_stack('test_stack')
        self.m.ReplayAll()
        stack.create()

        self.assertNotEqual(stack.resources['WebServer'], None)
        self.assertTrue(stack.resources['WebServer'].instance_id > 0)
        self.assertNotEqual(stack.resources['WebServer'].ipaddress, '0.0.0.0')

    def test_wordpress_single_instance_stack_delete(self):
        stack = self.start_wordpress_stack('test_stack')
        self.m.ReplayAll()
        rt = {}
        rt['template'] = stack.t
        rt['StackName'] = stack.name
        new_rt = db_api.raw_template_create(None, rt)
        ct = {'username': 'fred',
                   'password': 'mentions_fruit'}
        new_creds = db_api.user_creds_create(ct)
        s = {}
        s['name'] = stack.name
        s['raw_template_id'] = new_rt.id
        s['user_creds_id'] = new_creds.id
        s['username'] = ct['username']
        new_s = db_api.stack_create(None, s)
        stack.id = new_s.id
        stack.create()
        self.assertNotEqual(stack.resources['WebServer'], None)
        self.assertTrue(stack.resources['WebServer'].instance_id > 0)

        stack.delete()

        self.assertEqual(stack.resources['WebServer'].state, 'DELETE_COMPLETE')
        self.assertEqual(new_s.status, 'DELETE_COMPLETE')

    def test_stack_event_list(self):
        stack = self.start_wordpress_stack('test_event_list_stack')
        self.m.ReplayAll()
        rt = {}
        rt['template'] = stack.t
        rt['StackName'] = stack.name
        new_rt = db_api.raw_template_create(None, rt)
        ct = {'username': 'fred',
                   'password': 'mentions_fruit'}
        new_creds = db_api.user_creds_create(ct)
        s = {}
        s['name'] = stack.name
        s['raw_template_id'] = new_rt.id
        s['user_creds_id'] = new_creds.id
        s['username'] = ct['username']
        new_s = db_api.stack_create(None, s)
        stack.id = new_s.id
        stack.create()

        self.assertNotEqual(stack.resources['WebServer'], None)
        self.assertTrue(stack.resources['WebServer'].instance_id > 0)

        m = manager.EngineManager()
        events = db_api.event_get_all_by_stack(None, stack.id)
        for ev in events:
            result = m.parse_event(ev)
            self.assertTrue(result['EventId'] > 0)
            self.assertEqual(result['StackName'], "test_event_list_stack")
            self.assertTrue(result['ResourceStatus'] in ('IN_PROGRESS',
                                                         'CREATE_COMPLETE'))
            self.assertEqual(result['ResourceType'], 'AWS::EC2::Instance')
            self.assertEqual(result['ResourceStatusReason'], 'state changed')
            self.assertEqual(result['LogicalResourceId'], 'WebServer')
            # Big long user data field.. it mentions 'wordpress'
            # a few times so this should work.
            user_data = result['ResourceProperties']['UserData']
            self.assertNotEqual(user_data.find('wordpress'), -1)
            self.assertEqual(result['ResourceProperties']['ImageId'],
                             'F16-x86_64-gold')
            self.assertEqual(result['ResourceProperties']['InstanceType'],
                             'm1.large')

    def test_stack_list(self):
        stack = self.start_wordpress_stack('test_stack_list')
        rt = {}
        rt['template'] = stack.t
        rt['StackName'] = stack.name
        new_rt = db_api.raw_template_create(None, rt)
        ct = {'username': 'fred',
              'password': 'mentions_fruit'}
        new_creds = db_api.user_creds_create(ct)

        ctx = context.get_admin_context()
        self.m.StubOutWithMock(ctx, 'username')
        ctx.username = 'fred'
        self.m.StubOutWithMock(auth, 'authenticate')
        auth.authenticate(ctx).AndReturn(True)

        s = {}
        s['name'] = stack.name
        s['raw_template_id'] = new_rt.id
        s['user_creds_id'] = new_creds.id
        s['username'] = ct['username']
        new_s = db_api.stack_create(ctx, s)
        stack.id = new_s.id
        instances.Instance.nova().AndReturn(self.fc)
        self.m.ReplayAll()
        stack.create()

        f = open("%s/WordPress_Single_Instance_gold.template" % self.path)
        t = json.loads(f.read())
        params = {}
        parameters = {}
        t['Parameters']['KeyName']['Value'] = 'test'
        stack = parser.Stack(ctx, 'test_stack_list', t, 0, params)

        man = manager.EngineManager()
        sl = man.list_stacks(ctx, params)

        self.assertTrue(len(sl['stacks']) > 0)
        for s in sl['stacks']:
            self.assertTrue(s['StackId'] > 0)
            self.assertNotEqual(s['TemplateDescription'].find('WordPress'), -1)

    # allows testing of the test directly
    if __name__ == '__main__':
        sys.argv.append(__file__)
        nose.main()