def test_did_downloader_is_downloading__lockfile_exists__pid_not_exists__should_return_false(
mocker):
mocker.patch.object(os.path, 'isfile', return_value=True)
mocker.patch.object(psutil, 'pid_exists', return_value=False)
with patch("builtins.open", mock_open(read_data="123")) as mock_file:
result = DIDDownloader.is_downloading('/path')
assert not result, 'Invalid return value'
mock_file.assert_called_with(os.path.join('/path', '.lockfile'), 'r')
def test_did_downloader_write_temp_config_file__should_make_correct_directory(
mocker):
mock_config = {
'rucio_host': 'https://rucio',
'auth_host': 'https://rucio-auth',
'auth_type': 'userpass',
'username': '******',
'password': '******',
'account': 'root',
'ca_cert': '/opt/certs/rucio.pem'
}
mocker.patch.object(os, 'makedirs', return_value=None)
with patch("builtins.open", mock_open()) as mock_file:
DIDDownloader.write_temp_config_file('/path', mock_config)
mock_file.assert_called_with(os.path.join('/path', 'etc', 'rucio.cfg'),
'w')
os.makedirs.assert_called_once_with(os.path.join('/path', 'etc'),
exist_ok=True) # pylint: disable=no-member
def test_did_downloader_is_downloading__lockfile_exists__pid_exists__process_running__status_zombie__should_return_false(
mocker):
mocker.patch.object(os.path, 'isfile', return_value=True)
mocker.patch.object(psutil, 'pid_exists', return_value=True)
class MockProcess:
def __init__(self, pid):
pass
def is_running(self): # pylint: disable=no-self-use
return True
def status(self): # pylint: disable=no-self-use
return 'zombie'
mocker.patch('rucio_jupyterlab.mode_handlers.download.psutil.Process',
MockProcess)
with patch("builtins.open", mock_open(read_data="123")) as mock_file:
result = DIDDownloader.is_downloading('/path')
assert not result, 'Invalid return value'
mock_file.assert_called_with(os.path.join('/path', '.lockfile'), 'r')
def test_did_downloader_write_lockfile__should_write_pid(mocker):
mocker.patch.object(os, 'getpid', return_value=123)
with patch("builtins.open", mock_open()) as mock_file:
DIDDownloader.write_lockfile('/path')
mock_file.assert_called_with(os.path.join('/path', '.lockfile'), 'w')
mock_file.return_value.write.assert_called_once_with('123')
def test_did_downloader_is_downloading__lockfile_not_exists__should_return_false(
mocker):
mocker.patch.object(os.path, 'isfile', return_value=False)
result = DIDDownloader.is_downloading('/path')
assert not result, 'Invalid return value'