ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2024-12-21 05:53:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

use attr.s

Browse Source
This commit is contained in:
meejah 2020-06-13 17:28:28 -06:00 committed by Sajith Sasidharan
parent af70911e2f
commit e0287a7bfe
1 changed files with 13 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/allmydata/testing/web.py
View File

@ -133,6 +133,7 @@ def capability_generator(kind):
yield cap.encode("ascii") yield cap.encode("ascii")
@attr.s
class _FakeTahoeUriHandler(Resource, object): class _FakeTahoeUriHandler(Resource, object):
""" """
An in-memory fake of (some of) the `/uri` endpoint of a Tahoe An in-memory fake of (some of) the `/uri` endpoint of a Tahoe
@ -140,8 +141,9 @@ class _FakeTahoeUriHandler(Resource, object):
""" """
isLeaf = True isLeaf = True
_data = None
_capability_generators = None data = attr.ib(default=attr.Factory(dict))
capability_generators = attr.ib(default=attr.Factory(dict))
def _generate_capability(self, kind): def _generate_capability(self, kind):
""" """
@ -149,12 +151,9 @@ class _FakeTahoeUriHandler(Resource, object):
:returns: the next capability-string for the given kind :returns: the next capability-string for the given kind
""" """
if self._capability_generators is None: if kind not in self.capability_generators:
self._capability_generators = dict() self.capability_generators[kind] = capability_generator(kind)
capability = next(self.capability_generators[kind])
if kind not in self._capability_generators:
self._capability_generators[kind] = capability_generator(kind)
capability = next(self._capability_generators[kind])
return capability return capability
def add_data(self, kind, data, allow_duplicate=False): def add_data(self, kind, data, allow_duplicate=False):
@ -166,11 +165,8 @@ class _FakeTahoeUriHandler(Resource, object):
if not isinstance(data, bytes): if not isinstance(data, bytes):
raise TypeError("'data' must be bytes") raise TypeError("'data' must be bytes")
if self._data is None: for k in self.data:
self._data = dict() if self.data[k] == data:
for k in self._data:
if self._data[k] == data:
if allow_duplicate: if allow_duplicate:
return k return k
raise ValueError( raise ValueError(
@ -178,9 +174,9 @@ class _FakeTahoeUriHandler(Resource, object):
) )
cap = self._generate_capability(kind) cap = self._generate_capability(kind)
if cap in self._data: if cap in self.data:
raise ValueError("already have '{}'".format(cap)) raise ValueError("already have '{}'".format(cap))
self._data[cap] = data self.data[cap] = data
return cap return cap
def render_PUT(self, request): def render_PUT(self, request):
@ -217,10 +213,10 @@ class _FakeTahoeUriHandler(Resource, object):
) )
) )
if self._data is None or capability not in self._data: if capability not in self.data:
return u"No data for '{}'".format(capability).decode("ascii") return u"No data for '{}'".format(capability).decode("ascii")
return self._data[capability] return self.data[capability]
def create_fake_tahoe_root(): def create_fake_tahoe_root():