snaps/domain/test/network_tests.py

   1 # Copyright (c) 2017 Cable Television Laboratories, Inc. ("CableLabs")
   2 #                    and others.  All rights reserved.
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at:
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 import unittest
  17 from snaps.domain.network import SecurityGroup, SecurityGroupRule
  18
  19
  20 class SecurityGroupDomainObjectTests(unittest.TestCase):
  21     """
  22     Tests the construction of the snaps.domain.test.SecurityGroup class
  23     """
  24
  25     def test_construction_proj_id_kwargs(self):
  26         sec_grp = SecurityGroup(
  27             **{'name': 'name', 'id': 'id',
  28                'project_id': 'foo'})
  29         self.assertEqual('name', sec_grp.name)
  30         self.assertEqual('id', sec_grp.id)
  31         self.assertEqual('foo', sec_grp.project_id)
  32
  33     def test_construction_tenant_id_kwargs(self):
  34         sec_grp = SecurityGroup(
  35             **{'name': 'name', 'id': 'id',
  36                'tenant_id': 'foo'})
  37         self.assertEqual('name', sec_grp.name)
  38         self.assertEqual('id', sec_grp.id)
  39         self.assertEqual('foo', sec_grp.project_id)
  40
  41     def test_construction_named(self):
  42         sec_grp = SecurityGroup(tenant_id='foo', id='id', name='name')
  43         self.assertEqual('name', sec_grp.name)
  44         self.assertEqual('id', sec_grp.id)
  45         self.assertEqual('foo', sec_grp.project_id)
  46
  47
  48 class SecurityGroupRuleDomainObjectTests(unittest.TestCase):
  49     """
  50     Tests the construction of the snaps.domain.test.SecurityGroupRule class
  51     """
  52
  53     def test_construction_kwargs(self):
  54         sec_grp_rule = SecurityGroupRule(
  55             **{'id': 'id', 'security_group_id': 'grp_id', 'description': 'desc',
  56                'direction': 'dir', 'ethertype': 'eType',
  57                'port_range_min': '10.0.0.100', 'port_range_max': '10.0.0.200',
  58                'protocol': 'proto', 'remote_group_id': 'group_id',
  59                'remote_ip_prefix': 'ip_prefix'})
  60         self.assertEqual('id', sec_grp_rule.id)
  61         self.assertEqual('grp_id', sec_grp_rule.security_group_id)
  62         self.assertEqual('desc', sec_grp_rule.description)
  63         self.assertEqual('dir', sec_grp_rule.direction)
  64         self.assertEqual('eType', sec_grp_rule.ethertype)
  65         self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
  66         self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
  67         self.assertEqual('proto', sec_grp_rule.protocol)
  68         self.assertEqual('group_id', sec_grp_rule.remote_group_id)
  69         self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)
  70
  71     def test_construction_named(self):
  72         sec_grp_rule = SecurityGroupRule(
  73             remote_ip_prefix='ip_prefix', remote_group_id='group_id',
  74             protocol='proto', port_range_min='10.0.0.100',
  75             port_range_max='10.0.0.200', ethertype='eType',
  76             direction='dir', description='desc', security_group_id='grp_id',
  77             id='id')
  78         self.assertEqual('id', sec_grp_rule.id)
  79         self.assertEqual('grp_id', sec_grp_rule.security_group_id)
  80         self.assertEqual('desc', sec_grp_rule.description)
  81         self.assertEqual('dir', sec_grp_rule.direction)
  82         self.assertEqual('eType', sec_grp_rule.ethertype)
  83         self.assertEqual('10.0.0.100', sec_grp_rule.port_range_min)
  84         self.assertEqual('10.0.0.200', sec_grp_rule.port_range_max)
  85         self.assertEqual('proto', sec_grp_rule.protocol)
  86         self.assertEqual('group_id', sec_grp_rule.remote_group_id)
  87         self.assertEqual('ip_prefix', sec_grp_rule.remote_ip_prefix)