mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
migrate test_haproxyconn to pytest
This commit is contained in:
Frode Gundersen
Megan Wilhite
parent
cc669ee27f
commit
a95cb8ca86
2 changed files with 214 additions and 199 deletions
214
tests/pytests/unit/modules/test_haproxyconn.py
Normal file
214
tests/pytests/unit/modules/test_haproxyconn.py
Normal file
|
|
@ -0,0 +1,214 @@
|
|||
"""
|
||||
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
|
||||
|
||||
Test cases for salt.modules.haproxyconn
|
||||
"""
|
||||
|
||||
|
||||
import pytest
|
||||
|
||||
import salt.modules.haproxyconn as haproxyconn
|
||||
|
||||
|
||||
class Mockcmds:
|
||||
"""
|
||||
Mock of cmds
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.backend = None
|
||||
self.server = None
|
||||
self.weight = None
|
||||
|
||||
def listServers(self, backend):
|
||||
"""
|
||||
Mock of listServers method
|
||||
"""
|
||||
self.backend = backend
|
||||
return (
|
||||
"Name: server01 Status: UP Weight: 1 bIn: 22 bOut: 12\n"
|
||||
"Name: server02 Status: MAINT Weight: 2 bIn: 0 bOut: 0"
|
||||
)
|
||||
|
||||
def enableServer(self, server, backend):
|
||||
"""
|
||||
Mock of enableServer method
|
||||
"""
|
||||
self.backend = backend
|
||||
self.server = server
|
||||
return "server enabled"
|
||||
|
||||
def disableServer(self, server, backend):
|
||||
"""
|
||||
Mock of disableServer method
|
||||
"""
|
||||
self.backend = backend
|
||||
self.server = server
|
||||
return "server disabled"
|
||||
|
||||
def getWeight(self, server, backend, weight=0):
|
||||
"""
|
||||
Mock of getWeight method
|
||||
"""
|
||||
self.backend = backend
|
||||
self.server = server
|
||||
self.weight = weight
|
||||
return "server weight"
|
||||
|
||||
@staticmethod
|
||||
def showFrontends():
|
||||
"""
|
||||
Mock of showFrontends method
|
||||
"""
|
||||
return "frontend-alpha\nfrontend-beta\nfrontend-gamma"
|
||||
|
||||
@staticmethod
|
||||
def showBackends():
|
||||
"""
|
||||
Mock of showBackends method
|
||||
"""
|
||||
return "backend-alpha\nbackend-beta\nbackend-gamma"
|
||||
|
||||
|
||||
class Mockhaproxy:
|
||||
"""
|
||||
Mock of haproxy
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.cmds = Mockcmds()
|
||||
|
||||
|
||||
class MockHaConn:
|
||||
"""
|
||||
Mock of HaConn
|
||||
"""
|
||||
|
||||
def __init__(self, socket=None):
|
||||
self.ha_cmd = None
|
||||
|
||||
def sendCmd(self, ha_cmd, objectify=False):
|
||||
"""
|
||||
Mock of sendCmd method
|
||||
"""
|
||||
self.ha_cmd = ha_cmd
|
||||
self.objectify = objectify
|
||||
return ha_cmd
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def configure_loader_modules():
|
||||
return {haproxyconn: {"haproxy": Mockhaproxy(), "_get_conn": MockHaConn}}
|
||||
|
||||
|
||||
# 'list_servers' function tests: 1
|
||||
|
||||
|
||||
def test_list_servers():
|
||||
"""
|
||||
Test list_servers
|
||||
"""
|
||||
assert haproxyconn.list_servers("mysql")
|
||||
|
||||
|
||||
# 'enable_server' function tests: 1
|
||||
|
||||
|
||||
def test_enable_server():
|
||||
"""
|
||||
Test enable_server
|
||||
"""
|
||||
assert haproxyconn.enable_server("web1.salt.com", "www")
|
||||
|
||||
|
||||
# 'disable_server' function tests: 1
|
||||
|
||||
|
||||
def test_disable_server():
|
||||
"""
|
||||
Test disable_server
|
||||
"""
|
||||
assert haproxyconn.disable_server("db1.salt.com", "mysql")
|
||||
|
||||
|
||||
# 'get_weight' function tests: 1
|
||||
|
||||
|
||||
def test_get_weight():
|
||||
"""
|
||||
Test get the weight of a server
|
||||
"""
|
||||
assert haproxyconn.get_weight("db1.salt.com", "mysql")
|
||||
|
||||
|
||||
# 'set_weight' function tests: 1
|
||||
|
||||
|
||||
def test_set_weight():
|
||||
"""
|
||||
Test setting the weight of a given server
|
||||
"""
|
||||
assert haproxyconn.set_weight("db1.salt.com", "mysql", weight=11)
|
||||
|
||||
|
||||
# 'show_frontends' function tests: 1
|
||||
|
||||
|
||||
def test_show_frontends():
|
||||
"""
|
||||
Test print all frontends received from the HAProxy socket
|
||||
"""
|
||||
assert haproxyconn.show_frontends()
|
||||
|
||||
|
||||
def test_list_frontends():
|
||||
"""
|
||||
Test listing all frontends
|
||||
"""
|
||||
assert sorted(haproxyconn.list_frontends()) == sorted(
|
||||
["frontend-alpha", "frontend-beta", "frontend-gamma"]
|
||||
)
|
||||
|
||||
|
||||
# 'show_backends' function tests: 1
|
||||
|
||||
|
||||
def test_show_backends():
|
||||
"""
|
||||
Test print all backends received from the HAProxy socket
|
||||
"""
|
||||
assert haproxyconn.show_backends()
|
||||
|
||||
|
||||
def test_list_backends():
|
||||
"""
|
||||
Test listing of all backends
|
||||
"""
|
||||
assert sorted(haproxyconn.list_backends()) == sorted(
|
||||
["backend-alpha", "backend-beta", "backend-gamma"]
|
||||
)
|
||||
|
||||
|
||||
def test_get_backend():
|
||||
"""
|
||||
Test get_backend and compare returned value
|
||||
"""
|
||||
expected_data = {
|
||||
"server01": {"status": "UP", "weight": 1, "bin": 22, "bout": 12},
|
||||
"server02": {"status": "MAINT", "weight": 2, "bin": 0, "bout": 0},
|
||||
}
|
||||
assert haproxyconn.get_backend("test") == expected_data
|
||||
|
||||
|
||||
def test_wait_state_true():
|
||||
"""
|
||||
Test a successful wait for state
|
||||
"""
|
||||
assert haproxyconn.wait_state("test", "server01")
|
||||
|
||||
|
||||
def test_wait_state_false():
|
||||
"""
|
||||
Test a failed wait for state, with a timeout of 0
|
||||
"""
|
||||
assert not haproxyconn.wait_state("test", "server02", "up", 0)
|
||||
|
|
@ -1,199 +0,0 @@
|
|||
"""
|
||||
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
|
||||
"""
|
||||
|
||||
|
||||
import salt.modules.haproxyconn as haproxyconn
|
||||
from tests.support.mixins import LoaderModuleMockMixin
|
||||
from tests.support.unit import TestCase
|
||||
|
||||
|
||||
class Mockcmds:
|
||||
"""
|
||||
Mock of cmds
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.backend = None
|
||||
self.server = None
|
||||
self.weight = None
|
||||
|
||||
def listServers(self, backend):
|
||||
"""
|
||||
Mock of listServers method
|
||||
"""
|
||||
self.backend = backend
|
||||
return (
|
||||
"Name: server01 Status: UP Weight: 1 bIn: 22 bOut: 12\n"
|
||||
"Name: server02 Status: MAINT Weight: 2 bIn: 0 bOut: 0"
|
||||
)
|
||||
|
||||
def enableServer(self, server, backend):
|
||||
"""
|
||||
Mock of enableServer method
|
||||
"""
|
||||
self.backend = backend
|
||||
self.server = server
|
||||
return "server enabled"
|
||||
|
||||
def disableServer(self, server, backend):
|
||||
"""
|
||||
Mock of disableServer method
|
||||
"""
|
||||
self.backend = backend
|
||||
self.server = server
|
||||
return "server disabled"
|
||||
|
||||
def getWeight(self, server, backend, weight=0):
|
||||
"""
|
||||
Mock of getWeight method
|
||||
"""
|
||||
self.backend = backend
|
||||
self.server = server
|
||||
self.weight = weight
|
||||
return "server weight"
|
||||
|
||||
@staticmethod
|
||||
def showFrontends():
|
||||
"""
|
||||
Mock of showFrontends method
|
||||
"""
|
||||
return "frontend-alpha\nfrontend-beta\nfrontend-gamma"
|
||||
|
||||
@staticmethod
|
||||
def showBackends():
|
||||
"""
|
||||
Mock of showBackends method
|
||||
"""
|
||||
return "backend-alpha\nbackend-beta\nbackend-gamma"
|
||||
|
||||
|
||||
class Mockhaproxy:
|
||||
"""
|
||||
Mock of haproxy
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.cmds = Mockcmds()
|
||||
|
||||
|
||||
class MockHaConn:
|
||||
"""
|
||||
Mock of HaConn
|
||||
"""
|
||||
|
||||
def __init__(self, socket=None):
|
||||
self.ha_cmd = None
|
||||
|
||||
def sendCmd(self, ha_cmd, objectify=False):
|
||||
"""
|
||||
Mock of sendCmd method
|
||||
"""
|
||||
self.ha_cmd = ha_cmd
|
||||
self.objectify = objectify
|
||||
return ha_cmd
|
||||
|
||||
|
||||
class HaproxyConnTestCase(TestCase, LoaderModuleMockMixin):
|
||||
"""
|
||||
Test cases for salt.modules.haproxyconn
|
||||
"""
|
||||
|
||||
def setup_loader_modules(self):
|
||||
return {haproxyconn: {"haproxy": Mockhaproxy(), "_get_conn": MockHaConn}}
|
||||
|
||||
# 'list_servers' function tests: 1
|
||||
|
||||
def test_list_servers(self):
|
||||
"""
|
||||
Test list_servers
|
||||
"""
|
||||
self.assertTrue(haproxyconn.list_servers("mysql"))
|
||||
|
||||
# 'enable_server' function tests: 1
|
||||
|
||||
def test_enable_server(self):
|
||||
"""
|
||||
Test enable_server
|
||||
"""
|
||||
self.assertTrue(haproxyconn.enable_server("web1.salt.com", "www"))
|
||||
|
||||
# 'disable_server' function tests: 1
|
||||
|
||||
def test_disable_server(self):
|
||||
"""
|
||||
Test disable_server
|
||||
"""
|
||||
self.assertTrue(haproxyconn.disable_server("db1.salt.com", "mysql"))
|
||||
|
||||
# 'get_weight' function tests: 1
|
||||
|
||||
def test_get_weight(self):
|
||||
"""
|
||||
Test get the weight of a server
|
||||
"""
|
||||
self.assertTrue(haproxyconn.get_weight("db1.salt.com", "mysql"))
|
||||
|
||||
# 'set_weight' function tests: 1
|
||||
|
||||
def test_set_weight(self):
|
||||
"""
|
||||
Test setting the weight of a given server
|
||||
"""
|
||||
self.assertTrue(haproxyconn.set_weight("db1.salt.com", "mysql", weight=11))
|
||||
|
||||
# 'show_frontends' function tests: 1
|
||||
|
||||
def test_show_frontends(self):
|
||||
"""
|
||||
Test print all frontends received from the HAProxy socket
|
||||
"""
|
||||
self.assertTrue(haproxyconn.show_frontends())
|
||||
|
||||
def test_list_frontends(self):
|
||||
"""
|
||||
Test listing all frontends
|
||||
"""
|
||||
self.assertEqual(
|
||||
sorted(haproxyconn.list_frontends()),
|
||||
sorted(["frontend-alpha", "frontend-beta", "frontend-gamma"]),
|
||||
)
|
||||
|
||||
# 'show_backends' function tests: 1
|
||||
|
||||
def test_show_backends(self):
|
||||
"""
|
||||
Test print all backends received from the HAProxy socket
|
||||
"""
|
||||
self.assertTrue(haproxyconn.show_backends())
|
||||
|
||||
def test_list_backends(self):
|
||||
"""
|
||||
Test listing of all backends
|
||||
"""
|
||||
self.assertEqual(
|
||||
sorted(haproxyconn.list_backends()),
|
||||
sorted(["backend-alpha", "backend-beta", "backend-gamma"]),
|
||||
)
|
||||
|
||||
def test_get_backend(self):
|
||||
"""
|
||||
Test get_backend and compare returned value
|
||||
"""
|
||||
expected_data = {
|
||||
"server01": {"status": "UP", "weight": 1, "bin": 22, "bout": 12},
|
||||
"server02": {"status": "MAINT", "weight": 2, "bin": 0, "bout": 0},
|
||||
}
|
||||
self.assertDictEqual(haproxyconn.get_backend("test"), expected_data)
|
||||
|
||||
def test_wait_state_true(self):
|
||||
"""
|
||||
Test a successful wait for state
|
||||
"""
|
||||
self.assertTrue(haproxyconn.wait_state("test", "server01"))
|
||||
|
||||
def test_wait_state_false(self):
|
||||
"""
|
||||
Test a failed wait for state, with a timeout of 0
|
||||
"""
|
||||
self.assertFalse(haproxyconn.wait_state("test", "server02", "up", 0))
|
||||
Loading…
Add table
Reference in a new issue