def test_is_instance_available_some_terminated(self):
with patch('cloudweatherreport.cloudresource.aws.make_client',
autospec=True, return_value=make_fake_client(
running_nodes=19, terminated_nodes=1)) as aws_mock:
aws = AWS('access', 'secret', 'region')
instance_available = aws.is_instance_available(1)
self.assertIs(instance_available, True)
aws_mock.assert_called_once_with('access', 'secret', 'region')
def test_is_security_group_available_max_groups(self):
with patch('cloudweatherreport.cloudresource.aws.make_client',
autospec=True, return_value=make_fake_client(
number_of_security_group=500)) as aws_mock:
aws = AWS('access', 'secret', 'region')
instance_available = aws.is_security_group_available(1)
self.assertIs(instance_available, False)
aws_mock.assert_called_once_with('access', 'secret', 'region')
def test_is_instance_available_max(self):
with patch('cloudweatherreport.cloudresource.aws.make_client',
autospec=True, return_value=make_fake_client()
) as aws_mock:
aws = AWS('access', 'secret', 'region')
instance_available = aws.is_instance_available(20)
self.assertIs(instance_available, True)
aws_mock.assert_called_once_with('access', 'secret', 'region')