def test_fetch_file_local_name_specified(OSF_project, os_path_exists,
os_makedirs):
# check that `osf fetch` opens the right files with the right name
# and mode when specifying a local filename
args = MockArgs(project='1234',
remote='osfstorage/a/a/a',
local='foobar.txt')
mock_open_func = mock_open()
with patch('osfclient.cli.open', mock_open_func):
fetch(args)
OSF_project.assert_called_once_with('1234')
# check that the project and the files have been accessed
project = OSF_project.return_value
store = project._storage_mock.return_value
assert store._name_mock.return_value == 'osfstorage'
expected = [call._path_mock(), call.write_to(mock_open_func())]
assert expected == store.files[0].mock_calls
# second file should not have been looked at
assert not store.files[1].mock_calls
# should create a file in the same directory when no local
# filename is specified
assert mock.call('foobar.txt', 'wb') in mock_open_func.mock_calls
assert not os_makedirs.called
def test_fetch_local_file_exists_force_overrides_update(
OSF_project, os_makedirs):
# check that `osf fetch --force --update` overwrites the local file even if
# it matches the remote file (force overrides update)
args = MockArgs(project='1234',
remote='osfstorage/a/a/a',
force=True,
update=True)
def exists(path):
if path == ".osfcli.config":
return False
else:
return True
def simple_checksum(file_path):
return '0' * 32
mock_open_func = mock_open()
with patch('osfclient.cli.open', mock_open_func):
with patch('osfclient.cli.os.path.exists', side_effect=exists):
with patch('osfclient.cli.checksum', side_effect=simple_checksum):
fetch(args)
OSF_project.assert_called_once_with('1234')
# check that the project and the files have been accessed
store = OSF_project.return_value.storages[0]
assert store._name_mock.return_value == 'osfstorage'
# should create a file in the same directory when no local
# filename is specified.
# file should be created even though local matches remote and update is
# True, because force overrides update
assert mock.call('a', 'wb') in mock_open_func.mock_calls
def test_fetch_local_file_exists_update_files_differ(OSF_project, os_makedirs):
# check that `osf fetch --update` overwrites an existing local file if it
# differs from the remote
args = MockArgs(project='1234', remote='osfstorage/a/a/a', update=True)
def exists(path):
if path == ".osfcli.config":
return False
else:
return True
def simple_checksum(file_path):
return '1' * 32
mock_open_func = mock_open()
with patch('osfclient.cli.open', mock_open_func):
with patch('osfclient.cli.os.path.exists', side_effect=exists):
with patch('osfclient.cli.checksum', side_effect=simple_checksum):
fetch(args)
OSF_project.assert_called_once_with('1234')
# check that the project and the files have been accessed
store = OSF_project.return_value.storages[0]
assert store._name_mock.return_value == 'osfstorage'
# should create a file in the same directory when no local
# filename is specified
assert mock.call('a', 'wb') in mock_open_func.mock_calls
def download_erpcore(task="MMN", subject=1, localpath="local/bids/"):
project = "9f5w7" # after recent change they put everything as "sessions" in one big BIDS file
arguments = args(project) # project ID
for extension in [
"channels.tsv", "events.tsv", "eeg.fdt", "eeg.json", "eeg.set"
]:
targetpath = '/sub-{:03d}/ses-{}/eeg/sub-{:03d}_ses-{}_task-{}_{}'.format(
subject, task, subject, task, task, extension)
print("Downloading {}".format(targetpath))
arguments.remote = "\\ERP_CORE_BIDS_Raw_Files/" + targetpath
arguments.local = localpath + targetpath
cli.fetch(arguments)
def test_fetch_file(OSF_project, os_path_exists, os_makedirs):
# check that `osf fetch` opens the right files with the right name and mode
args = MockArgs(project='1234', remote='osfstorage/a/a/a')
mock_open_func = mock_open()
with patch('osfclient.cli.open', mock_open_func):
fetch(args)
OSF_project.assert_called_once_with('1234')
# check that the project and the files have been accessed
store = OSF_project.return_value.storages[0]
assert store._name_mock.return_value == 'osfstorage'
# should create a file in the same directory when no local
# filename is specified
assert mock.call('a', 'wb') in mock_open_func.mock_calls
def test_fetch_local_file_exists(OSF_project):
# check that `osf fetch` opens the right files with the right name
# and mode when specifying a local filename
args = MockArgs(project='1234',
remote='osfstorage/a/a/a',
local='subdir/foobar.txt')
def exists(path):
if path == ".osfcli.config":
return False
else:
return True
with patch('osfclient.cli.os.path.exists', side_effect=exists):
with pytest.raises(SystemExit) as e:
fetch(args)
assert 'already exists, not overwriting' in e.value.args[0]
def test_fetch_file_local_dir_specified(OSF_project, os_path_exists,
os_makedirs):
# check that `osf fetch` opens the right files with the right name
# and mode when specifying a local filename
args = MockArgs(project='1234',
remote='osfstorage/a/a/a',
local='subdir/foobar.txt')
mock_open_func = mock_open()
with patch('osfclient.cli.open', mock_open_func):
fetch(args)
OSF_project.assert_called_once_with('1234')
# check that the project and the files have been accessed
store = OSF_project.return_value.storages[0]
assert store._name_mock.return_value == 'osfstorage'
assert (mock.call('subdir/foobar.txt', 'wb') in mock_open_func.mock_calls)
assert mock.call('subdir', exist_ok=True) in os_makedirs.mock_calls
