d54272bf57
This change allows for defining a load balancer supporting multiple simultaneous protocols instead of one LB per protocol. Redundant worker tests are also removed. Change-Id: I9ef13771af7d0513997c675374fc171d515d4b43
60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
import unittest
|
|
from libra.worker.drivers.haproxy.driver import HAProxyDriver
|
|
|
|
|
|
class TestHAProxyDriver(unittest.TestCase):
|
|
def setUp(self):
|
|
self.driver = HAProxyDriver('mock_objects.FakeOSServices')
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def testInit(self):
|
|
""" Test the HAProxy init() method """
|
|
self.driver.init()
|
|
self.assertIsInstance(self.driver._config, dict)
|
|
|
|
def testAddProtocol(self):
|
|
""" Test the HAProxy set_protocol() method """
|
|
proto = 'http'
|
|
self.driver.add_protocol(proto, None)
|
|
self.assertIn(proto, self.driver._config)
|
|
self.assertEqual(self.driver._config[proto]['bind_address'], '0.0.0.0')
|
|
self.assertEqual(self.driver._config[proto]['bind_port'], 80)
|
|
|
|
proto = 'tcp'
|
|
self.driver.add_protocol(proto, 443)
|
|
self.assertIn(proto, self.driver._config)
|
|
self.assertEqual(self.driver._config[proto]['bind_address'], '0.0.0.0')
|
|
self.assertEqual(self.driver._config[proto]['bind_port'], 443)
|
|
|
|
def testAddTCPRequiresPort(self):
|
|
with self.assertRaises(Exception):
|
|
self.driver.add_protocol('tcp', None)
|
|
|
|
def testAddServer(self):
|
|
""" Test the HAProxy add_server() method """
|
|
proto = 'http'
|
|
self.driver.add_protocol(proto, None)
|
|
self.driver.add_server(proto, '1.2.3.4', 7777)
|
|
self.driver.add_server(proto, '5.6.7.8', 8888)
|
|
self.assertIn(proto, self.driver._config)
|
|
self.assertIn('servers', self.driver._config[proto])
|
|
servers = self.driver._config[proto]['servers']
|
|
self.assertEqual(len(servers), 2)
|
|
self.assertEqual(servers[0], ('1.2.3.4', 7777))
|
|
self.assertEqual(servers[1], ('5.6.7.8', 8888))
|
|
|
|
def testSetAlgorithm(self):
|
|
""" Test the HAProxy set_algorithm() method """
|
|
proto = 'http'
|
|
self.driver.add_protocol(proto, None)
|
|
self.driver.set_algorithm(proto, self.driver.ROUNDROBIN)
|
|
self.assertIn(proto, self.driver._config)
|
|
self.assertIn('algorithm', self.driver._config[proto])
|
|
self.assertEqual(self.driver._config[proto]['algorithm'], 'roundrobin')
|
|
self.driver.set_algorithm(proto, self.driver.LEASTCONN)
|
|
self.assertEqual(self.driver._config[proto]['algorithm'], 'leastconn')
|
|
with self.assertRaises(Exception):
|
|
self.driver.set_algorithm(proto, 99)
|