def partial_update(self, guid, body, code=200):
url = os.path.join(self.url, guid, "partial")
response, status = Atlas.http_put_post_request(url=url,
data=json.dumps(body),
method='PUT')
assert status == code
return response
def associate_term_to_entities(self, guid, entity_guids_list, code=204):
url = os.path.join(self.base_url, 'api/atlas/v2/glossary/terms', guid,
"assignedEntities")
body = list()
for each in entity_guids_list:
body.append({"guid": each})
_response, status = Atlas.http_put_post_request(url=url,
data=json.dumps(body))
assert status == code
def disassociate_term_to_entities(self, guid, entity_guids_dict, code=204):
url = os.path.join(self.base_url, 'api/atlas/v2/glossary/terms', guid,
"assignedEntities")
body = list()
for entity_guid, relationship_guid in entity_guids_dict.iteritems():
body.append({
"guid": entity_guid,
"relationshipGuid": relationship_guid
})
status = Atlas.http_delete_request(url=url, body=json.dumps(body))
assert status == code
def run_smoke_test(cls):
'''
Run smoke test for yarn
:param smoketestnumber: Used for unique output log location
'''
from beaver.component.rollingupgrade.ruUpgrade import UpgradePerNode
host = Atlas.get_host()
if not host:
logger.error("No ACTIVE Atlas instance is available")
else:
logger.info("Active instance of Atlas is found: " + host)
response, response_status = Atlas.create_entities(input_json_string=cls.entity_create_request_payload)
if response_status != 201:
UpgradePerNode.reportProgress("[FAILED][Atlas][Smoke] Entity creation failed = %d" % response_status)
else:
guid = response["definition"]["id"]["id"]
logger.info("Entity creation was successful. GUID = %s" % guid)
cls.guid_list.append(guid)
UpgradePerNode.reportProgress("[INFO][Atlas][Smoke] Smoke test for Atlas component finished")
def verifyLongRunningJob(cls):
'''
Validate long running background job after end of all component upgrade
'''
from beaver.component.rollingupgrade.ruUpgrade import UpgradePerNode
for guid in cls.guid_list:
response, response_status = Atlas.get_entities(guid)
if response_status != 200:
UpgradePerNode.reportProgress("[FAILED][Atlas][Smoke] Get Entity failed for guid %d" % guid)
else:
guid_in_response = response["definition"]["id"]["id"]
if guid == guid_in_response:
logger.info("Get Entity was successful for GUID %s" % guid)
else:
logger.error("Get entity guid error for guid %s. Response - %s" % (guid, response))
def smoke_test_setup(cls):
'''
Setup required to run Smoke test
'''
Atlas.create_type(input_json_string=cls.type_request_payload)
def get_associated_terms_headers(self, guid, code=200):
url = os.path.join(self.url, guid, "terms", "headers")
response, status = Atlas.http_get_request(url=url)
assert status == code
return response
def get_associated_categories(self, guid, code=200):
url = os.path.join(self.url, guid, "categories")
response, status = Atlas.http_get_request(url=url)
assert status == code
return response
def get_detailed(self, guid, code=200):
url = os.path.join(self.url, guid, "detailed")
response, status = Atlas.http_get_request(url=url)
assert status == code
return response
def get_all(self, code=200):
response, status = Atlas.http_get_request(url=self.url)
assert status == code
return response
def create_categories(self, body, code=200):
url = os.path.join(self.base_url, 'api/atlas/v2/glossary/categories')
response, status = Atlas.http_put_post_request(url=url,
data=json.dumps(body))
assert status == code
return response
def get(self, guid, code=200):
response, status = Atlas.http_get_request(
url=os.path.join(self.url, guid))
assert status == code
return response
def delete(self, guid, code=204):
status = Atlas.http_delete_request(url=os.path.join(self.url, guid))
assert status == code
def create(self, body, code=200):
response, status = Atlas.http_put_post_request(url=self.url,
data=json.dumps(body))
assert status == code
return response
def __init__(self, url):
self.base_url = Atlas.get_base_url()
self.url = os.path.join(self.base_url, url)
def get_related_terms(self, guid, code=200):
url = os.path.join(self.base_url, 'api/atlas/v2/glossary/terms', guid,
"related")
response, status = Atlas.http_get_request(url=url)
assert status == code
return response
