cloudstack/tools/testClient/testcase/BVT-tests/test_primary_storage.py

# -*- encoding: utf-8 -*-
#
# Copyright (c) 2012 Citrix.  All rights reserved.
#
""" BVT tests for Primary Storage
"""
#Import Local Modules
from cloudstackTestCase import *
from cloudstackAPI import *
from utils import *
from base import *

#Import System modules
import time

class Services:
    """Test Primary storage Services
    """

    def __init__(self):
        self.services = {
                        "nfs": {
                            0: {
                                "url": "nfs://192.168.100.131/Primary",
                                # Format: File_System_Type/Location/Path
                                "name": "Primary XEN",
                                "clusterid": 1, # XEN Cluster
                                "hypervisor": 'XEN',
                            },
                            1: {
                                "url": "nfs://192.168.100.131/Primary",
                                "name": "Primary KVM",
                                "clusterid": 40, # KVM Cluster
                                "hypervisor": 'KVM',
                            },
                            2: {
                                "url": "nfs://192.168.100.131/Primary",
                                "name": "Primary VMWare",
                                "clusterid": 33, # VMWare Cluster
                                "hypervisor": 'VMWare',
                            },
                        },
                        "iscsi": {
                            0: {
                                "url": "iscsi://192.168.100.21/iqn.2012-01.localdomain.clo-cstack-cos6:iser/1",
                                # Format : iscsi://IP Address/IQN number/LUN#
                                "name": "Primary iSCSI",
                                "clusterid": 1, # XEN Cluster
                                "hypervisor": 'XEN',
                            },
                            1: {
                                "url": "iscsi://192.168.100.21/export",
                                "name": "Primary KVM",
                                "clusterid": 1, # KVM Cluster
                                "hypervisor": 'KVM',
                            },
                        },
                 }

class TestPrimaryStorageServices(cloudstackTestCase):

    def setUp(self):

        self.apiclient = self.testClient.getApiClient()
        self.services = Services().services
        self.cleanup = []
        # Get Zone and pod
        self.zone = get_zone(self.apiclient)
        self.pod = get_pod(self.apiclient, self.zone.id)

        self.services["nfs"][0]["zoneid"] = self.zone.id
        self.services["nfs"][1]["zoneid"] = self.zone.id
        self.services["nfs"][2]["zoneid"] = self.zone.id

        self.services["nfs"][0]["podid"] = self.pod.id
        self.services["nfs"][1]["podid"] = self.pod.id
        self.services["nfs"][2]["podid"] = self.pod.id

        self.services["iscsi"][0]["zoneid"] = self.zone.id
        self.services["iscsi"][1]["zoneid"] = self.zone.id

        self.services["iscsi"][0]["podid"] = self.pod.id
        self.services["iscsi"][1]["podid"] = self.pod.id

        return

    def tearDown(self):
        try:
            #Clean up, terminate the created templates
            cleanup_resources(self.apiclient, self.cleanup)

        except Exception as e:
            raise Exception("Warning: Exception during cleanup : %s" % e)
        return

    def test_01_primary_storage(self):
        """Test primary storage pools - XEN, KVM, VMWare
        """

        # Validate the following:
        # 1. verify hypervisortype returned by api is Xen/KVM/VMWare
        # 2. verify that the cluster is in 'Enabled' allocation state
        # 3. verify that the host is added successfully and
        #    in Up state with listHosts api response

        #Create NFS storage pools with on XEN/KVM/VMWare clusters
        for k, v in self.services["nfs"].items():

            #Host should be present before adding primary storage
            cmd = listHosts.listHostsCmd()
            cmd.clusterid = v["clusterid"]
            list_hosts_response = self.apiclient.listHosts(cmd)

            self.assertNotEqual(
                len(list_hosts_response),
                0,
                "Check list Hosts for hypervisor: " + v["hypervisor"]
                )

            storage = StoragePool.create(self.apiclient, v)
            self.cleanup.append(storage)

            self.assertEqual(
                storage.state,
                'Up',
                "Check primary storage state for hypervisor: " + v["hypervisor"]
                )

            self.assertEqual(
                storage.type,
                'NetworkFilesystem',
                "Check storage pool type for hypervisor : " + v["hypervisor"]
                )

           #Verify List Storage pool Response has newly added storage pool
            cmd = listStoragePools.listStoragePoolsCmd()
            cmd.id = storage.id
            storage_pools_response = self.apiclient.listStoragePools(cmd)

            self.assertNotEqual(
                            len(storage_pools_response),
                            0,
                            "Check list Hosts response"
                        )

            storage_response = storage_pools_response[0]
            self.assertEqual(
                    storage_response.id,
                    storage.id,
                    "Check storage pool ID for hypervisor: " + v["hypervisor"]
                    )
            self.assertEqual(
                    storage.type,
                    storage_response.type,
                    "Check storage pool type for hypervisor: " + v["hypervisor"]
                )
            # Call cleanup for reusing primary storage
            cleanup_resources(self.apiclient, self.cleanup)
            self.cleanup = []

        # Create iSCSI storage pools with on XEN/KVM clusters
        for k, v in self.services["iscsi"].items():
            storage = StoragePool.create(self.apiclient, v)
            self.cleanup.append(storage)

            self.assertEqual(
                storage.state,
                'Up',
                "Check primary storage state for hypervisor: " + v["hypervisor"]
                )

            #Verify List Storage pool Response has newly added storage pool
            cmd = listStoragePools.listStoragePoolsCmd()
            cmd.id = storage.id
            storage_pools_response = self.apiclient.listStoragePools(cmd)

            self.assertNotEqual(
                len(storage_pools_response),
                0,
                "Check Hosts response for hypervisor: " + v["hypervisor"]
                        )

            storage_response = storage_pools_response[0]
            self.assertEqual(
                    storage_response.id,
                    storage.id,
                    "Check storage pool ID for hypervisor: " + v["hypervisor"]
                )
            self.assertEqual(
                    storage.type,
                    storage_response.type,
                    "Check storage pool type hypervisor: " + v["hypervisor"]
                )

            # Call cleanup for reusing primary storage
            cleanup_resources(self.apiclient, self.cleanup)
            self.cleanup = []
        return