def setup_application(self):
if "yaml_template" in self.config:
with open(self.config["yaml_template"], "rb") as f:
self.yaml_template = f.read()
else:
self.yaml_template = resource_string(__name__,
"toy_decision_tree.yaml")
self.r_server = redis.Redis(**self.config.get('redis', {}))
self.session_manager = SessionManager(
self.r_server,
"%(worker_name)s:%(transport_name)s" % self.config,
max_session_length=self.MAX_SESSION_LENGTH)
def startWorker(self):
"""
Setup session manager.
"""
self.redis_config = self.config.get('redis_config', {})
self.redis_server = redis.Redis(**self.redis_config)
self.session_manager = SessionManager(
r_server=self.redis_server,
prefix="%(worker_name)s:%(transport_name)s" % self.config,
max_session_length=getattr(self, 'MAX_SESSION_LENGTH', None))
if self.flush_sessions:
self.redis_server.flushdb()
yield super(MenuApplicationWorker, self).startWorker()
def setUp(self):
self.fake_redis = FakeRedis()
self.add_cleanup(self.fake_redis.teardown)
self.sm = SessionManager(self.fake_redis, prefix="test")
self.add_cleanup(self.sm.stop)
def setUp(self):
self.fake_redis = FakeRedis()
self.sm = SessionManager(self.fake_redis, prefix="test")
