test_internet_accounts.py

from mock import patch

import internet_accounts
import config

@patch('internet_accounts.backup_internet_accounts')
def test_backup(internet_accounts_mock):
    internet_accounts.backup()
    internet_accounts_mock.assert_called_once()


@patch('internet_accounts.restore_internet_accounts')
def test_restore(internet_accounts_mock):
    internet_accounts.restore()
    internet_accounts_mock.assert_called_once()


@patch('internet_accounts.copy_dir')
def test_backup_internet_accounts(copy_mock):
    source = config.get_internet_accounts_dir()
    dest = config.get_internet_accounts_backup_dir()
    internet_accounts.backup_internet_accounts()
    copy_mock.assert_called_with(
        source, dest
    )


@patch('internet_accounts.ensure_dir_owned_by_user')
@patch('internet_accounts.copy_dir')
def test_restore_internet_accounts(copy_mock, owned_mock):
    internet_accounts.restore_internet_accounts()
    copy_mock.assert_called_with(
        config.get_internet_accounts_backup_dir(),
        config.get_internet_accounts_dir(), with_sudo=True
    )
    owned_mock.assert_called_with(
        config.get_internet_accounts_dir(), config.get_user()
    )