python-group-based-policy-c.../gbpclient/tests/unit/test_cli20_policytargetgroup.py
Sumit Naiksatam 92c0197abe CLI for Applicaiton Policy Group
Supports CRUD operations for APGs, and associating/disassociating
APG with PTG(s).

Change-Id: I5f7798269a2414427971b868d0527e99e4a09e93
Implements: blueprint application-group
2017-03-13 21:49:30 -07:00

200 lines
8.6 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 logging
import sys
from gbpclient.gbp.v2_0 import groupbasedpolicy as gbp
from gbpclient.tests.unit import test_cli20
class CLITestV20PolicyTargetGroupJSON(test_cli20.CLITestV20Base):
LOG = logging.getLogger(__name__)
def setUp(self):
super(CLITestV20PolicyTargetGroupJSON, self).setUp()
def test_create_policy_target_group_with_mandatory_params(self):
"""policy-target-group-create with all mandatory params."""
resource = 'policy_target_group'
cmd = gbp.CreatePolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
name = 'my-name'
tenant_id = 'my-tenant'
my_id = 'my-id'
args = ['--tenant-id', tenant_id,
name]
position_names = ['name', ]
position_values = [name, ]
self._test_create_resource(resource, cmd, name, my_id, args,
position_names, position_values,
tenant_id=tenant_id)
def test_create_policy_target_group_with_all_params(self):
"""policy-target-group-create with all params."""
resource = 'policy_target_group'
cmd = gbp.CreatePolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
my_id = 'my-id'
tenant_id = 'my-tenant'
name = 'my-name'
description = 'ptg description'
application_policy_group_id = 'application_policy_group_id'
l2_policy_id = 'l2_policy_id'
provided_prs = "icmp-prs=false,web-prs=true"
consumed_prs = "ssh-prs=true,ftp-prs=false"
network_service_policy_id = 'network_service_policy_id'
shared = 'true'
intra_ptg_allow = 'false'
args = [name,
'--tenant-id', tenant_id,
'--description', description,
'--application-policy-group-id', application_policy_group_id,
'--l2-policy-id', l2_policy_id,
'--provided-policy-rule-sets', provided_prs,
'--consumed-policy-rule-sets', consumed_prs,
'--network-service-policy-id', network_service_policy_id,
'--shared', shared,
'--intra-ptg-allow', intra_ptg_allow]
position_names = ['name', 'description', 'application_policy_group_id',
'l2_policy_id',
'provided_policy_rule_sets',
'consumed_policy_rule_sets',
'network_service_policy_id']
provided_policy_rule_sets = {
'icmp-prs': 'false',
'web-prs': 'true'}
consumed_policy_rule_sets = {
'ssh-prs': 'true',
'ftp-prs': 'false'}
position_values = [name, description, application_policy_group_id,
l2_policy_id,
provided_policy_rule_sets,
consumed_policy_rule_sets,
network_service_policy_id]
self._test_create_resource(resource, cmd, name, my_id, args,
position_names, position_values,
tenant_id=tenant_id, shared=shared,
intra_ptg_allow=intra_ptg_allow)
def test_list_policy_target_groups(self):
"""policy-target-group-list."""
resource = 'policy_target_groups'
cmd = gbp.ListPolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
self._test_list_resources(resource, cmd, True)
def test_show_policy_target_group_name(self):
"""policy-target-group-show."""
resource = 'policy_target_group'
cmd = gbp.ShowPolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
args = ['--fields', 'id', self.test_id]
self._test_show_resource(resource, cmd, self.test_id, args, ['id'])
def test_update_policy_target_group_action(self):
"""policy-policy-action-update."""
resource = 'policy_target_group'
cmd = gbp.UpdatePolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
self._test_update_resource(resource, cmd, 'myid',
['myid', '--name', 'myname',
'--tags', 'a', 'b'],
{'name': 'myname', 'tags': ['a', 'b'], })
def test_update_policy_target_group_with_all_params(self):
"""policy-target-group-update."""
resource = 'policy_target_group'
cmd = gbp.UpdatePolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
my_id = 'my-id'
name = 'ptg'
description = 'ptg description'
application_policy_group_id = 'application_policy_group_id'
l2_policy_id = 'l2_policy_id'
provided_prs = "icmp-prs=false,web-prs=true"
consumed_prs = "ssh-prs=true,ftp-prs=false"
network_service_policy_id = 'network_service_policy_id'
shared = 'true'
intra_ptg_allow = 'false'
args = [my_id,
'--name', name,
'--description', description,
'--application-policy-group-id', application_policy_group_id,
'--l2-policy-id', l2_policy_id,
'--provided-policy-rule-sets', provided_prs,
'--consumed-policy-rule-sets', consumed_prs,
'--network-service-policy-id', network_service_policy_id,
'--shared', shared,
'--intra-ptg-allow', intra_ptg_allow]
provided_policy_rule_sets = {
'icmp-prs': 'false',
'web-prs': 'true'}
consumed_policy_rule_sets = {
'ssh-prs': 'true',
'ftp-prs': 'false'}
params = {
'name': name,
'description': description,
'application_policy_group_id': application_policy_group_id,
'l2_policy_id': l2_policy_id,
'provided_policy_rule_sets': provided_policy_rule_sets,
'consumed_policy_rule_sets': consumed_policy_rule_sets,
'network_service_policy_id': network_service_policy_id,
'shared': shared,
'intra_ptg_allow': intra_ptg_allow
}
self._test_update_resource(resource, cmd, my_id, args, params)
def test_update_policy_target_group_unset_prs(self):
"""policy-target-group-update."""
resource = 'policy_target_group'
cmd = gbp.UpdatePolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
my_id = 'my-id'
provided_prs = ""
consumed_prs = ""
args = [my_id,
'--provided-policy-rule-sets', provided_prs,
'--consumed-policy-rule-sets', consumed_prs
]
params = {
'provided_policy_rule_sets': {},
'consumed_policy_rule_sets': {}
}
self._test_update_resource(resource, cmd, my_id, args, params)
def test_update_policy_target_group_unset_nsp(self):
"""policy-target-group-update."""
resource = 'policy_target_group'
cmd = gbp.UpdatePolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
my_id = 'my-id'
network_service_policy_id = ''
args = [my_id,
'--network-service-policy', network_service_policy_id]
params = {'network_service_policy_id': None}
self._test_update_resource(resource, cmd, my_id, args, params)
def test_update_policy_target_group_unset_apg(self):
"""policy-target-group-update."""
resource = 'policy_target_group'
cmd = gbp.UpdatePolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
my_id = 'my-id'
application_policy_group_id = ''
args = [my_id,
'--application-policy-group', application_policy_group_id]
params = {'application_policy_group_id': None}
self._test_update_resource(resource, cmd, my_id, args, params)
def test_delete_policy_target_group_name(self):
"""policy-target-group-delete."""
resource = 'policy_target_group'
cmd = gbp.DeletePolicyTargetGroup(test_cli20.MyApp(sys.stdout), None)
my_id = 'my-id'
args = [my_id]
self._test_delete_resource(resource, cmd, my_id, args)