Python mocked_gfal2 Examples

Example #1

File: test_dumper.py Project: zzaiin/Rucio

def test_srm_download_to_file():
    srm_file = StringIO()
    local_file = StringIO()
    srm_file.write('content')
    srm_file.seek(0)

    with gfal2.mocked_gfal2(dumper, files={'srm://example.com/file': srm_file}):
        dumper.srm_download_to_file('srm://example.com/file', local_file)

    local_file.seek(0)
    eq_(local_file.read(), 'content')

Example #2

File: test_dumper.py Project: zzaiin/Rucio

def test_srm_download_creates_file_with_content():
    srm_file = StringIO()
    local_file = StringIO()
    srm_file.write('content')
    srm_file.seek(0)

    with gfal2.mocked_gfal2(dumper, files={'srm://example.com/file': srm_file}):
        with mock_open(dumper, local_file):
            dumper.srm_download('srm://example.com/file', 'filename')

    local_file.seek(0)
    eq_(local_file.read(), 'content')