def __release_public_ports(self, expr_id, host_server, host_ports):
ep = Endpoint(Service(self.load_azure_key_id(expr_id)))
host_server_name = host_server.vm_name
host_server_dns = host_server.public_dns.split('.')[0]
self.log.debug("starting to release ports ... ")
ep.release_public_endpoints(host_server_dns, 'Production',
host_server_name, host_ports)
def report_health(self):
azure_key = self.db.find_first_object(AzureKey)
if not azure_key:
return {
STATUS: HEALTH_STATUS.WARNING,
DESCRIPTION: "No Azure key found"
}
azure = Service(azure_key.id)
if azure.ping():
return {STATUS: HEALTH_STATUS.OK}
else:
return {STATUS: HEALTH_STATUS.ERROR}
def setUp(self):
db_adapter_url = 'hackathon.azureformation.service.db_adapter'
with mock.patch(db_adapter_url) as db_adapter:
subscription_id = 'hter45sd-gf34-gfd4-ht34-ds43gree23gr'
pem_url = '/home/azureformation.pem'
management_host = 'das342.fdsf34.com'
db_adapter.get_object.return_value = AzureKey(
subscription_id=subscription_id,
pem_url=pem_url,
management_host=management_host)
azure_key_id = 0
self.service = Service(azure_key_id)
def __get_available_public_ports(self, expr_id, host_server, host_ports):
self.log.debug("starting to get azure ports")
ep = Endpoint(Service(self.load_azure_key_id(expr_id)))
host_server_name = host_server.vm_name
host_server_dns = host_server.public_dns.split('.')[0]
public_endpoints = ep.assign_public_endpoints(host_server_dns,
'Production',
host_server_name,
host_ports)
if not isinstance(public_endpoints, list):
self.log.debug("failed to get public ports")
return internal_server_error('cannot get public ports')
self.log.debug("public ports : %s" % public_endpoints)
return public_endpoints
def __init__(self, azure_key_id):
self.azure_key_id = azure_key_id
self.service = Service(self.azure_key_id)
self.subscription = Subscription(self.service)
