def get_installed_ext_handler(self):
lastest_version = self.get_installed_version()
if lastest_version is None:
return None
installed_handler = ExtHandler()
set_properties("ExtHandler", installed_handler, get_properties(self.ext_handler))
installed_handler.properties.version = lastest_version
return ExtHandlerInstance(installed_handler, self.protocol)
def get_installed_ext_handler(self):
lastest_version = self.get_installed_version()
if lastest_version is None:
return None
installed_handler = ExtHandler()
set_properties("ExtHandler", installed_handler, get_properties(self.ext_handler))
installed_handler.properties.version = lastest_version
return ExtHandlerInstance(installed_handler, self.protocol)
def get_handler_status(self):
state_dir = self.get_conf_dir()
status_file = os.path.join(state_dir, "HandlerStatus")
if not os.path.isfile(status_file):
return None
try:
data = json.loads(fileutil.read_file(status_file))
handler_status = ExtHandlerStatus()
set_properties("ExtHandlerStatus", handler_status, data)
return handler_status
except (IOError, ValueError) as e:
self.logger.error("Failed to get handler status: {0}", e)
def get_handler_status(self):
state_dir = self.get_conf_dir()
status_file = os.path.join(state_dir, "HandlerStatus")
if not os.path.isfile(status_file):
return None
try:
data = json.loads(fileutil.read_file(status_file))
handler_status = ExtHandlerStatus()
set_properties("ExtHandlerStatus", handler_status, data)
return handler_status
except (IOError, ValueError) as e:
self.logger.error("Failed to get handler status: {0}", e)
def _setup_image_origin_assert(publisher, offer, sku, version):
s = '''{{
"publisher": "{0}",
"offer": "{1}",
"sku": "{2}",
"version": "{3}"
}}'''.format(publisher, offer, sku, version)
data = json.loads(s, encoding='utf-8')
compute_info = imds.ComputeInfo()
set_properties("compute", compute_info, data)
return compute_info.image_origin
def _setup_image_origin_assert(publisher, offer, sku, version):
s = '''{{
"publisher": "{0}",
"offer": "{1}",
"sku": "{2}",
"version": "{3}"
}}'''.format(publisher, offer, sku, version)
data = json.loads(s, encoding='utf-8')
compute_info = imds.ComputeInfo()
set_properties("compute", compute_info, data)
return compute_info.image_origin
def test_deserialize_ComputeInfo(self):
s = '''{
"location": "westcentralus",
"name": "unit_test",
"offer": "UnitOffer",
"osType": "Linux",
"placementGroupId": "",
"platformFaultDomain": "0",
"platformUpdateDomain": "0",
"publisher": "UnitPublisher",
"resourceGroupName": "UnitResourceGroupName",
"sku": "UnitSku",
"subscriptionId": "e4402c6c-2804-4a0a-9dee-d61918fc4d28",
"tags": "Key1:Value1;Key2:Value2",
"vmId": "f62f23fb-69e2-4df0-a20b-cb5c201a3e7a",
"version": "UnitVersion",
"vmSize": "Standard_D1_v2",
"vmScaleSetName": "MyScaleSet",
"zone": "In"
}'''
data = json.loads(s, encoding='utf-8')
compute_info = ComputeInfo()
set_properties("compute", compute_info, data)
self.assertEqual('westcentralus', compute_info.location)
self.assertEqual('unit_test', compute_info.name)
self.assertEqual('UnitOffer', compute_info.offer)
self.assertEqual('Linux', compute_info.osType)
self.assertEqual('', compute_info.placementGroupId)
self.assertEqual('0', compute_info.platformFaultDomain)
self.assertEqual('0', compute_info.platformUpdateDomain)
self.assertEqual('UnitPublisher', compute_info.publisher)
self.assertEqual('UnitResourceGroupName',
compute_info.resourceGroupName)
self.assertEqual('UnitSku', compute_info.sku)
self.assertEqual('e4402c6c-2804-4a0a-9dee-d61918fc4d28',
compute_info.subscriptionId)
self.assertEqual('Key1:Value1;Key2:Value2', compute_info.tags)
self.assertEqual('f62f23fb-69e2-4df0-a20b-cb5c201a3e7a',
compute_info.vmId)
self.assertEqual('UnitVersion', compute_info.version)
self.assertEqual('Standard_D1_v2', compute_info.vmSize)
self.assertEqual('MyScaleSet', compute_info.vmScaleSetName)
self.assertEqual('In', compute_info.zone)
self.assertEqual('UnitPublisher:UnitOffer:UnitSku:UnitVersion',
compute_info.image_info)
def test_deserialize_ComputeInfo(self):
s = '''{
"location": "westcentralus",
"name": "unit_test",
"offer": "UnitOffer",
"osType": "Linux",
"placementGroupId": "",
"platformFaultDomain": "0",
"platformUpdateDomain": "0",
"publisher": "UnitPublisher",
"resourceGroupName": "UnitResourceGroupName",
"sku": "UnitSku",
"subscriptionId": "e4402c6c-2804-4a0a-9dee-d61918fc4d28",
"tags": "Key1:Value1;Key2:Value2",
"vmId": "f62f23fb-69e2-4df0-a20b-cb5c201a3e7a",
"version": "UnitVersion",
"vmSize": "Standard_D1_v2",
"vmScaleSetName": "MyScaleSet",
"zone": "In"
}'''
data = json.loads(s, encoding='utf-8')
compute_info = imds.ComputeInfo()
set_properties("compute", compute_info, data)
self.assertEqual('westcentralus', compute_info.location)
self.assertEqual('unit_test', compute_info.name)
self.assertEqual('UnitOffer', compute_info.offer)
self.assertEqual('Linux', compute_info.osType)
self.assertEqual('', compute_info.placementGroupId)
self.assertEqual('0', compute_info.platformFaultDomain)
self.assertEqual('0', compute_info.platformUpdateDomain)
self.assertEqual('UnitPublisher', compute_info.publisher)
self.assertEqual('UnitResourceGroupName', compute_info.resourceGroupName)
self.assertEqual('UnitSku', compute_info.sku)
self.assertEqual('e4402c6c-2804-4a0a-9dee-d61918fc4d28', compute_info.subscriptionId)
self.assertEqual('Key1:Value1;Key2:Value2', compute_info.tags)
self.assertEqual('f62f23fb-69e2-4df0-a20b-cb5c201a3e7a', compute_info.vmId)
self.assertEqual('UnitVersion', compute_info.version)
self.assertEqual('Standard_D1_v2', compute_info.vmSize)
self.assertEqual('MyScaleSet', compute_info.vmScaleSetName)
self.assertEqual('In', compute_info.zone)
self.assertEqual('UnitPublisher:UnitOffer:UnitSku:UnitVersion', compute_info.image_info)
def get_compute(self):
"""
Fetch compute information.
:return: instance of a ComputeInfo
:rtype: ComputeInfo
"""
resp = restutil.http_get(self.compute_url, headers=self._headers)
if restutil.request_failed(resp):
raise HttpError("{0} - GET: {1}".format(resp.status, self.compute_url))
data = resp.read()
data = json.loads(ustr(data, encoding="utf-8"))
compute_info = ComputeInfo()
set_properties('compute', compute_info, data)
return compute_info
def get_compute(self):
"""
Fetch compute information.
:return: instance of a ComputeInfo
:rtype: ComputeInfo
"""
resp = restutil.http_get(self.compute_url, headers=self._headers)
if restutil.request_failed(resp):
raise HttpError("{0} - GET: {1}".format(resp.status, self.compute_url))
data = resp.read()
data = json.loads(ustr(data, encoding="utf-8"))
compute_info = ComputeInfo()
set_properties('compute', compute_info, data)
return compute_info
def parse_json_event(data_str):
data = json.loads(data_str)
event = TelemetryEvent()
set_properties("TelemetryEvent", event, data)
return event