krkn/tests/test_vmware_plugin.py

import unittest
import os
import os
import logging
from arcaflow_plugin_sdk import plugin
from kraken.plugins.vmware.kubernetes_functions import Actions
from kraken.plugins.vmware import vmware_plugin


class NodeScenariosTest(unittest.TestCase):
    def setUp(self):
        vsphere_env_vars = ["VSPHERE_IP", "VSPHERE_USERNAME", "VSPHERE_PASSWORD"]
        self.credentials_present = all(
            env_var in os.environ for env_var in vsphere_env_vars
        )

    def test_serialization(self):
        plugin.test_object_serialization(
            vmware_plugin.NodeScenarioConfig(name="test", skip_openshift_checks=True),
            self.fail,
        )
        plugin.test_object_serialization(
            vmware_plugin.NodeScenarioSuccessOutput(
                nodes={}, action=Actions.START
            ),
            self.fail,
        )
        plugin.test_object_serialization(
            vmware_plugin.NodeScenarioErrorOutput(
                error="Hello World", action=Actions.START
            ),
            self.fail,
        )

    def test_node_start(self):
        if not self.credentials_present:
            self.skipTest(
                "Check if the environmental variables 'VSPHERE_IP', 'VSPHERE_USERNAME', 'VSPHERE_PASSWORD' are set"
            )
        vsphere = vmware_plugin.vSphere(verify=False)
        vm_id, vm_name = vsphere.create_default_vm()
        if vm_id is None:
            self.fail("Could not create test VM")

        output_id, output_data = vmware_plugin.node_start(
            vmware_plugin.NodeScenarioConfig(
                name=vm_name, skip_openshift_checks=True, verify_session=False
            )
        )
        if output_id == "error":
            logging.error(output_data.error)
            self.fail("The VMware VM did not start because an error occurred")
        vsphere.release_instances(vm_name)

    def test_node_stop(self):
        if not self.credentials_present:
            self.skipTest(
                "Check if the environmental variables 'VSPHERE_IP', 'VSPHERE_USERNAME', 'VSPHERE_PASSWORD' are set"
            )
        vsphere = vmware_plugin.vSphere(verify=False)
        vm_id, vm_name = vsphere.create_default_vm()
        if vm_id is None:
            self.fail("Could not create test VM")
        vsphere.start_instances(vm_name)

        output_id, output_data = vmware_plugin.node_stop(
            vmware_plugin.NodeScenarioConfig(
                name=vm_name, skip_openshift_checks=True, verify_session=False
            )
        )
        if output_id == "error":
            logging.error(output_data.error)
            self.fail("The VMware VM did not stop because an error occurred")
        vsphere.release_instances(vm_name)

    def test_node_reboot(self):
        if not self.credentials_present:
            self.skipTest(
                "Check if the environmental variables 'VSPHERE_IP', 'VSPHERE_USERNAME', 'VSPHERE_PASSWORD' are set"
            )
        vsphere = vmware_plugin.vSphere(verify=False)
        vm_id, vm_name = vsphere.create_default_vm()
        if vm_id is None:
            self.fail("Could not create test VM")
        vsphere.start_instances(vm_name)

        output_id, output_data = vmware_plugin.node_reboot(
            vmware_plugin.NodeScenarioConfig(
                name=vm_name, skip_openshift_checks=True, verify_session=False
            )
        )
        if output_id == "error":
            logging.error(output_data.error)
            self.fail("The VMware VM did not reboot because an error occurred")
        vsphere.release_instances(vm_name)

    def test_node_terminate(self):
        if not self.credentials_present:
            self.skipTest(
                "Check if the environmental variables 'VSPHERE_IP', 'VSPHERE_USERNAME', 'VSPHERE_PASSWORD' are set"
            )
        vsphere = vmware_plugin.vSphere(verify=False)
        vm_id, vm_name = vsphere.create_default_vm()
        if vm_id is None:
            self.fail("Could not create test VM")
        vsphere.start_instances(vm_name)

        output_id, output_data = vmware_plugin.node_terminate(
            vmware_plugin.NodeScenarioConfig(
                name=vm_name, skip_openshift_checks=True, verify_session=False
            )
        )
        if output_id == "error":
            logging.error(output_data.error)
            self.fail("The VMware VM did not reboot because an error occurred")


if __name__ == "__main__":
    unittest.main()