tests/test_apex_network_environment.py

   1 ##############################################################################
   2 # Copyright (c) 2016 Dan Radez (Red Hat)
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10 import ipaddress
  11
  12 from apex.common.constants import (
  13     PUBLIC_NETWORK,
  14     PRIVATE_NETWORK,
  15     STORAGE_NETWORK,
  16     API_NETWORK)
  17 from apex.network_settings import NetworkSettings
  18 from apex.network_environment import (
  19     NetworkEnvironment,
  20     NetworkEnvException,
  21     EXTERNAL_RESOURCES,
  22     TENANT_RESOURCES,
  23     STORAGE_RESOURCES,
  24     API_RESOURCES)
  25
  26 from nose.tools import assert_equal
  27 from nose.tools import assert_raises
  28 from nose.tools import assert_is_instance
  29 from nose.tools import assert_not_equal
  30
  31
  32 class TestNetworkEnvironment(object):
  33     @classmethod
  34     def setup_class(klass):
  35         """This method is run once for each class before any tests are run"""
  36
  37     @classmethod
  38     def teardown_class(klass):
  39         """This method is run once for each class _after_ all tests are run"""
  40
  41     def setUp(self):
  42         """This method is run once before _each_ test method is executed"""
  43
  44     def teardown(self):
  45         """This method is run once after _each_ test method is executed"""
  46
  47     def test_init(self):
  48         assert_raises(NetworkEnvException, NetworkEnvironment,
  49                       None, '../build/network-environment.yaml')
  50
  51     def test_netenv_settings_public_network(self):
  52         ns = NetworkSettings('../config/network/network_settings.yaml', True)
  53         # test vlans
  54         ns[PUBLIC_NETWORK]['vlan'] = 100
  55         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
  56         assert_equal(ne['parameter_defaults']['NeutronExternalNetworkBridge'],
  57                      '""')
  58         assert_equal(ne['parameter_defaults']['ExternalNetworkVlanID'], 100)
  59
  60         # Test IPv6
  61         ns[PUBLIC_NETWORK]['cidr'] = ipaddress.ip_network('::1/128')
  62         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
  63         regstr = ne['resource_registry'][next(iter(EXTERNAL_RESOURCES.keys()))]
  64         assert_equal(regstr.split('/')[-1], 'external_v6.yaml')
  65
  66     def test_netenv_settings_private_network(self):
  67         ns = NetworkSettings('../config/network/network_settings.yaml', True)
  68         # test vlans
  69         ns[PRIVATE_NETWORK]['vlan'] = 100
  70         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
  71         assert_equal(ne['parameter_defaults']['TenantNetworkVlanID'], 100)
  72
  73         # Test IPv6
  74         ns[PRIVATE_NETWORK]['cidr'] = ipaddress.ip_network('::1/128')
  75         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
  76         regstr = ne['resource_registry'][next(iter(TENANT_RESOURCES.keys()))]
  77         assert_equal(regstr.split('/')[-1], 'tenant_v6.yaml')
  78
  79         # Test removing PRIVATE_NETWORK
  80         ns.enabled_network_list.remove(PRIVATE_NETWORK)
  81         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
  82         regstr = ne['resource_registry'][next(iter(TENANT_RESOURCES.keys()))]
  83         assert_equal(regstr.split('/')[-1], 'noop.yaml')
  84
  85     def test_netenv_settings_storage_network(self):
  86         ns = NetworkSettings('../config/network/network_settings.yaml', True)
  87         # test vlans
  88         ns[STORAGE_NETWORK]['vlan'] = 100
  89         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
  90         assert_equal(ne['parameter_defaults']['StorageNetworkVlanID'], 100)
  91
  92         # Test IPv6
  93         ns[STORAGE_NETWORK]['cidr'] = ipaddress.ip_network('::1/128')
  94         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
  95         regstr = ne['resource_registry'][next(iter(STORAGE_RESOURCES.keys()))]
  96         assert_equal(regstr.split('/')[-1], 'storage_v6.yaml')
  97
  98         # Test removing STORAGE_NETWORK
  99         ns.enabled_network_list.remove(STORAGE_NETWORK)
 100         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
 101         regstr = ne['resource_registry'][next(iter(STORAGE_RESOURCES.keys()))]
 102         assert_equal(regstr.split('/')[-1], 'noop.yaml')
 103
 104     def test_netenv_settings_api_network(self):
 105         ns = NetworkSettings('../config/network/network_settings.yaml', True)
 106         # test vlans
 107         ns.enabled_network_list.append(API_NETWORK)
 108         ns[API_NETWORK] = {'vlan': 100,
 109                            'cidr': ipaddress.ip_network('10.10.10.0/24'),
 110                            'usable_ip_range': '10.10.10.10,10.10.10.100'}
 111         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
 112         assert_equal(ne['parameter_defaults']['InternalApiNetworkVlanID'], 100)
 113
 114         # Test IPv6
 115         ns[API_NETWORK]['cidr'] = ipaddress.ip_network('::1/128')
 116         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
 117         regstr = ne['resource_registry'][next(iter(API_RESOURCES.keys()))]
 118         assert_equal(regstr.split('/')[-1], 'internal_api_v6.yaml')
 119
 120         # Test removing API_NETWORK
 121         ns.enabled_network_list.remove(API_NETWORK)
 122         ne = NetworkEnvironment(ns, '../build/network-environment.yaml')
 123         regstr = ne['resource_registry'][next(iter(API_RESOURCES.keys()))]
 124         assert_equal(regstr.split('/')[-1], 'noop.yaml')
 125
 126     def test_numa_configs(self):
 127         ns = NetworkSettings('../config/network/network_settings.yaml', True)
 128         ne = NetworkEnvironment(ns, '../build/network-environment.yaml',
 129                                 compute_pre_config=True,
 130                                 controller_pre_config=True)
 131         assert_is_instance(ne, dict)
 132         assert_not_equal(ne, {})
 133
 134     def test_exception(self):
 135         e = NetworkEnvException("test")
 136         print(e)
 137         assert_is_instance(e, NetworkEnvException)