def test_next_token(self):
elb = ELBConnection(aws_access_key_id="aws_access_key_id", aws_secret_access_key="aws_secret_access_key")
mock_response = mock.Mock()
mock_response.read.return_value = DISABLE_RESPONSE
mock_response.status = 200
elb.make_request = mock.Mock(return_value=mock_response)
disabled = elb.disable_availability_zones("mine", ["sample-zone"])
self.assertEqual(disabled, ["sample-zone"])
def test_next_token(self):
elb = ELBConnection(aws_access_key_id='aws_access_key_id',
aws_secret_access_key='aws_secret_access_key')
mock_response = mock.Mock()
mock_response.read.return_value = DISABLE_RESPONSE
mock_response.status = 200
elb.make_request = mock.Mock(return_value=mock_response)
disabled = elb.disable_availability_zones('mine', ['sample-zone'])
self.assertEqual(disabled, ['sample-zone'])
def test_detach_subnets(self):
elb = ELBConnection(aws_access_key_id="aws_access_key_id", aws_secret_access_key="aws_secret_access_key")
lb = LoadBalancer(elb, "mylb")
mock_response = mock.Mock()
mock_response.read.return_value = DETACH_RESPONSE
mock_response.status = 200
elb.make_request = mock.Mock(return_value=mock_response)
lb.detach_subnets("s-xxx")
def _setup_mock(self):
"""Sets up a mock elb request.
Returns: response, elb connection and LoadBalancer
"""
mock_response = mock.Mock()
mock_response.status = 200
elb = ELBConnection(aws_access_key_id='aws_access_key_id',
aws_secret_access_key='aws_secret_access_key')
elb.make_request = mock.Mock(return_value=mock_response)
return mock_response, elb, LoadBalancer(elb, 'test_elb')
def _setup_mock(self):
"""Sets up a mock elb request.
Returns: response, elb connection and LoadBalancer
"""
mock_response = mock.Mock()
mock_response.status = 200
elb = ELBConnection(aws_access_key_id='aws_access_key_id',
aws_secret_access_key='aws_secret_access_key')
elb.make_request = mock.Mock(return_value=mock_response)
return mock_response, elb, LoadBalancer(elb, 'test_elb')
def test_detach_subnets(self):
elb = ELBConnection(aws_access_key_id='aws_access_key_id',
aws_secret_access_key='aws_secret_access_key')
lb = LoadBalancer(elb, "mylb")
mock_response = mock.Mock()
mock_response.read.return_value = DETACH_RESPONSE
mock_response.status = 200
elb.make_request = mock.Mock(return_value=mock_response)
lb.detach_subnets("s-xxx")
def test_request_with_marker(self):
elb = ELBConnection(aws_access_key_id="aws_access_key_id", aws_secret_access_key="aws_secret_access_key")
mock_response = mock.Mock()
mock_response.read.return_value = DESCRIBE_RESPONSE
mock_response.status = 200
elb.make_request = mock.Mock(return_value=mock_response)
load_balancers1 = elb.get_all_load_balancers()
self.assertEqual("1234", load_balancers1.marker)
load_balancers2 = elb.get_all_load_balancers(marker=load_balancers1.marker)
self.assertEqual(len(load_balancers2), 1)
def test_request_with_marker(self):
elb = ELBConnection(aws_access_key_id='aws_access_key_id',
aws_secret_access_key='aws_secret_access_key')
mock_response = mock.Mock()
mock_response.read.return_value = DESCRIBE_RESPONSE
mock_response.status = 200
elb.make_request = mock.Mock(return_value=mock_response)
load_balancers1 = elb.get_all_load_balancers()
self.assertEqual('1234', load_balancers1.marker)
load_balancers2 = elb.get_all_load_balancers(
marker=load_balancers1.marker)
self.assertEqual(len(load_balancers2), 1)
def test_other_policy(self):
elb = ELBConnection(aws_access_key_id="aws_access_key_id", aws_secret_access_key="aws_secret_access_key")
mock_response = mock.Mock()
mock_response.read.return_value = DESCRIBE_RESPONSE
mock_response.status = 200
elb.make_request = mock.Mock(return_value=mock_response)
load_balancers = elb.get_all_load_balancers()
self.assertEqual(len(load_balancers), 1)
lb = load_balancers[0]
self.assertEqual(len(lb.policies.other_policies), 2)
self.assertEqual(
lb.policies.other_policies[0].policy_name, "AWSConsole-SSLNegotiationPolicy-my-test-loadbalancer"
)
self.assertEqual(lb.policies.other_policies[1].policy_name, "EnableProxyProtocol")
self.assertEqual(len(lb.backends), 1)
self.assertEqual(len(lb.backends[0].policies), 1)
self.assertEqual(lb.backends[0].policies[0].policy_name, "EnableProxyProtocol")
self.assertEqual(lb.backends[0].instance_port, 80)
def test_other_policy(self):
elb = ELBConnection(aws_access_key_id='aws_access_key_id',
aws_secret_access_key='aws_secret_access_key')
mock_response = mock.Mock()
mock_response.read.return_value = DESCRIBE_RESPONSE
mock_response.status = 200
elb.make_request = mock.Mock(return_value=mock_response)
load_balancers = elb.get_all_load_balancers()
self.assertEqual(len(load_balancers), 1)
lb = load_balancers[0]
self.assertEqual(len(lb.policies.other_policies), 2)
self.assertEqual(
lb.policies.other_policies[0].policy_name,
'AWSConsole-SSLNegotiationPolicy-my-test-loadbalancer')
self.assertEqual(lb.policies.other_policies[1].policy_name,
'EnableProxyProtocol')
self.assertEqual(len(lb.backends), 1)
self.assertEqual(len(lb.backends[0].policies), 1)
self.assertEqual(lb.backends[0].policies[0].policy_name,
'EnableProxyProtocol')
self.assertEqual(lb.backends[0].instance_port, 80)
