def test_register_instance_from_elb():
cloud_connection = MagicMock()
connection = MagicMock()
cloud_connection.get_connection.return_value = connection
with mock.patch.object(
AwsClbManager,
'list_lbs_from_autoscale',
new=lambda a, b, c: [
get_aws_data('elb--describe-load-balancers')[
"LoadBalancerDescriptions"][0]["LoadBalancerName"]
]):
clb_manager = AwsClbManager(cloud_connection, 'region')
ret = clb_manager.register_instances_from_lbs('test-asg',
['id0', 'id1'], LOG_FILE)
assert ret == True
connection.register_instances.assert_called_once_with(
'test-elb', ['id0', 'id1'])
def test_register_instance_from_elb():
cloud_connection = MagicMock()
connection = MagicMock()
cloud_connection.get_connection.return_value = connection
ret = AwsClbManager(cloud_connection,
'region').register_instances_from_lbs(['test-elb'],
['id0', 'id1'],
LOG_FILE)
assert ret == True
connection.register_instances.assert_called_once_with(
'test-elb', ['id0', 'id1'])
def test_copy_elb():
cloud_connection = MagicMock()
connection = MagicMock()
cloud_connection.get_connection.return_value = connection
def describe_load_balancers(LoadBalancerNames=None):
if LoadBalancerNames == ['test-elb']:
return get_aws_data('elb--describe-load-balancers')
if LoadBalancerNames == ['copied_elb']:
return None
raise Exception(
'describe_load_balancers : Invalid parameter LoadBalancerNames')
connection.describe_load_balancers.side_effect = describe_load_balancers
connection.describe_tags.return_value = get_aws_data('elb--describe-tags')
connection.describe_load_balancer_attributes.return_value = get_aws_data(
'elb--describe-load-balancers-attributes')
connection.create_load_balancer.return_value.__getitem__.side_effect = {
"DNSName": "test-dns"
}.__getitem__
dns = AwsClbManager(cloud_connection,
'region').copy_lb('copied_elb', 'test-elb',
{'foo': 'bar'}, LOG_FILE)
assert dns == "test-dns"
assert connection.describe_load_balancers.call_count == 2
connection.describe_tags.assert_called_once_with(
LoadBalancerNames=['test-elb'])
connection.describe_load_balancer_attributes.assert_called_once_with(
LoadBalancerName='test-elb')
connection.create_load_balancer.assert_called_once_with(
LoadBalancerName='copied_elb',
Listeners=[{
"LoadBalancerPort": 80,
"Protocol": "HTTP",
"InstancePort": 80,
"InstanceProtocol": "HTTP"
}, {
"LoadBalancerPort":
443,
"Protocol":
"HTTPS",
"InstancePort":
80,
"InstanceProtocol":
"HTTP",
"SSLCertificateId":
"arn:aws:iam::0123456789:server-certificate/test-certificate"
}],
Subnets=["subnet-test"],
SecurityGroups=["sg-test"],
Scheme="internet-facing",
Tags=[
{
"Key": "test-key",
"Value": "test-value"
},
{
"Key": "foo",
"Value": "bar"
},
])
connection.configure_health_check.assert_called_once_with(
LoadBalancerName='copied_elb',
HealthCheck={
"Interval": 20,
"Timeout": 6,
"Target": "TCP:80",
"UnhealthyThreshold": 4,
"HealthyThreshold": 2
})
connection.modify_load_balancer_attributes.assert_called_once_with(
LoadBalancerName='copied_elb',
LoadBalancerAttributes={
"AccessLog": {
"Enabled": False
},
"CrossZoneLoadBalancing": {
"Enabled": True
},
"ConnectionDraining": {
"Enabled": True,
"Timeout": 400
},
"ConnectionSettings": {
"IdleTimeout": 400
}
})