def setup_method(self, method):
messageClass = 'SMS'
if 'EMAIL' in method.__name__:
messageClass = 'EMAIL'
self.nsObj = NSAdminHelper.getConnObj(newConnection=True)
self.masterNsObj = NSAdminHelper.getMasterConnObj()
self.nsadminHelper = NSAdminHelper(constant.config['orgId'],
messageClass)
self.nsadminHelper.disableDomainPropertiesGatewayMap()
Logger.logMethodName(method.__name__)
def updateCredit(credit=0,channel='EMAIL'):
try:
nsObj = NSAdminHelper.getMasterConnObj() if constant.config['cluster'] in ['nightly','staging','china'] else NSAdminHelper.getConnObj(newConnection=True)
messageClass = 1 if channel == 'EMAIL' else 0
creditDetails1 = {
"orgId": int(constant.config['orgId']),
"bulkCredits": int(credit),
'messageClass': messageClass
}
creditDetails1 = NSAdminObject.OrgCreditDetails(creditDetails1)
if credit == 0:
currVal = nsObj.getCreditDetailsByOrgAndChannel(constant.config['orgId'], messageClass, 'test_{}'.format(int(time.time())))
currVal = int(currVal.bulkCredits) * -1
creditDetails1.bulkCredits = currVal
nsObj.addCredits(creditDetails1)
except Exception,exp:
raise Exception('NotAbleToUpdateCreditTo:{}'.format(credit))
Logger.log(exp)