diff --git a/tests/test_vmware_plugin.py b/tests/test_vmware_plugin.py index b8fea27b..b00c8bbc 100644 --- a/tests/test_vmware_plugin.py +++ b/tests/test_vmware_plugin.py @@ -1,6 +1,5 @@ import unittest import os -import os import logging from arcaflow_plugin_sdk import plugin from kraken.plugins.vmware.kubernetes_functions import Actions @@ -9,14 +8,21 @@ from kraken.plugins.vmware import vmware_plugin class NodeScenariosTest(unittest.TestCase): def setUp(self): - vsphere_env_vars = ["VSPHERE_IP", "VSPHERE_USERNAME", "VSPHERE_PASSWORD"] + 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), + vmware_plugin.NodeScenarioConfig( + name="test", + skip_openshift_checks=True + ), self.fail, ) plugin.test_object_serialization( @@ -35,7 +41,8 @@ class NodeScenariosTest(unittest.TestCase): 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" + "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() @@ -55,7 +62,8 @@ class NodeScenariosTest(unittest.TestCase): 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" + "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() @@ -76,7 +84,8 @@ class NodeScenariosTest(unittest.TestCase): 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" + "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() @@ -97,7 +106,8 @@ class NodeScenariosTest(unittest.TestCase): 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" + "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()