def setUp(self):
twisted.internet.base.DelayedCall.debug = True
conf = AutopushConfig(
hostname="localhost",
statsd_host=None,
bear_hash_key='AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAB=',
)
self.fernet_mock = conf.fernet = Mock(spec=Fernet)
self.db = db = test_db()
db.router = Mock(spec=Router)
db.router.register_user.return_value = (True, {}, {})
db.router.get_uaid.return_value = {
"router_type": "test",
"router_data": dict()
}
db.create_initial_message_tables()
self.routers = routers = routers_from_config(conf, db, Mock())
routers["test"] = Mock(spec=IRouter)
app = EndpointHTTPFactory(conf, db=db, routers=routers)
self.client = Client(app)
self.request_mock = Mock(body=b'', arguments={}, headers={})
self.reg = NewRegistrationHandler(app, self.request_mock)
self.auth = ("WebPush %s" %
generate_hash(conf.bear_hash_key[0], dummy_uaid.hex))
self.conf = conf
def _for_handler(cls, conf, db, routers=None, **kwargs):
if routers is None:
routers = routers_from_config(
conf,
db=db,
agent=agent_from_config(conf)
)
return cls(conf, db=db, routers=routers, **kwargs)
def __init__(self, conf, resource=None):
# type: (AutopushConfig, DynamoDBResource) -> None
super(EndpointApplication, self).__init__(conf, resource=resource)
self.routers = routers_from_config(conf, self.db, self.agent)
def __init__(self, conf):
# type: (AutopushConfig) -> None
super(EndpointApplication, self).__init__(conf)
self.routers = routers_from_config(conf, self.db, self.agent)