def test_is_absent_in_load_balancer_error(self):
with mock.patch('requests.get', self.mockRequestsInternalServerError):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
self.assertRaises(
Exception,
manager._is_absent_in_load_balancer,
)
def test_register_port_success(self):
with mock.patch('requests.post',
self.mockRequestsPostRegisterPortWithLoadBalancer):
with mock.patch(self.TARGET_WAIT_LB_STATUS,
mock.MagicMock(return_value=True)):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
manager._register_port()
self.assertEqual(True, manager.changed)
def test_create_loadbalancer_success(self):
with mock.patch('requests.post',
self.mockRequestsPostCreateLoadBalancer):
with mock.patch(self.TARGET_WAIT_LB_STATUS,
mock.MagicMock(return_value=True)):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
manager._create_load_balancer()
self.assertEqual(True, manager.changed)
def test_deregister_instances_internal_error(self):
with mock.patch(
'requests.post', self.
mockRequestsPostDeregisterInstancesFromLoadBalancer): # noqa
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
self.assertRaises(
Exception,
manager._deregister_instances,
)
def test_sync_ssl_policy_no_change(self):
with mock.patch('requests.post',
self.mockRequestsPostConfigureHealthCheck):
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
manager._sync_ssl_policy()
self.assertEqual(False, manager.changed)
def test_sync_filter_no_change(self):
with mock.patch('requests.post',
self.mockRequestsPostSetFilterForLoadBalancer):
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
manager._sync_filter()
self.assertEqual(False, manager.changed)
def test_register_port_internal_error(self):
with mock.patch('requests.post', self.mockRequestsInternalServerError):
with mock.patch(self.TARGET_WAIT_LB_STATUS,
mock.MagicMock(return_value=False)):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
self.assertRaises(
Exception,
manager._register_port,
)
def test_sync_instances_no_change(self):
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
with mock.patch(self.TARGET_REGISTER_INSTANCES, self.mockEmpty):
with mock.patch(self.TARGET_DEREGISTER_INSTANCES,
self.mockEmpty):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
manager._sync_instances()
self.assertEqual(False, manager.changed)
def test_register_port_wait_failed(self):
with mock.patch('requests.post',
self.mockRequestsPostRegisterPortWithLoadBalancer):
with mock.patch(self.TARGET_WAIT_LB_STATUS,
mock.MagicMock(return_value=False)):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
self.assertRaises(
Exception,
manager._register_port,
)
def test_create_loadbalancer_wait_failed(self):
with mock.patch('requests.post',
self.mockRequestsPostCreateLoadBalancer):
with mock.patch(self.TARGET_WAIT_LB_STATUS,
mock.MagicMock(return_value=False)):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
self.assertRaises(
Exception,
manager._create_load_balancer,
)
def test_describe_load_balancers(self):
with mock.patch('requests.get',
self.mockRequestsGetDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
info = manager._describe_load_balancers(dict())
self.assertEqual(info['status'], 200)
self.assertEqual(info['xml_namespace'], dict(nc=self.xmlnamespace))
self.assertEqual(
etree.tostring(info['xml_body']),
etree.tostring(etree.fromstring(
self.xml['describeLoadBalancers'])))
def test_sync_filter_change_type(self):
mockModule = mock.MagicMock(
params=copy.deepcopy(self.mockModule.params),
fail_json=self.mockModule.fail_json,
check_mode=False,
)
mockModule.params['filter_type'] = 2
with mock.patch('requests.post',
self.mockRequestsPostSetFilterForLoadBalancer):
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(mockModule)
manager._sync_filter()
self.assertEqual(True, manager.changed)
def test_sync_ssl_policy_changed(self):
mockModule = mock.MagicMock(
params=copy.deepcopy(self.mockModule.params),
fail_json=self.mockModule.fail_json,
check_mode=False,
)
mockModule.params['ssl_policy_name'] = 'Standard Ciphers A ver1'
with mock.patch('requests.post',
self.mockRequestsPostConfigureHealthCheck):
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(mockModule)
manager._sync_ssl_policy()
self.assertEqual(True, manager.changed)
def test_sync_ssl_policy_internal_error(self):
mockModule = mock.MagicMock(
params=copy.deepcopy(self.mockModule.params),
fail_json=self.mockModule.fail_json,
check_mode=False,
)
mockModule.params['ssl_policy_name'] = 'Standard Ciphers A ver1'
with mock.patch('requests.post', self.mockRequestsInternalServerError):
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(mockModule)
self.assertRaises(
Exception,
manager._sync_ssl_policy,
)
def test_sync_health_check_internal_error(self):
mockModule = mock.MagicMock(
params=copy.deepcopy(self.mockModule.params),
fail_json=self.mockModule.fail_json,
check_mode=False,
)
mockModule.params['health_check_target'] = 'ICMP'
with mock.patch('requests.post', self.mockRequestsInternalServerError):
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(mockModule)
self.assertRaises(
Exception,
manager._sync_health_check,
)
def test_sync_health_check_changed(self):
mockModule = mock.MagicMock(
params=copy.deepcopy(self.mockModule.params),
fail_json=self.mockModule.fail_json,
check_mode=False,
)
mockModule.params['health_check_target'] = 'ICMP'
mockModule.params['health_check_interval'] = 5
mockModule.params['health_check_unhealthy_threshold'] = 10
with mock.patch('requests.post',
self.mockRequestsPostConfigureHealthCheck):
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(mockModule)
manager._sync_health_check()
self.assertEqual(True, manager.changed)
def test_sync_filter_internal_error(self):
mockModule = mock.MagicMock(
params=copy.deepcopy(self.mockModule.params),
fail_json=self.mockModule.fail_json,
check_mode=False,
)
addresses = ['192.168.0.1']
mockModule.params['filter_ip_addresses'] = addresses
with mock.patch('requests.post', self.mockRequestsInternalServerError):
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(mockModule)
self.assertRaises(
Exception,
manager._sync_filter,
)
def test_sync_instances_deregister_instance(self):
mockModule = mock.MagicMock(
params=copy.deepcopy(self.mockModule.params),
fail_json=self.mockModule.fail_json,
check_mode=False,
)
instance_ids = []
mockModule.params['instance_ids'] = instance_ids
with mock.patch(self.TARGET_DESCRIBE_CURRENT,
self.mockDescribeLoadBalancers):
with mock.patch(self.TARGET_REGISTER_INSTANCES, self.mockEmpty):
with mock.patch(
'requests.post', self.
mockRequestsPostDeregisterInstancesFromLoadBalancer
): # noqa
manager = nifcloud_lb.LoadBalancerManager(mockModule)
manager._sync_instances()
self.assertEqual(True, manager.changed)
def test_get_state_instance_in_load_balancer_port_not_found(self):
with mock.patch('requests.get',
self.mockRequestsGetDescribeLoadBalancersPortNotFound):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
self.assertEqual('port-not-found',
manager._get_state_instance_in_load_balancer())
def test_is_absent_in_load_balancer_absent(self):
with mock.patch('requests.get',
self.mockRequestsGetDescribeLoadBalancersNameNotFound):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
self.assertEqual(True, manager._is_absent_in_load_balancer())
def test_get_state_instance_in_load_balancer_present(self):
with mock.patch('requests.get',
self.mockRequestsGetDescribeLoadBalancers):
manager = nifcloud_lb.LoadBalancerManager(self.mockModule)
self.assertEqual('present',
manager._get_state_instance_in_load_balancer())
