def setProdCommonDB():
Session.set_database(Cconfiguration.get('DB'))
Session.connect('ProdCommon')
Session.set_session('ProdCommon')
Session.start_transaction()
def testA(self):
job_parameters={'numberOfJobs':20,
'prefix':'Wave1'}
job_result=[{'jobSpecId':'jobspec1',\
'URL':'http://somewhere',\
'start_event':300,\
'end_event':350},
{'jobSpecId':'jobspec2',\
'URL':'http://somewhere',\
'start_event':351,\
'end_event':400},
{'jobSpecId':'jobspec3',\
'URL':'http://somewhere',\
'start_event':400,\
'end_event':450}]
allocation_result=["myRequest/1","myRequest/2","myRequest/3","myRequest4"]
try:
print("testA")
Session.set_database(configuration.get('DB'))
for i in xrange(0,self.prodagents):
for j in xrange(0,self.components):
serviceCall.log("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"acquireAllocation",\
["prodAgentID","myRequest",15],\
allocation_result,"my_tag1")
logged_result=serviceCall.retrieve("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"my_tag1","acquireAllocation")
self.assertEqual(logged_result[0],allocation_result)
try:
logged_result=serviceCall.retrieve("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"my_wrong_tag1","acquireAllocation")
except Exception,ex:
print(str(ex))
print('INTENTIONAL ERROR!')
serviceCall.log("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"acquireJobs",\
["prodAgentID","myRequest",job_parameters],\
job_result)
logged_result=serviceCall.retrieve("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"0","acquireJobs")
self.assertEqual(logged_result[0],job_result)
serviceCall.log("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"releaseAllocation",\
["prodAgentID","myRequest/1"],\
True)
logged_result=serviceCall.retrieve("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"0","releaseAllocation")
self.assertEqual(logged_result[0],True)
serviceCall.log("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"releaseJob",\
["prodAgentID","jobspec1",30],\
True)
logged_result=serviceCall.retrieve("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"0","releaseJob")
self.assertEqual(logged_result[0],True)
try:
serviceCall.log("DN-ProdAgent-"+str(i)+"-Component-1","acquireAllocation",\
["prodAgentID","myRequest",15],\
allocation_result)
except:
print("Handling error")
pass
print("Pretending the client crashes")
for i in xrange(0,self.prodagents):
for j in xrange(0,self.components):
serviceCall.remove("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"acquireAllocation")
serviceCall.remove("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"acquireJobs")
serviceCall.remove("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"releaseAllocation")
serviceCall.remove("DN-ProdAgent-"+str(i)+"-Component-"+str(j),"releaseJob")
def buildQuery(queryType, parameters):
if configuration.get('DB')['dbType'] == 'mysql':
query = retrieveHandler(queryType, "MySQL")
if configuration.get('DB')['dbType'] == 'oracle':
query = retrieveHandler(queryType, "Oracle")
return query.build(parameters)
def setProdCommonDB():
Session.set_database(Cconfiguration.get("DB"))
Session.connect("ProdCommon")
Session.set_session("ProdCommon")
Session.start_transaction()
def buildQuery(queryType,parameters):
if configuration.get('DB')['dbType']=='mysql':
query=retrieveHandler(queryType,"MySQL")
if configuration.get('DB')['dbType']=='oracle':
query=retrieveHandler(queryType,"Oracle")
return query.build(parameters)