def test_remove_exception(self): actor = elbv2_actor.DeregisterInstance('UTA', { 'target_group': 'test', 'region': 'us-east-1', 'instances': 'test'}) actor.elbv2_conn = mock.Mock() exc = botocore.exceptions.ClientError({'Error': {'Code': ''}}, 'Test') actor.elbv2_conn.deregister_targets.side_effect = exc with self.assertRaises(exceptions.UnrecoverableActorFailure): yield actor._remove('target_group_arn', ['i-un173s7'])
def test_execute_self(self): actor = elbv2_actor.DeregisterInstance('UTA', { 'target_group': 'elb-test', 'region': 'us-east-1'}) actor._find_target_group = mock.Mock() actor._find_target_group.return_value = helper.tornado_value('arn') actor._get_meta_data = helper.mock_tornado('i-test') actor._remove = mock.Mock() actor._remove.return_value = helper.tornado_value(mock.Mock()) yield actor._execute() actor._remove.assert_called_with('arn', ['i-test'])
def test_remove(self): actor = elbv2_actor.DeregisterInstance('UTA', { 'target_group': 'test', 'region': 'us-east-1', 'instances': 'test'}) actor.elbv2_conn = mock.Mock() actor.elbv2_conn.deregister_targets.return_value = {} yield actor._remove('target_group_arn', ['i-un173s7']) actor.elbv2_conn.deregister_targets.assert_called_with( TargetGroupArn='target_group_arn', Targets=[{'Id': 'i-un173s7'}])