Merge "Added unit tests for vping."

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())