from tests import factories from tests.base import OsfTestCase, fake from tests import utils as test_utils from tests.utils import unique as _unique SILENT_LOGGERS = ( 'framework.celery_tasks.utils', 'website.archiver.tasks', ) for each in SILENT_LOGGERS: logging.getLogger(each).setLevel(logging.CRITICAL) sha256_factory = _unique(fake.sha256) name_factory = _unique(fake.ean13) def file_factory(sha256=None): fname = name_factory() return { 'path': '/' + fname, 'name': fname, 'kind': 'file', 'size': random.randint(4, 4000), 'extra': { 'hashes': { 'sha256': sha256 or sha256_factory() } } }
from osf_tests import factories from tests.factories import MockOAuthAddonNodeSettings from tests.base import OsfTestCase, fake from tests import utils as test_utils from tests.utils import unique as _unique SILENT_LOGGERS = ( 'framework.celery_tasks.utils', 'framework.celery_tasks.signals', 'website.app', 'website.archiver.tasks', ) for each in SILENT_LOGGERS: logging.getLogger(each).setLevel(logging.CRITICAL) sha256_factory = _unique(fake.sha256) name_factory = _unique(fake.ean13) def file_factory(name=None, sha256=None): fname = name or name_factory() return { 'path': '/' + fname, 'name': fname, 'kind': 'file', 'size': random.randint(4, 4000), 'extra': { 'hashes': { 'sha256': sha256 or sha256_factory() } } }
from tests import factories from tests.base import OsfTestCase, fake from tests import utils as test_utils from tests.utils import unique as _unique SILENT_LOGGERS = ( 'framework.tasks.utils', 'website.archiver.tasks', ) for each in SILENT_LOGGERS: logging.getLogger(each).setLevel(logging.CRITICAL) sha256_factory = _unique(fake.sha256) name_factory = _unique(fake.word) def file_factory(sha256=None): fname = name_factory() return { 'path': '/' + fname, 'name': fname, 'kind': 'file', 'size': random.randint(4, 4000), 'extra': { 'hashes': { 'sha256': sha256 or sha256_factory() } } }