def test_write_content_to_file(mocker): filename = "filename" content = "content" sftp = mocker.patch("paramiko.SFTPClient") mock_open = mocker.patch("paramiko.SFTPClient.open") ssh.write_content_to_file(sftp, filename, content) mock_open.assert_called_once_with(filename, 'w')
def change_repositories(node, repos): ssh.remove_all_files_from_dirs(['/etc/apt/sources.list.d', '/etc/apt/preferences.d'], node) sftp = ssh.sftp(node) for repo in repos: filename_source, content_source = apt.create_repo_source(repo) ssh.write_content_to_file(sftp, filename_source, content_source) if repo['priority']: filename_pref, content_pref = apt.create_repo_preferences(repo) ssh.write_content_to_file(sftp, filename_pref, content_pref) ssh.call(['apt-get', 'update'], node=node)