def deploy(job): job.set_status(JobStatus.running) project_attributes = job.get_project() new_instances = ec2.create_and_start_instances(project_attributes) if ec2.is_running(new_instances, project_attributes): if project_attributes.get('environment_tier') == EnvironmentTier.WORKER.value: old_instances = ec2.get_instances(filters={'tag:Project': project_attributes['name']}) old_instance_ids = list(set([instance.id for instance in old_instances]) - set(new_instances)) else: load_balancer = loadbalancer.get_loadbalancer(project_attributes) old_instance_ids = [instance.id for instance in load_balancer.instances] loadbalancer.attach(load_balancer, new_instances) loadbalancer.dettach(load_balancer, old_instance_ids) ec2.terminate(old_instance_ids) job.set_status(JobStatus.done) else: ec2.terminate(new_instances) job.set_status(JobStatus.failed)
def test_we_return_true_if_terminate_succeed(self, mock_landlord, mock_ec2): properties = {'region': 'myRegion', 'environment': 'STAGE', 'domain': 'this.is.awesome'} instances = ['i-278219', 'i-82715'] mock_landlord.Tenant = StubLandlord mock_connection = Mock() mock_ec2.connect_to_region.return_value = mock_connection mock_connection.terminate_instances.return_value = ['i-278219', 'i-82715'] result = ec2.terminate(instances) mock_ec2.connect_to_region.assert_called_with('deploy.region', aws_access_key_id='aws.id', aws_secret_access_key='aws.secret') mock_connection.terminate_instances.assert_called_with(instances) self.assertEquals(True, result)
def test_we_return_false_when_we_pass_an_empty_array(self): properties = {'region': 'myRegion', 'environment': 'STAGE', 'domain': 'this.is.awesome'} self.assertEquals(False, ec2.terminate([]))