def test_unauthorize_should_log_failure_when_ec2_conn_raises_exception(self):
def fail_to_authorize(*args, **kwargs):
raise EC2ResponseError(status=500, reason="I've failed, my friend")
err = self.mocker.replace("logging.error")
err("500 - I've failed, my friend")
self.mocker.result(None)
self.mocker.replay()
instance = Instance(name="semblance_of_liberty", ec2_id="i-022")
client = Client()
client._ec2_conn = mocks.FakeEC2Conn()
client._ec2_conn.revoke_security_group = fail_to_authorize
client.unauthorize(instance)
self.mocker.verify()
def test_unauthorize_should_return_False_when_ec2_conn_raises_exception(self):
def fail_to_authorize(*args, **kwargs):
raise EC2ResponseError(status=500, reason="I've failed, my friend")
instance = Instance(name="semblance_of_liberty", ec2_id="i-022")
client = Client()
client._ec2_conn = mocks.FakeEC2Conn()
client._ec2_conn.revoke_security_group = fail_to_authorize
self.assertFalse(client.unauthorize(instance))
def test_unauthorize_should_use_ec2_to_revoke_access_to_the_instance(self):
fake = mocks.FakeEC2Conn()
instance = Instance(name="tides_of_time", ec2_id="i-021")
client = Client()
client._ec2_conn = fake
authorized = client.authorize(instance)
self.assertTrue(authorized)
unauthorized = client.unauthorize(instance)
self.assertTrue(unauthorized)
authorization_string = "cidr_ip=%s/32 from_port=%s group_name=default ip_protocol=tcp to_port=%s" % (
instance.host,
instance.port,
instance.port,
)
self.assertNotIn(authorization_string, fake.authorizations)
def test_unauthorize_should_return_False_when_revoking_fail_at_ec2_conn(self):
instance = Instance()
client = Client()
client._ec2_conn = mocks.FailingEC2Conn()
self.assertFalse(client.unauthorize(instance))
