""" SoftLayer.tests.managers.ipsec_tests ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ :license: MIT, see LICENSE for more details. """ from mock import MagicMock import SoftLayer from SoftLayer.exceptions import SoftLayerAPIError from SoftLayer import testing class IPSECTests(testing.TestCase): def set_up(self): self.ipsec = SoftLayer.IPSECManager(self.client) def test_add_internal_subnet(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'addPrivateSubnetToNetworkTunnel') mock.return_value = True self.assertEqual(self.ipsec.add_internal_subnet(445, 565787), True) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'addPrivateSubnetToNetworkTunnel', args=(565787,), identifier=445) def test_add_remote_subnet(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'addCustomerSubnetToNetworkTunnel') mock.return_value = True self.assertEqual(self.ipsec.add_remote_subnet(445, 565787), True) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'addCustomerSubnetToNetworkTunnel', args=(565787,), identifier=445) def test_add_service_subnet(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'addServiceSubnetToNetworkTunnel') mock.return_value = True self.assertEqual(self.ipsec.add_service_subnet(445, 565787), True) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'addServiceSubnetToNetworkTunnel', args=(565787,), identifier=445) def test_apply_configuration(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'applyConfigurationsToDevice') mock.return_value = True self.assertEqual(self.ipsec.apply_configuration(445), True) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'applyConfigurationsToDevice', args=(), identifier=445) def test_create_remote_subnet(self): mock = self.set_mock('SoftLayer_Network_Customer_Subnet', 'createObject') mock.return_value = {'id': 565787, 'networkIdentifier': '50.0.0.0', 'cidr': 29, 'accountId': 999000} result = self.ipsec.create_remote_subnet(999000, '50.0.0.0', 29) self.assertEqual(result, mock.return_value) self.assert_called_with('SoftLayer_Network_Customer_Subnet', 'createObject', args=({'networkIdentifier': '50.0.0.0', 'cidr': 29, 'accountId': 999000},)) def test_create_translation(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'createAddressTranslation') mock.return_value = {'id': 787989, 'customerIpAddress': '50.0.0.0', 'customerIpAddressId': 672634, 'internalIpAddress': '10.0.0.0', 'internalIpAddressId': 871231, 'notes': 'first translation'} result = self.ipsec.create_translation(445, '10.0.0.0', '50.0.0.0', 'first translation') self.assertEqual(result, mock.return_value) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'createAddressTranslation', args=({'customerIpAddress': '50.0.0.0', 'internalIpAddress': '10.0.0.0', 'notes': 'first translation'},), identifier=445) def test_delete_remote_subnet(self): mock = self.set_mock('SoftLayer_Network_Customer_Subnet', 'deleteObject') mock.return_value = True self.assertEqual(self.ipsec.delete_remote_subnet(565787), True) self.assert_called_with('SoftLayer_Network_Customer_Subnet', 'deleteObject', identifier=565787) def test_get_tunnel_context(self): _filter = {'networkTunnelContexts': {'id': {'operation': 445}}} _mask = '[mask[id]]' mock = self.set_mock('SoftLayer_Account', 'getNetworkTunnelContexts') mock.return_value = [{'id': 445}] result = self.ipsec.get_tunnel_context(445, mask=_mask) self.assertEqual(result, mock.return_value[0]) self.assert_called_with('SoftLayer_Account', 'getNetworkTunnelContexts', filter=_filter, mask=_mask) def test_get_tunnel_context_raises_error(self): mock = self.set_mock('SoftLayer_Account', 'getNetworkTunnelContexts') mock.return_value = [] self.assertRaises(SoftLayerAPIError, self.ipsec.get_tunnel_context, 445) def test_get_translation(self): mock = self.set_mock('SoftLayer_Account', 'getNetworkTunnelContexts') mock.return_value = [{'id': 445, 'addressTranslations': [{'id': 234123}, {'id': 872341}]}] self.assertEqual(self.ipsec.get_translation(445, 872341), {'id': 872341, 'customerIpAddress': '', 'internalIpAddress': ''}) self.assert_called_with('SoftLayer_Account', 'getNetworkTunnelContexts') def test_get_translation_raises_error(self): mock = self.set_mock('SoftLayer_Account', 'getNetworkTunnelContexts') mock.return_value = [{'id': 445, 'addressTranslations': [{'id': 234123}]}] self.assertRaises(SoftLayerAPIError, self.ipsec.get_translation, 445, 872341) def test_get_translations(self): _mask = ('[mask[addressTranslations[customerIpAddressRecord,' 'internalIpAddressRecord]]]') _filter = {'networkTunnelContexts': {'id': {'operation': 445}}} mock = self.set_mock('SoftLayer_Account', 'getNetworkTunnelContexts') mock.return_value = [{'id': 445, 'addressTranslations': [{ 'id': 234123, 'customerIpAddressRecord': {'ipAddress': '50.0.0.0'}, 'customerIpAddressId': 234112, 'internalIpAddressRecord': {'ipAddress': '10.0.0.0'}, 'internalIpAddressId': 234442 }]}] self.assertEqual(self.ipsec.get_translations(445), [{'id': 234123, 'customerIpAddress': '50.0.0.0', 'customerIpAddressId': 234112, 'internalIpAddress': '10.0.0.0', 'internalIpAddressId': 234442}]) self.assert_called_with('SoftLayer_Account', 'getNetworkTunnelContexts', filter=_filter, mask=_mask) def test_get_tunnel_contexts(self): _mask = '[mask[addressTranslations]]' mock = self.set_mock('SoftLayer_Account', 'getNetworkTunnelContexts') mock.return_value = [{'id': 445}, {'id': 446}] self.assertEqual(self.ipsec.get_tunnel_contexts(mask=_mask), mock.return_value) self.assert_called_with('SoftLayer_Account', 'getNetworkTunnelContexts', mask=_mask) def test_remove_internal_subnet(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'removePrivateSubnetFromNetworkTunnel') mock.return_value = True self.assertEqual(self.ipsec.remove_internal_subnet(445, 565787), True) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'removePrivateSubnetFromNetworkTunnel', args=(565787,), identifier=445) def test_remove_remote_subnet(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'removeCustomerSubnetFromNetworkTunnel') mock.return_value = True self.assertEqual(self.ipsec.remove_remote_subnet(445, 565787), True) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'removeCustomerSubnetFromNetworkTunnel', args=(565787,), identifier=445) def test_remove_service_subnet(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'removeServiceSubnetFromNetworkTunnel') mock.return_value = True self.assertEqual(self.ipsec.remove_service_subnet(445, 565787), True) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'removeServiceSubnetFromNetworkTunnel', args=(565787,), identifier=445) def test_remove_translation(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'deleteAddressTranslation') mock.return_value = True self.assertEqual(self.ipsec.remove_translation(445, 787547), True) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'deleteAddressTranslation', args=(787547,), identifier=445) def test_update_translation(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'editAddressTranslation') mock.return_value = True translation = {'id': 234123, 'customerIpAddress': '50.0.0.0', 'customerIpAddressId': 234112, 'internalIpAddress': '10.0.0.0', 'internalIpAddressId': 234442} self.ipsec.get_translation = MagicMock(return_value=translation) result = self.ipsec.update_translation(445, 234123, static_ip='10.0.0.2', remote_ip='50.0.0.2', notes='do not touch') self.assertEqual(result, True) self.ipsec.get_translation.assert_called_with(445, 234123) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'editAddressTranslation', args=({'id': 234123, 'customerIpAddress': '50.0.0.2', 'internalIpAddress': '10.0.0.2', 'notes': 'do not touch'},), identifier=445) def test_update_tunnel_context(self): mock = self.set_mock('SoftLayer_Network_Tunnel_Module_Context', 'editObject') mock.return_value = True context = {'id': 445, 'name': 'der tunnel', 'friendlyName': 'the tunnel', 'internalPeerIpAddress': '10.0.0.1', 'customerPeerIpAddress': '50.0.0.1', 'advancedConfigurationFlag': 0, 'presharedKey': 'secret', 'phaseOneAuthentication': 'MD5', 'phaseOneDiffieHellmanGroup': 1, 'phaseOneEncryption': 'DES', 'phaseOneKeylife': 600, 'phaseTwoAuthentication': 'MD5', 'phaseTwoDiffieHellmanGroup': 1, 'phaseTwoEncryption': 'DES', 'phaseTwoKeylife': 600, 'phaseTwoPerfectForwardSecrecy': 0} self.ipsec.get_tunnel_context = MagicMock(return_value=context) result = self.ipsec.update_tunnel_context(445, friendly_name='ipsec tunnel', remote_peer='50.0.0.2', preshared_key='enigma', phase1_auth='SHA256', phase1_dh=5, phase1_crypto='AES256', phase1_key_ttl=120, phase2_auth='SHA128', phase2_dh=2, phase2_crypto='AES192', phase2_key_ttl=240, phase2_forward_secrecy=1) self.assertEqual(result, True) self.ipsec.get_tunnel_context.assert_called_with(445) self.assert_called_with('SoftLayer_Network_Tunnel_Module_Context', 'editObject', args=({'id': 445, 'name': 'der tunnel', 'friendlyName': 'ipsec tunnel', 'internalPeerIpAddress': '10.0.0.1', 'customerPeerIpAddress': '50.0.0.2', 'advancedConfigurationFlag': 0, 'presharedKey': 'enigma', 'phaseOneAuthentication': 'SHA256', 'phaseOneDiffieHellmanGroup': 5, 'phaseOneEncryption': 'AES256', 'phaseOneKeylife': 120, 'phaseTwoAuthentication': 'SHA128', 'phaseTwoDiffieHellmanGroup': 2, 'phaseTwoEncryption': 'AES192', 'phaseTwoKeylife': 240, 'phaseTwoPerfectForwardSecrecy': 1},), identifier=445)