def test_validToken(self):
self.assertTrue(
validToken(
"2d0d55cd7f98bcb81c6e24abcdc35168254c7846a43e2828b1ba5a8f82e219df"
))
self.assertFalse(
validToken(
"d0d55cd7f98bcb81c6e24abcdc35168254c7846a43e2828b1ba5a8f82e219df"
))
self.assertFalse(validToken("foo"))
self.assertFalse(validToken(""))
def processSubscription(self, request):
"""
Given an authenticated request, use the token and key arguments
to add a subscription entry to the database.
@param request: The request to process
@type request: L{twext.web2.server.Request}
"""
token = request.args.get("token", ("",))[0].replace(" ", "").lower()
key = request.args.get("key", ("",))[0]
userAgent = request.headers.getHeader("user-agent", "-")
host = request.remoteAddr.host
fwdHeaders = request.headers.getRawHeaders("x-forwarded-for", [])
if fwdHeaders:
host = fwdHeaders[0]
if not (key and token):
code = responsecode.BAD_REQUEST
msg = "Invalid request: both 'token' and 'key' must be provided"
elif not validToken(token):
code = responsecode.BAD_REQUEST
msg = "Invalid request: bad 'token' %s" % (token,)
else:
principal = self.principalFromRequest(request)
uid = principal.record.uid
try:
yield self.addSubscription(token, key, uid, userAgent, host)
code = responsecode.OK
msg = None
except InvalidSubscriptionValues:
code = responsecode.BAD_REQUEST
msg = "Invalid subscription values"
returnValue((code, msg))
def processSubscription(self, request):
"""
Given an authenticated request, use the token and key arguments
to add a subscription entry to the database.
@param request: The request to process
@type request: L{txweb2.server.Request}
"""
token = request.args.get("token", ("", ))[0].replace(" ", "").lower()
key = request.args.get("key", ("", ))[0]
userAgent = request.headers.getHeader("user-agent", "-")
host = request.remoteAddr.host
fwdHeaders = request.headers.getRawHeaders("x-forwarded-for", [])
if fwdHeaders:
host = fwdHeaders[0]
if not (key and token):
code = responsecode.BAD_REQUEST
msg = "Invalid request: both 'token' and 'key' must be provided"
elif not validToken(token):
code = responsecode.BAD_REQUEST
msg = "Invalid request: bad 'token' %s" % (token, )
else:
uid = request.authnUser.record.uid
try:
yield self.addSubscription(token, key, uid, userAgent, host)
code = responsecode.OK
msg = None
except InvalidSubscriptionValues:
code = responsecode.BAD_REQUEST
msg = "Invalid subscription values"
returnValue((code, msg))
def test_validToken(self):
self.assertTrue(validToken("2d0d55cd7f98bcb81c6e24abcdc35168254c7846a43e2828b1ba5a8f82e219df"))
self.assertFalse(validToken("d0d55cd7f98bcb81c6e24abcdc35168254c7846a43e2828b1ba5a8f82e219df"))
self.assertFalse(validToken("foo"))
self.assertFalse(validToken(""))
