def test_start_agent_elb(self, resource, kms, dynamo, elb, repeat, manager, agent_db, tables, get_instance_id): get_instance_id.return_value = 'i-123456' start_agent(ENVIRONMENT, SERVICE, REGION, ELB, 0.1, 0.1) elb.assert_called_with(REGION)
def test_start_agent_messaging(self, resource, kms, dynamo, elb, repeat, manager, agent_db, tables, get_instance_id, get_queue): get_instance_id.return_value = 'i-123456' get_queue.return_value = MagicMock() start_agent(ENVIRONMENT, SERVICE, REGION, ELB, 0.1, 0.1) self.assertEquals(3, repeat.call_count)
def test_start_agent_no_elb(self, resource, kms, dynamo, elb, repeat, manager, tables, get_instance_id, get_queue): get_queue.return_value = None get_instance_id.return_value = 'i-123456' start_agent(ENVIRONMENT, SERVICE, REGION, None, 0.1, 0.1) dynamo.assert_called_with(REGION) kms.assert_called_with(REGION) elb.assert_not_called() self.assertEquals(2, repeat.call_count)