def test_exceptions(self):
stats = HttpAPIStatsCollector().get()
self.assertRaises(KeyNotFound, stats.get, 'anything')
self.assertRaises(KeyNotFound, stats.set, 'anything', 22)
self.assertRaises(KeyNotFound, stats.inc, 'anything')
stats.set('created_at', datetime.now())
self.assertRaises(KeyNotIncrementable, stats.inc, 'created_at')
def __init__(self,
RouterPB,
SMPPClientManagerPB,
config,
interceptor=None):
Resource.__init__(self)
# Setup stats collector
stats = HttpAPIStatsCollector().get()
stats.set('created_at', datetime.now())
# Set up a dedicated logger
log = logging.getLogger(LOG_CATEGORY)
if len(log.handlers) != 1:
log.setLevel(config.log_level)
handler = TimedRotatingFileHandler(filename=config.log_file,
when=config.log_rotate)
formatter = logging.Formatter(config.log_format,
config.log_date_format)
handler.setFormatter(formatter)
log.addHandler(handler)
log.propagate = False
self.log = log
# Set http url routings
log.debug("Setting http url routing for /send")
self.putChild(
b'send',
Send(config, RouterPB, SMPPClientManagerPB, stats, log,
interceptor))
log.debug("Setting http url routing for /rate")
self.putChild(b'rate', Rate(config, RouterPB, stats, log, interceptor))
log.debug("Setting http url routing for /balance")
self.putChild(b'balance', Balance(RouterPB, stats, log))
log.debug("Setting http url routing for /ping")
self.putChild(b'ping', Ping(log))
def test_stats_set(self):
stats = HttpAPIStatsCollector().get()
stats.set('success_count', 1000)
self.assertEqual(stats.get('success_count'), 1000)
