tahoe-lafs/src/allmydata/test/test_json_metadata.py
Itamar Turner-Trauring 61c76902ca Port to Python 3.
2020-12-16 11:16:34 -05:00

68 lines
2.2 KiB
Python

"""
Ported to Python 3.
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from future.utils import PY2
if PY2:
from future.builtins import filter, map, zip, ascii, chr, hex, input, next, oct, open, pow, round, super, bytes, dict, list, object, range, str, max, min # noqa: F401
from twisted.trial.unittest import TestCase
from allmydata.web.common import get_filenode_metadata, SDMF_VERSION, MDMF_VERSION
class MockFileNode(object):
def __init__(self, size, mutable_version=None):
self.size = size
self.mutable_version = mutable_version
def get_size(self):
return self.size
def is_mutable(self):
return self.mutable_version is not None
def get_version(self):
if self.mutable_version is None:
raise AttributeError()
return self.mutable_version
class CommonFixture(object):
def test_size_is_0(self):
"""If get_size doesn't return None the returned metadata must contain "size"."""
mockfilenode = MockFileNode(0, mutable_version=self.mutable_version)
metadata = get_filenode_metadata(mockfilenode)
self.failUnlessEqual(metadata['size'], 0)
def test_size_is_1000(self):
"""1000 is sufficiently large to guarantee the cap is not a literal."""
mockfilenode = MockFileNode(1000, mutable_version=self.mutable_version)
metadata = get_filenode_metadata(mockfilenode)
self.failUnlessEqual(metadata['size'], 1000)
def test_size_is_None(self):
"""If get_size returns None the returned metadata must not contain "size"."""
mockfilenode = MockFileNode(None, mutable_version=self.mutable_version)
metadata = get_filenode_metadata(mockfilenode)
self.failIfIn('size', metadata)
class Test_GetFileNodeMetaData_Immutable(CommonFixture, TestCase):
def setUp(self):
self.mutable_version = None
class Test_GetFileNodeMetaData_SDMF(CommonFixture, TestCase):
def setUp(self):
self.mutable_version = SDMF_VERSION
class Test_GetFileNodeMetaData_MDMF(CommonFixture, TestCase):
def setUp(self):
self.mutable_version = MDMF_VERSION