Exemple #1
0
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()
            }
        }
    }
Exemple #2
0
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()
            }
        }
    }