def test_recursive_upload_with_subdir(OSF_project):
# test that an extra level of subdirectory is created on the remote side
# this is because args.source does not end in a /
args = MockArgs(username='******',
project='1234',
source='foobar',
recursive=True,
destination='BAR/')
def simple_getenv(key):
if key == 'OSF_PASSWORD':
return 'secret'
fake_open = mock_open()
fake_storage = OSF_project.return_value.storage.return_value
# it is important we use foobar and not foobar/ to match with args.source
# to mimick the behaviour of os.walk()
dir_contents = [('foobar', None, ['bar.txt', 'abc.txt']),
('foobar/baz', None, ['bar.txt', 'abc.txt'])]
with patch('osfclient.cli.open', fake_open):
with patch('os.walk', return_value=iter(dir_contents)):
with patch('osfclient.cli.os.getenv', side_effect=simple_getenv):
with patch('osfclient.cli.os.path.isdir', return_value=True):
upload(args)
assert call('foobar/bar.txt', 'rb') in fake_open.mock_calls
assert call('foobar/abc.txt', 'rb') in fake_open.mock_calls
assert call('foobar/baz/bar.txt', 'rb') in fake_open.mock_calls
assert call('foobar/baz/abc.txt', 'rb') in fake_open.mock_calls
# two directories with two files each -> four calls plus all the
# context manager __enter__ and __exit__ calls
assert len(fake_open.mock_calls) == 4 + 4 * 2
fake_storage.assert_has_calls([
call.create_file('BAR/foobar/./bar.txt',
mock.ANY,
update=False,
force=False),
call.create_file('BAR/foobar/./abc.txt',
mock.ANY,
update=False,
force=False),
call.create_file('BAR/foobar/baz/bar.txt',
mock.ANY,
update=False,
force=False),
call.create_file('BAR/foobar/baz/abc.txt',
mock.ANY,
update=False,
force=False)
])
# two directories with two files each -> four calls
assert len(fake_storage.mock_calls) == 4
def test_select_project(OSF_project):
args = MockArgs(username='******',
project='1234',
source='foo/bar.txt',
destination='bar/bar/foo.txt')
def simple_getenv(key):
if key == 'OSF_PASSWORD':
return 'secret'
fake_open = mock_open()
with patch('osfclient.cli.open', fake_open):
with patch('osfclient.cli.os.getenv', side_effect=simple_getenv):
upload(args)
assert call('foo/bar.txt', 'rb') in fake_open.mock_calls
# the mock project created by calling OSF().project()
fake_project = OSF_project.return_value
expected = [call('osfstorage')]
assert fake_project._storage_mock.mock_calls == expected
# assert fake_project.mock_calls == expected
expected = [call.create_file('bar/bar/foo.txt', fake_open.return_value,
force=False, update=False)]
# we should call the create_file method on the return
# value of _storage_mock
assert fake_project._storage_mock.return_value.mock_calls == expected
