group-based-policy-automation/gbpautomation/heat/tests/test_grouppolicy.py

612 lines
22 KiB
Python

# 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 copy
from gbpautomation.heat.engine.resources.neutron import grouppolicy
from gbpclient.v2_0 import client as gbpclient
from heat.common import exception
from heat.common import template_format
from heat.tests.common import HeatTestCase
from heat.engine import scheduler
from heat.tests import utils
endpoint_template = '''
{
"AWSTemplateFormatVersion" : "2010-09-09",
"Description" : "Template to test neutron endpoint resource",
"Parameters" : {},
"Resources" : {
"endpoint": {
"Type": "OS::Neutron::Endpoint",
"Properties": {
"name": "test-endpoint",
"endpoint_group_id": "epg-id",
"description": "test endpoint resource"
}
}
}
}
'''
endpoint_group_template = '''
{
"AWSTemplateFormatVersion" : "2010-09-09",
"Description" : "Template to test neutron endpoint group resource",
"Parameters" : {},
"Resources" : {
"endpoint_group": {
"Type": "OS::Neutron::EndpointGroup",
"Properties": {
"name": "test-endpoint-group",
"description": "test endpoint group resource",
"l2_policy_id": "l2-policy-id",
"provided_contracts": [
{"contract_id": "contract1", "contract_scope": "scope1"},
{"contract_id": "contract2", "contract_scope": "scope2"}
],
"consumed_contracts": [
{"contract_id": "contract3", "contract_scope": "scope3"},
{"contract_id": "contract4", "contract_scope": "scope4"}
]
}
}
}
}
'''
l2_policy_template = '''
{
"AWSTemplateFormatVersion" : "2010-09-09",
"Description" : "Template to test neutron l2 policy",
"Parameters" : {},
"Resources" : {
"l2_policy": {
"Type": "OS::Neutron::L2Policy",
"Properties": {
"name": "test-l2-policy",
"description": "test L2 policy resource",
"l3_policy_id": "l3-policy-id"
}
}
}
}
'''
l3_policy_template = '''
{
"AWSTemplateFormatVersion" : "2010-09-09",
"Description" : "Template to test neutron l3 policy",
"Parameters" : {},
"Resources" : {
"l3_policy": {
"Type": "OS::Neutron::L3Policy",
"Properties": {
"name": "test-l3-policy",
"description": "test L3 policy resource",
"ip_version": "4",
"ip_pool": "10.20.20.0",
"subnet_prefix_length": 24
}
}
}
}
'''
class EndpointTest(HeatTestCase):
def setUp(self):
super(EndpointTest, self).setUp()
self.m.StubOutWithMock(gbpclient.Client, 'create_endpoint')
self.m.StubOutWithMock(gbpclient.Client, 'delete_endpoint')
self.m.StubOutWithMock(gbpclient.Client, 'show_endpoint')
self.m.StubOutWithMock(gbpclient.Client, 'update_endpoint')
self.stub_keystoneclient()
def create_endpoint(self):
gbpclient.Client.create_endpoint({
'endpoint': {
'name': 'test-endpoint',
'endpoint_group_id': 'epg-id',
"description": "test endpoint resource"
}
}).AndReturn({'endpoint': {'id': '5678'}})
snippet = template_format.parse(endpoint_template)
stack = utils.parse_stack(snippet)
resource_defns = stack.t.resource_definitions(stack)
return grouppolicy.Endpoint(
'endpoint', resource_defns['endpoint'], stack)
def test_create(self):
rsrc = self.create_endpoint()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
self.assertEqual((rsrc.CREATE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_create_failed(self):
gbpclient.Client.create_endpoint({
'endpoint': {
'name': 'test-endpoint',
'endpoint_group_id': 'epg-id',
"description": "test endpoint resource"
}
}).AndRaise(grouppolicy.NeutronClientException())
self.m.ReplayAll()
snippet = template_format.parse(endpoint_template)
stack = utils.parse_stack(snippet)
resource_defns = stack.t.resource_definitions(stack)
rsrc = grouppolicy.Endpoint(
'endpoint', resource_defns['endpoint'], stack)
error = self.assertRaises(exception.ResourceFailure,
scheduler.TaskRunner(rsrc.create))
self.assertEqual(
'NeutronClientException: An unknown exception occurred.',
str(error))
self.assertEqual((rsrc.CREATE, rsrc.FAILED), rsrc.state)
self.m.VerifyAll()
def test_delete(self):
gbpclient.Client.delete_endpoint('5678')
gbpclient.Client.show_endpoint('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=404))
rsrc = self.create_endpoint()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
scheduler.TaskRunner(rsrc.delete)()
self.assertEqual((rsrc.DELETE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_delete_already_gone(self):
gbpclient.Client.delete_endpoint('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=404))
rsrc = self.create_endpoint()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
scheduler.TaskRunner(rsrc.delete)()
self.assertEqual((rsrc.DELETE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_delete_failed(self):
gbpclient.Client.delete_endpoint('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=400))
rsrc = self.create_endpoint()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
error = self.assertRaises(exception.ResourceFailure,
scheduler.TaskRunner(rsrc.delete))
self.assertEqual(
'NeutronClientException: An unknown exception occurred.',
str(error))
self.assertEqual((rsrc.DELETE, rsrc.FAILED), rsrc.state)
self.m.VerifyAll()
def test_attribute(self):
rsrc = self.create_endpoint()
gbpclient.Client.show_endpoint('5678').MultipleTimes(
).AndReturn(
{'endpoint': {'neutron_port_id': '1234'}})
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
self.assertEqual('1234', rsrc.FnGetAtt('neutron_port_id'))
self.m.VerifyAll()
def test_attribute_failed(self):
rsrc = self.create_endpoint()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
error = self.assertRaises(exception.InvalidTemplateAttribute,
rsrc.FnGetAtt, 'l2_policy_id')
self.assertEqual(
'The Referenced Attribute (endpoint l2_policy_id) is '
'incorrect.', str(error))
self.m.VerifyAll()
def test_update(self):
rsrc = self.create_endpoint()
gbpclient.Client.update_endpoint(
'5678', {'endpoint': {'endpoint_group_id': 'epg_id_update'}})
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
update_template = copy.deepcopy(rsrc.t)
update_template['Properties']['endpoint_group_id'] = 'epg_id_update'
scheduler.TaskRunner(rsrc.update, update_template)()
self.m.VerifyAll()
class EndpointGroupTest(HeatTestCase):
def setUp(self):
super(EndpointGroupTest, self).setUp()
self.m.StubOutWithMock(gbpclient.Client, 'create_endpoint_group')
self.m.StubOutWithMock(gbpclient.Client, 'delete_endpoint_group')
self.m.StubOutWithMock(gbpclient.Client, 'show_endpoint_group')
self.m.StubOutWithMock(gbpclient.Client, 'update_endpoint_group')
self.stub_keystoneclient()
def create_endpoint_group(self):
gbpclient.Client.create_endpoint_group({
"endpoint_group": {
"name": "test-endpoint-group",
"description": "test endpoint group resource",
"l2_policy_id": "l2-policy-id",
"provided_contracts": {
"contract1": "scope1",
"contract2": "scope2"
},
"consumed_contracts": {
"contract3": "scope3",
"contract4": "scope4"
}
}
}).AndReturn({'endpoint_group': {'id': '5678'}})
snippet = template_format.parse(endpoint_group_template)
stack = utils.parse_stack(snippet)
resource_defns = stack.t.resource_definitions(stack)
return grouppolicy.EndpointGroup(
'endpoint_group', resource_defns['endpoint_group'], stack)
def test_create(self):
rsrc = self.create_endpoint_group()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
self.assertEqual((rsrc.CREATE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_create_failed(self):
gbpclient.Client.create_endpoint_group({
"endpoint_group": {
"name": "test-endpoint-group",
"description": "test endpoint group resource",
"l2_policy_id": "l2-policy-id",
"provided_contracts": {
"contract1": "scope1",
"contract2": "scope2"
},
"consumed_contracts": {
"contract3": "scope3",
"contract4": "scope4"
}
}
}).AndRaise(grouppolicy.NeutronClientException())
self.m.ReplayAll()
snippet = template_format.parse(endpoint_group_template)
stack = utils.parse_stack(snippet)
resource_defns = stack.t.resource_definitions(stack)
rsrc = grouppolicy.EndpointGroup(
'endpoint_group', resource_defns['endpoint_group'], stack)
error = self.assertRaises(exception.ResourceFailure,
scheduler.TaskRunner(rsrc.create))
self.assertEqual(
'NeutronClientException: An unknown exception occurred.',
str(error))
self.assertEqual((rsrc.CREATE, rsrc.FAILED), rsrc.state)
self.m.VerifyAll()
def test_delete(self):
gbpclient.Client.delete_endpoint_group('5678')
gbpclient.Client.show_endpoint_group('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=404))
rsrc = self.create_endpoint_group()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
scheduler.TaskRunner(rsrc.delete)()
self.assertEqual((rsrc.DELETE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_delete_already_gone(self):
gbpclient.Client.delete_endpoint_group('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=404))
rsrc = self.create_endpoint_group()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
scheduler.TaskRunner(rsrc.delete)()
self.assertEqual((rsrc.DELETE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_delete_failed(self):
gbpclient.Client.delete_endpoint_group('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=400))
rsrc = self.create_endpoint_group()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
error = self.assertRaises(exception.ResourceFailure,
scheduler.TaskRunner(rsrc.delete))
self.assertEqual(
'NeutronClientException: An unknown exception occurred.',
str(error))
self.assertEqual((rsrc.DELETE, rsrc.FAILED), rsrc.state)
self.m.VerifyAll()
def test_attribute_failed(self):
rsrc = self.create_endpoint_group()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
error = self.assertRaises(exception.InvalidTemplateAttribute,
rsrc.FnGetAtt, 'l3_policy_id')
self.assertEqual(
'The Referenced Attribute (endpoint_group l3_policy_id) is '
'incorrect.', str(error))
self.m.VerifyAll()
def test_update(self):
rsrc = self.create_endpoint_group()
gbpclient.Client.update_endpoint_group(
'5678', {'endpoint_group': {'l2_policy_id': 'l2_id_update'}})
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
update_template = copy.deepcopy(rsrc.t)
update_template['Properties']['l2_policy_id'] = 'l2_id_update'
scheduler.TaskRunner(rsrc.update, update_template)()
self.m.VerifyAll()
class L2PolicyTest(HeatTestCase):
def setUp(self):
super(L2PolicyTest, self).setUp()
self.m.StubOutWithMock(gbpclient.Client, 'create_l2_policy')
self.m.StubOutWithMock(gbpclient.Client, 'delete_l2_policy')
self.m.StubOutWithMock(gbpclient.Client, 'show_l2_policy')
self.m.StubOutWithMock(gbpclient.Client, 'update_l2_policy')
self.stub_keystoneclient()
def create_l2_policy(self):
gbpclient.Client.create_l2_policy({
'l2_policy': {
"name": "test-l2-policy",
"description": "test L2 policy resource",
"l3_policy_id": "l3-policy-id"
}
}).AndReturn({'l2_policy': {'id': '5678'}})
snippet = template_format.parse(l2_policy_template)
stack = utils.parse_stack(snippet)
resource_defns = stack.t.resource_definitions(stack)
return grouppolicy.L2Policy(
'l2_policy', resource_defns['l2_policy'], stack)
def test_create(self):
rsrc = self.create_l2_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
self.assertEqual((rsrc.CREATE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_create_failed(self):
gbpclient.Client.create_l2_policy({
'l2_policy': {
"name": "test-l2-policy",
"description": "test L2 policy resource",
"l3_policy_id": "l3-policy-id"
}
}).AndRaise(grouppolicy.NeutronClientException())
self.m.ReplayAll()
snippet = template_format.parse(l2_policy_template)
stack = utils.parse_stack(snippet)
resource_defns = stack.t.resource_definitions(stack)
rsrc = grouppolicy.L2Policy(
'l2_policy', resource_defns['l2_policy'], stack)
error = self.assertRaises(exception.ResourceFailure,
scheduler.TaskRunner(rsrc.create))
self.assertEqual(
'NeutronClientException: An unknown exception occurred.',
str(error))
self.assertEqual((rsrc.CREATE, rsrc.FAILED), rsrc.state)
self.m.VerifyAll()
def test_delete(self):
gbpclient.Client.delete_l2_policy('5678')
gbpclient.Client.show_l2_policy('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=404))
rsrc = self.create_l2_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
scheduler.TaskRunner(rsrc.delete)()
self.assertEqual((rsrc.DELETE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_delete_already_gone(self):
gbpclient.Client.delete_l2_policy('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=404))
rsrc = self.create_l2_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
scheduler.TaskRunner(rsrc.delete)()
self.assertEqual((rsrc.DELETE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_delete_failed(self):
gbpclient.Client.delete_l2_policy('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=400))
rsrc = self.create_l2_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
error = self.assertRaises(exception.ResourceFailure,
scheduler.TaskRunner(rsrc.delete))
self.assertEqual(
'NeutronClientException: An unknown exception occurred.',
str(error))
self.assertEqual((rsrc.DELETE, rsrc.FAILED), rsrc.state)
self.m.VerifyAll()
def test_attribute_failed(self):
rsrc = self.create_l2_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
error = self.assertRaises(exception.InvalidTemplateAttribute,
rsrc.FnGetAtt, 'endpoint_id')
self.assertEqual(
'The Referenced Attribute (l2_policy endpoint_id) is '
'incorrect.', str(error))
self.m.VerifyAll()
def test_update(self):
rsrc = self.create_l2_policy()
gbpclient.Client.update_l2_policy(
'5678', {'l2_policy': {'l3_policy_id': 'l3_id_update'}})
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
update_template = copy.deepcopy(rsrc.t)
update_template['Properties']['l3_policy_id'] = 'l3_id_update'
scheduler.TaskRunner(rsrc.update, update_template)()
self.m.VerifyAll()
class L3PolicyTest(HeatTestCase):
def setUp(self):
super(L3PolicyTest, self).setUp()
self.m.StubOutWithMock(gbpclient.Client, 'create_l3_policy')
self.m.StubOutWithMock(gbpclient.Client, 'delete_l3_policy')
self.m.StubOutWithMock(gbpclient.Client, 'show_l3_policy')
self.m.StubOutWithMock(gbpclient.Client, 'update_l3_policy')
self.stub_keystoneclient()
def create_l3_policy(self):
gbpclient.Client.create_l3_policy({
'l3_policy': {
"name": "test-l3-policy",
"description": "test L3 policy resource",
"ip_version": "4",
"ip_pool": "10.20.20.0",
"subnet_prefix_length": 24
}
}).AndReturn({'l3_policy': {'id': '5678'}})
snippet = template_format.parse(l3_policy_template)
stack = utils.parse_stack(snippet)
resource_defns = stack.t.resource_definitions(stack)
return grouppolicy.L3Policy(
'l3_policy', resource_defns['l3_policy'], stack)
def test_create(self):
rsrc = self.create_l3_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
self.assertEqual((rsrc.CREATE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_create_failed(self):
gbpclient.Client.create_l3_policy({
'l3_policy': {
"name": "test-l3-policy",
"description": "test L3 policy resource",
"ip_version": "4",
"ip_pool": "10.20.20.0",
"subnet_prefix_length": 24
}
}).AndRaise(grouppolicy.NeutronClientException())
self.m.ReplayAll()
snippet = template_format.parse(l3_policy_template)
stack = utils.parse_stack(snippet)
resource_defns = stack.t.resource_definitions(stack)
rsrc = grouppolicy.L3Policy(
'l3_policy', resource_defns['l3_policy'], stack)
error = self.assertRaises(exception.ResourceFailure,
scheduler.TaskRunner(rsrc.create))
self.assertEqual(
'NeutronClientException: An unknown exception occurred.',
str(error))
self.assertEqual((rsrc.CREATE, rsrc.FAILED), rsrc.state)
self.m.VerifyAll()
def test_delete(self):
gbpclient.Client.delete_l3_policy('5678')
gbpclient.Client.show_l3_policy('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=404))
rsrc = self.create_l3_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
scheduler.TaskRunner(rsrc.delete)()
self.assertEqual((rsrc.DELETE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_delete_already_gone(self):
gbpclient.Client.delete_l3_policy('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=404))
rsrc = self.create_l3_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
scheduler.TaskRunner(rsrc.delete)()
self.assertEqual((rsrc.DELETE, rsrc.COMPLETE), rsrc.state)
self.m.VerifyAll()
def test_delete_failed(self):
gbpclient.Client.delete_l3_policy('5678').AndRaise(
grouppolicy.NeutronClientException(status_code=400))
rsrc = self.create_l3_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
error = self.assertRaises(exception.ResourceFailure,
scheduler.TaskRunner(rsrc.delete))
self.assertEqual(
'NeutronClientException: An unknown exception occurred.',
str(error))
self.assertEqual((rsrc.DELETE, rsrc.FAILED), rsrc.state)
self.m.VerifyAll()
def test_attribute_failed(self):
rsrc = self.create_l3_policy()
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
error = self.assertRaises(exception.InvalidTemplateAttribute,
rsrc.FnGetAtt, 'subnet_id')
self.assertEqual(
'The Referenced Attribute (l3_policy subnet_id) is '
'incorrect.', str(error))
self.m.VerifyAll()
def test_update(self):
rsrc = self.create_l3_policy()
gbpclient.Client.update_l3_policy(
'5678', {'l3_policy': {'subnet_prefix_length': 28}})
self.m.ReplayAll()
scheduler.TaskRunner(rsrc.create)()
update_template = copy.deepcopy(rsrc.t)
update_template['Properties']['subnet_prefix_length'] = 28
scheduler.TaskRunner(rsrc.update, update_template)()
self.m.VerifyAll()