def test_get_service_host_notfound(self):
'''Test case for ensuring that None is returned when host can not be found.'''
self.assertEqual(None, aws_config.get_service_host(None, None))
self.assertEqual(None, aws_config.get_service_host("not found", None))
self.assertEqual(None, aws_config.get_service_host("eu-west-1", None))
self.assertEqual(None, aws_config.get_service_host("eu-west-1", "not found"))
def __init__(self, region, http_client=AwsHttpClient):
self._region = region
self._sqs_service_host = aws_config.get_service_host(region, SqsClient.SQS_SERVICE_NAME)
self._request_signer = AWSRequestSignerV4(
aws_config.AWS_ACCESS_KEY, aws_config.AWS_SECRET_KEY, region, SqsClient.SQS_SERVICE_NAME
)
self._http_client = http_client(httplib2.Http)
def setUpClass(cls):
'''Method used to setup global dependencies for test cases.'''
if aws_config.AWS_ACCESS_KEY == "Put your access key in here" or \
aws_config.AWS_SECRET_KEY == "Put your secret key in here":
raise SkipTest("This will be executed only if you provide valid AWS access / secret keys.")
sqs_client = SqsClient(cls.QUEUE_REGION)
cls.QUEUE_URL = sqs_client.get_queue_url(cls.QUEUE_NAME)
host = aws_config.get_service_host(cls.QUEUE_REGION, cls.SERVICE)
assert cls.QUEUE_URL is not None
assert not cls.QUEUE_URL.startswith("http://%s" % host)
assert cls.QUEUE_URL.endswith(cls.QUEUE_NAME)
def test_get_service_host_found(self):
'''Test case for ensuring that hosts can be found for existing regions and services.'''
host_name = aws_config.get_service_host("eu-west-1", "sqs")
self.assertEqual("sqs.eu-west-1.amazonaws.com", host_name)
