diff options
Diffstat (limited to 'functest/tests/unit/openstack/vping/test_vping.py')
| -rw-r--r-- | functest/tests/unit/openstack/vping/test_vping.py | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/functest/tests/unit/openstack/vping/test_vping.py b/functest/tests/unit/openstack/vping/test_vping.py new file mode 100644 index 00000000..b229c351 --- /dev/null +++ b/functest/tests/unit/openstack/vping/test_vping.py @@ -0,0 +1,157 @@ +# Copyright (c) 2017 Cable Television Laboratories, Inc. and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# +# http://www.apache.org/licenses/LICENSE-2.0 + +import unittest + +import mock + +from snaps.openstack.create_image import OpenStackImage +from snaps.openstack.create_instance import OpenStackVmInstance, \ + VmInstanceSettings +from snaps.openstack.create_keypairs import OpenStackKeypair, KeypairSettings +from snaps.openstack.create_network import OpenStackNetwork, NetworkSettings, \ + SubnetSettings, PortSettings +from snaps.openstack.create_router import OpenStackRouter, RouterSettings +from snaps.openstack.create_security_group import OpenStackSecurityGroup, \ + SecurityGroupSettings +from snaps.openstack.os_credentials import OSCreds + +from functest.core.testcase import TestCase +from functest.opnfv_tests.openstack.vping import vping_userdata, vping_ssh + + +class VPingUserdataTesting(unittest.TestCase): + """ + Ensures the VPingUserdata class can run in Functest. This test does not + actually connect with an OpenStack pod. + """ + + def setUp(self): + self.os_creds = OSCreds( + username='user', password='pass', + auth_url='http://foo.com:5000/v3', project_name='bar') + + self.vping_userdata = vping_userdata.VPingUserdata( + os_creds=self.os_creds) + + @mock.patch('snaps.openstack.utils.deploy_utils.create_vm_instance') + @mock.patch('functest.opnfv_tests.openstack.vping.vping_base.os.' + 'path.exists', return_value=True) + @mock.patch('snaps.openstack.create_flavor.OpenStackFlavor.create', + return_value=None) + @mock.patch('snaps.openstack.create_instance.OpenStackVmInstance.' + 'get_port_ip', return_value='10.0.0.1') + @mock.patch('snaps.openstack.create_instance.OpenStackVmInstance.' + 'vm_active', return_value=True) + def test_vping_userdata(self, deploy_vm, path_exists, create_flavor, + get_port_ip, vm_active): + os_vm_inst = mock.MagicMock(name='get_console_output') + os_vm_inst.get_console_output.return_value = 'vPing OK' + with mock.patch('snaps.openstack.utils.deploy_utils.create_image', + return_value=OpenStackImage(self.os_creds, None)), \ + mock.patch('snaps.openstack.utils.deploy_utils.create_network', + return_value=OpenStackNetwork( + self.os_creds, NetworkSettings(name='foo'))), \ + mock.patch('snaps.openstack.utils.deploy_utils.' + 'create_vm_instance', + return_value=OpenStackVmInstance( + self.os_creds, + VmInstanceSettings( + name='foo', flavor='bar', + port_settings=[PortSettings( + name='foo', network_name='bar')]), + None)), \ + mock.patch('snaps.openstack.create_instance.' + 'OpenStackVmInstance.get_os_vm_server_obj', + return_value=os_vm_inst): + self.assertEquals(TestCase.EX_OK, self.vping_userdata.run()) + + +class VPingSSHTesting(unittest.TestCase): + """ + Ensures the VPingUserdata class can run in Functest. This test does not + actually connect with an OpenStack pod. + """ + + def setUp(self): + self.os_creds = OSCreds( + username='user', password='pass', + auth_url='http://foo.com:5000/v3', project_name='bar') + + self.vping_ssh = vping_ssh.VPingSSH( + os_creds=self.os_creds) + + @mock.patch('snaps.openstack.utils.deploy_utils.create_vm_instance') + @mock.patch('functest.opnfv_tests.openstack.vping.vping_base.os.' + 'path.exists', return_value=True) + @mock.patch('snaps.openstack.create_flavor.OpenStackFlavor.create', + return_value=None) + @mock.patch('snaps.openstack.create_instance.OpenStackVmInstance.' + 'get_port_ip', return_value='10.0.0.1') + @mock.patch('snaps.openstack.create_instance.OpenStackVmInstance.' + 'vm_active', return_value=True) + @mock.patch('snaps.openstack.create_instance.OpenStackVmInstance.' + 'vm_ssh_active', return_value=True) + @mock.patch('snaps.openstack.create_instance.OpenStackVmInstance.' + 'ssh_client', return_value=True) + @mock.patch('scp.SCPClient') + @mock.patch('functest.opnfv_tests.openstack.vping.vping_ssh.' + 'VPingSSH._transfer_ping_script', return_value=True) + @mock.patch('functest.opnfv_tests.openstack.vping.vping_ssh.' + 'VPingSSH._do_vping_ssh', return_value=TestCase.EX_OK) + @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.' + 'get_ext_net_name', return_value='foo') + def test_vping_ssh(self, create_vm, path_exists, + flavor_create, get_port_ip, vm_active, ssh_active, + ssh_client, scp_client, trans_script, do_vping_ssh, + ext_net_name): + os_vm_inst = mock.MagicMock(name='get_console_output') + os_vm_inst.get_console_output.return_value = 'vPing OK' + ssh_client = mock.MagicMock(name='get_transport') + ssh_client.get_transport.return_value = None + scp_client = mock.MagicMock(name='put') + scp_client.put.return_value = None + + with mock.patch('snaps.openstack.utils.deploy_utils.create_image', + return_value=OpenStackImage(self.os_creds, None)), \ + mock.patch('snaps.openstack.utils.deploy_utils.create_network', + return_value=OpenStackNetwork( + self.os_creds, + NetworkSettings( + name='foo', + subnet_settings=[ + SubnetSettings( + name='bar', + cidr='10.0.0.1/24')]))), \ + mock.patch('snaps.openstack.utils.deploy_utils.' + 'create_vm_instance', + return_value=OpenStackVmInstance( + self.os_creds, + VmInstanceSettings( + name='foo', flavor='bar', + port_settings=[PortSettings( + name='foo', network_name='bar')]), + None)), \ + mock.patch('snaps.openstack.utils.deploy_utils.create_keypair', + return_value=OpenStackKeypair( + self.os_creds, KeypairSettings(name='foo'))), \ + mock.patch('snaps.openstack.utils.deploy_utils.create_router', + return_value=OpenStackRouter( + self.os_creds, RouterSettings(name='foo'))), \ + mock.patch('snaps.openstack.utils.deploy_utils.' + 'create_security_group', + return_value=OpenStackSecurityGroup( + self.os_creds, + SecurityGroupSettings(name='foo'))), \ + mock.patch('snaps.openstack.create_instance.' + 'OpenStackVmInstance.' + 'get_vm_inst', return_value=os_vm_inst), \ + mock.patch('snaps.openstack.create_instance.' + 'OpenStackVmInstance.' + 'ssh_client', return_value=ssh_client): + self.assertEquals(TestCase.EX_OK, self.vping_ssh.run()) |