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)