def setUp(self, mock_server, mock_plugin):
super(PluribusLoadBalancerTestCase, self).setUp()
self.service = PluribusLoadBalancerDriver(mock_plugin)
self.service.server = mock_server
self.pools['pool_id'] = self.pool_id
self.tenant_id = "tenant-1"
self.context = FakeContext()
class PluribusLoadBalancerTestCase(base.BaseTestCase,
Vip,
Pool,
Member,
HealthMonitor):
"""Test cases for Pluribus LoadBalancer"""
@mock.patch('oslo.utils.importutils.import_object')
@mock.patch('neutron.services.loadbalancer.drivers.common.'
'agent_driver_base.LoadBalancerCallbacks')
def setUp(self, mock_server, mock_plugin):
super(PluribusLoadBalancerTestCase, self).setUp()
self.service = PluribusLoadBalancerDriver(mock_plugin)
self.service.server = mock_server
self.pools['pool_id'] = self.pool_id
self.tenant_id = "tenant-1"
self.context = FakeContext()
def _create_pool_dict(self):
pool = {
'lb_method': self.lb_method,
'protocol': self.protocol,
'name': self.name,
'subnet_id': self.subnet_id,
'id': self.pool_id,
'provider': self.provider
}
return pool
def _create_member_dict(self):
member = {
'weight': self.weight,
'address': self.address,
'protocol_port': self.protocol_port,
'id': self.member_id
}
return member
def _create_vip_dict(self):
vip = {
'protocol': self.protocol,
'address': self.address,
'protocol_port': self.protocol_port,
'id': self.vip_id,
'vip_name': self.vip_name,
'subnet_id': self.subnet_id,
'connection_limit': self.connection_limit,
'pool_id': self.pool_id
}
return vip
def _create_health_monitor(self):
hm = {
'delay': self.delay,
'max_retries': self.max_retries,
'timeout': self.timeout,
'pools': self.pools,
'type': self.type,
'id': self.healthmonitor_id
}
return hm
def test_create_pool(self):
pool = self._create_pool_dict()
self.service.create_pool(self.context, pool)
self.service.server.create_pool.assert_called_once_with(
lb_method=self.lb_method,
protocol=self.protocol,
name=self.name,
subnet_id=self.subnet_id,
id=self.pool_id,
provider=self.provider
)
def test_delete_pool(self):
pool = self._create_pool_dict()
self.service.delete_pool(self.context, pool)
self.service.server.delete_pool.assert_called_once_with(
lb_method=self.lb_method,
protocol=self.protocol,
name=self.name,
subnet_id=self.subnet_id,
id=self.pool_id,
provider=self.provider
)
def test_create_member(self):
member = self._create_member_dict()
self.service.create_member(self.context, member)
self.service.server.create_member.assert_called_once_with(
weight=self.weight,
address=self.address,
protocol_port=self.protocol_port,
id=self.member_id
)
def test_delete_member(self):
member = self._create_member_dict()
self.service.delete_member(self.context, member)
self.service.server.delete_member.assert_called_once_with(
weight=self.weight,
address=self.address,
protocol_port=self.protocol_port,
id=self.member_id
)
def test_create_vip(self):
vip = self._create_vip_dict()
self.service.create_vip(self.context, vip)
self.service.server.create_vip.assert_called_once_with(
protocol=self.protocol,
address=self.address,
protocol_port=self.protocol_port,
id=self.vip_id,
vip_name=self.vip_name,
subnet_id=self.subnet_id,
connection_limit=self.connection_limit,
pool_id=self.pool_id
)
def test_delete_vip(self):
vip = self._create_vip_dict()
self.service.delete_vip(self.context, vip)
self.service.server.delete_vip.assert_called_once_with(
protocol=self.protocol,
address=self.address,
protocol_port=self.protocol_port,
id=self.vip_id,
vip_name=self.vip_name,
subnet_id=self.subnet_id,
connection_limit=self.connection_limit,
pool_id=self.pool_id
)
def test_create_pool_health_monitor(self):
hm = self._create_health_monitor()
self.service.create_pool_health_monitor(self.context, hm,
self.pool_id)
self.service.server.create_health.assert_called_once_with(
delay=self.delay,
max_retries=self.max_retries,
timeout=self.timeout,
pools=self.pools,
type=self.type,
id=self.healthmonitor_id,
)
def test_delete_pool_health_monitor(self):
hm = self._create_health_monitor()
self.service.delete_pool_health_monitor(self.context, hm,
self.pool_id)
self.service.server.delete_health.assert_called_once_with(
delay=self.delay,
max_retries=self.max_retries,
timeout=self.timeout,
pools=self.pools,
type=self.type,
id=self.healthmonitor_id,
)
