Commit 7c168be0 authored by Łukasz Nowak's avatar Łukasz Nowak

software/kvm: Flake8ize test file

parent 5adadd1b
...@@ -38,9 +38,10 @@ import unittest ...@@ -38,9 +38,10 @@ import unittest
from slapos.recipe.librecipe import generateHashFromFiles from slapos.recipe.librecipe import generateHashFromFiles
from slapos.testing.testcase import makeModuleSetUpAndTestCaseClass from slapos.testing.testcase import makeModuleSetUpAndTestCaseClass
has_kvm = os.access('/dev/kvm', os.R_OK|os.W_OK) has_kvm = os.access('/dev/kvm', os.R_OK | os.W_OK)
skipUnlessKvm = unittest.skipUnless(has_kvm, 'kvm not loaded or not allowed') skipUnlessKvm = unittest.skipUnless(has_kvm, 'kvm not loaded or not allowed')
skipIfPython3 = unittest.skipIf(six.PY3, 'rdiff-backup is not compatible with Python 3 yet') skipIfPython3 = unittest.skipIf(
six.PY3, 'rdiff-backup is not compatible with Python 3 yet')
if has_kvm: if has_kvm:
setUpModule, InstanceTestCase = makeModuleSetUpAndTestCaseClass( setUpModule, InstanceTestCase = makeModuleSetUpAndTestCaseClass(
...@@ -57,19 +58,22 @@ else: ...@@ -57,19 +58,22 @@ else:
bootstrap_common_param_dict = { bootstrap_common_param_dict = {
# the bootstrap script is vm-bootstrap # the bootstrap script is vm-bootstrap
"bootstrap-script-url": "http://shacache.org/shacache/05105cd25d1ad798b71fd46a206c9b73da2c285a078af33d0e739525a595886785725a68811578bc21f75d0a97700a66d5e75bce5b2721ca4556a0734cb13e65#c98825aa1b6c8087914d2bfcafec3058", "bootstrap-script-url":
"http://shacache.org/shacache/05105cd25d1ad798b71fd46a206c9b73da2c285a078"
"af33d0e739525a595886785725a68811578bc21f75d0a97700a66d5e75bce5b2721ca455"
"6a0734cb13e65#c98825aa1b6c8087914d2bfcafec3058",
"slave-frontend": { "slave-frontend": {
"slave-frontend-dict": {} "slave-frontend-dict": {}
}, },
"authorized-keys": [ "authorized-keys": [
"ssh-rsa %s key_one" % ("A" * 372), "ssh-rsa %s key_one" % ("A" * 372),
"ssh-rsa %s key_two" % ("B" * 372), "ssh-rsa %s key_two" % ("B" * 372),
"ssh-rsa %s key_three" % ("C" * 372) "ssh-rsa %s key_three" % ("C" * 372)
], ],
"fw-restricted-access": "off", "fw-restricted-access": "off",
"fw-authorized-sources": [], "fw-authorized-sources": [],
"fw-reject-sources": ["10.32.0.0/13"] "fw-reject-sources": ["10.32.0.0/13"]
} }
bootstrap_machine_param_dict = { bootstrap_machine_param_dict = {
"computer-guid": "local", "computer-guid": "local",
...@@ -80,20 +84,23 @@ bootstrap_machine_param_dict = { ...@@ -80,20 +84,23 @@ bootstrap_machine_param_dict = {
"cpu-count": 2, "cpu-count": 2,
"disk-size": 50, "disk-size": 50,
# Debian 10 image # Debian 10 image
"virtual-hard-drive-url": "http://shacache.org/shacache/9d3e6d017754fdd08e5ecf78093dec27fd792fb183df6146006adf003b6f4b98c0388d5a11566627101f7855d77f60e3dd4ba7ce66850f4a8f030573b904d5ab", "virtual-hard-drive-url":
"http://shacache.org/shacache/9d3e6d017754fdd08e5ecf78093dec27fd792fb183d"
"f6146006adf003b6f4b98c0388d5a11566627101f7855d77f60e3dd4ba7ce66850f4a8f0"
"30573b904d5ab",
"virtual-hard-drive-md5sum": "b7928d7b0a2b5e2888f5ddf68f5fe422", "virtual-hard-drive-md5sum": "b7928d7b0a2b5e2888f5ddf68f5fe422",
"virtual-hard-drive-gzipped": False, "virtual-hard-drive-gzipped": False,
"hard-drive-url-check-certificate": False, "hard-drive-url-check-certificate": False,
"use-tap": True, "use-tap": True,
"use-nat": True, "use-nat": True,
"nat-restrict-mode":True, "nat-restrict-mode": True,
"enable-vhost": True, "enable-vhost": True,
"external-disk-number": 1, "external-disk-number": 1,
"external-disk-size": 100, "external-disk-size": 100,
"external-disk-format": "qcow2", "external-disk-format": "qcow2",
"enable-monitor": True, "enable-monitor": True,
"keyboard-layout-language": "fr" "keyboard-layout-language": "fr"
} }
@skipUnlessKvm @skipUnlessKvm
class ServicesTestCase(InstanceTestCase): class ServicesTestCase(InstanceTestCase):
def test_hashes(self): def test_hashes(self):
...@@ -237,6 +244,7 @@ class TestAccessDefaultAdditional(MonitorAccessMixin, InstanceTestCase): ...@@ -237,6 +244,7 @@ class TestAccessDefaultAdditional(MonitorAccessMixin, InstanceTestCase):
) )
self.assertIn('<title>noVNC</title>', result.text) self.assertIn('<title>noVNC</title>', result.text)
@skipUnlessKvm @skipUnlessKvm
class TestAccessDefaultBootstrap(MonitorAccessMixin, InstanceTestCase): class TestAccessDefaultBootstrap(MonitorAccessMixin, InstanceTestCase):
__partition_reference__ = 'adb' __partition_reference__ = 'adb'
...@@ -244,7 +252,8 @@ class TestAccessDefaultBootstrap(MonitorAccessMixin, InstanceTestCase): ...@@ -244,7 +252,8 @@ class TestAccessDefaultBootstrap(MonitorAccessMixin, InstanceTestCase):
@classmethod @classmethod
def getInstanceParameterDict(cls): def getInstanceParameterDict(cls):
return {'_': json.dumps(dict(bootstrap_common_param_dict, **bootstrap_machine_param_dict))} return {'_': json.dumps(dict(
bootstrap_common_param_dict, **bootstrap_machine_param_dict))}
def test(self): def test(self):
connection_parameter_dict = self.computer_partition\ connection_parameter_dict = self.computer_partition\
...@@ -257,6 +266,7 @@ class TestAccessDefaultBootstrap(MonitorAccessMixin, InstanceTestCase): ...@@ -257,6 +266,7 @@ class TestAccessDefaultBootstrap(MonitorAccessMixin, InstanceTestCase):
) )
self.assertIn('<title>noVNC</title>', result.text) self.assertIn('<title>noVNC</title>', result.text)
@skipUnlessKvm @skipUnlessKvm
class TestAccessKvmCluster(MonitorAccessMixin, InstanceTestCase): class TestAccessKvmCluster(MonitorAccessMixin, InstanceTestCase):
__partition_reference__ = 'akc' __partition_reference__ = 'akc'
...@@ -328,6 +338,7 @@ class TestAccessKvmClusterAdditional(MonitorAccessMixin, InstanceTestCase): ...@@ -328,6 +338,7 @@ class TestAccessKvmClusterAdditional(MonitorAccessMixin, InstanceTestCase):
) )
self.assertIn('<title>noVNC</title>', result.text) self.assertIn('<title>noVNC</title>', result.text)
@skipIfPython3 @skipIfPython3
@skipUnlessKvm @skipUnlessKvm
class TestAccessKvmClusterBootstrap(MonitorAccessMixin, InstanceTestCase): class TestAccessKvmClusterBootstrap(MonitorAccessMixin, InstanceTestCase):
...@@ -345,28 +356,34 @@ class TestAccessKvmClusterBootstrap(MonitorAccessMixin, InstanceTestCase): ...@@ -345,28 +356,34 @@ class TestAccessKvmClusterBootstrap(MonitorAccessMixin, InstanceTestCase):
"test-machine1": bootstrap_machine_param_dict, "test-machine1": bootstrap_machine_param_dict,
"test-machine2": dict(bootstrap_machine_param_dict, **{ "test-machine2": dict(bootstrap_machine_param_dict, **{
# Debian 9 image # Debian 9 image
"virtual-hard-drive-url": "http://shacache.org/shacache/ce07873dbab7fa8501d1bf5565c2737b2eed6c8b9361b4997b21daf5f5d1590972db9ac00131cc5b27d9aa353f2f94071e073f9980cc61badd6d2427f592e6e8", "virtual-hard-drive-url":
"http://shacache.org/shacache/ce07873dbab7fa8501d1bf5565c2737b2"
"eed6c8b9361b4997b21daf5f5d1590972db9ac00131cc5b27d9aa353f2f940"
"71e073f9980cc61badd6d2427f592e6e8",
"virtual-hard-drive-md5sum": "2b113e3cd8276b9740189622603d6f99" "virtual-hard-drive-md5sum": "2b113e3cd8276b9740189622603d6f99"
}) })
} }
}))} }))}
def test(self): def test(self):
connection_parameter_dict = self.computer_partition\ connection_parameter_dict = self.computer_partition\
.getConnectionParameterDict() .getConnectionParameterDict()
result = requests.get(connection_parameter_dict['test-machine1-url'], verify=False) result = requests.get(
connection_parameter_dict['test-machine1-url'], verify=False)
self.assertEqual( self.assertEqual(
httplib.OK, httplib.OK,
result.status_code result.status_code
) )
self.assertIn('<title>noVNC</title>', result.text) self.assertIn('<title>noVNC</title>', result.text)
result = requests.get(connection_parameter_dict['test-machine2-url'], verify=False) result = requests.get(
connection_parameter_dict['test-machine2-url'], verify=False)
self.assertEqual( self.assertEqual(
httplib.OK, httplib.OK,
result.status_code result.status_code
) )
self.assertIn('<title>noVNC</title>', result.text) self.assertIn('<title>noVNC</title>', result.text)
@skipIfPython3 @skipIfPython3
@skipUnlessKvm @skipUnlessKvm
class TestInstanceResilient(InstanceTestCase): class TestInstanceResilient(InstanceTestCase):
...@@ -395,6 +412,7 @@ class TestInstanceResilient(InstanceTestCase): ...@@ -395,6 +412,7 @@ class TestInstanceResilient(InstanceTestCase):
'takeover-kvm-1-url', 'takeover-kvm-1-url',
'url'])) 'url']))
@skipIfPython3 @skipIfPython3
@skipUnlessKvm @skipUnlessKvm
class TestAccessResilientAdditional(InstanceTestCase): class TestAccessResilientAdditional(InstanceTestCase):
...@@ -430,6 +448,7 @@ class TestAccessResilientAdditional(InstanceTestCase): ...@@ -430,6 +448,7 @@ class TestAccessResilientAdditional(InstanceTestCase):
) )
self.assertIn('<title>noVNC</title>', result.text) self.assertIn('<title>noVNC</title>', result.text)
class TestInstanceNbdServer(InstanceTestCase): class TestInstanceNbdServer(InstanceTestCase):
__partition_reference__ = 'ins' __partition_reference__ = 'ins'
instance_max_retry = 5 instance_max_retry = 5
...@@ -448,7 +467,8 @@ class TestInstanceNbdServer(InstanceTestCase): ...@@ -448,7 +467,8 @@ class TestInstanceNbdServer(InstanceTestCase):
def test(self): def test(self):
connection_parameter_dict = self.computer_partition\ connection_parameter_dict = self.computer_partition\
.getConnectionParameterDict() .getConnectionParameterDict()
result = requests.get(connection_parameter_dict['upload_url'].strip(), verify=False) result = requests.get(
connection_parameter_dict['upload_url'].strip(), verify=False)
self.assertEqual( self.assertEqual(
httplib.OK, httplib.OK,
result.status_code result.status_code
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment