make sure we get bytes here

This commit is contained in:
Jean-Paul Calderone 2019-03-22 16:42:50 -04:00
parent 794314668a
commit febfa50a83
2 changed files with 7 additions and 2 deletions
src/allmydata

@ -49,7 +49,7 @@ class PrivacyTests(SyncTestCase):
Tests for the privacy features of the resources created by ``create_private_tree``.
"""
def setUp(self):
self.token = u"abcdef"
self.token = b"abcdef"
self.resource = create_private_tree(lambda: self.token)
self.agent = RequestTraversalAgent(self.resource)
self.client = HTTPClient(self.agent)

@ -46,6 +46,9 @@ from twisted.web.guard import (
from ..util.hashutil import (
timing_safe_compare,
)
from ..util.assertutil import (
precondition,
)
from .logs import (
create_log_resources,
@ -77,7 +80,9 @@ class TokenChecker(object):
credentialInterfaces = [IToken]
def requestAvatarId(self, credentials):
if credentials.equals(self.get_auth_token()):
required_token = self.get_auth_token()
precondition(isinstance(required_token, bytes))
if credentials.equals(required_token):
return succeed(ANONYMOUS)
return fail(Failure(UnauthorizedLogin()))