import pytest

from test.mock.mock_backup import MockBackupRoutine


@pytest.fixture(scope='module')
def br():
    _br = MockBackupRoutine()
    return _br


@pytest.mark.backup_routine
def test_backup_user_archive(br, mocker):
    """Test whether expected user archive is returned"""
    assert br.user_data is not None
    assert br.user_archive(username='mockuser') is not None
    assert br.user_archive(username='mockuser') == 'mockuser.tar.gz'
    assert br.user_archive(username='fakeuser') is None


@pytest.mark.backup_routine
def test_backup_user_data(br, mocker):
    """Test whether expected user-data is returned"""
    assert br.user_data is not None
    assert isinstance(br.user_data, list)


@pytest.mark.backup_routine
def test_assert_backup_users(br, mocker):
    """Test whether users are valid"""
    for user_acct in br.user_data:
        assert br.username in user_acct['user']
    br.backup = 0
    for user_acct in br.user_data:
        assert br.username not in user_acct['user']
    br.reset()
    br.suspended = 1
    for user_acct in br.user_data:
        assert br.username not in user_acct['user']
    br.reset()
    br.null_routed = True
    for user_acct in br.user_data:
        assert br.username not in user_acct['user']
    br.reset()


@pytest.mark.backup_routine
def test_backup_package_account(br, mocker):
    """Test whether account packaging is invoked when expected"""
    br.run()
    assert br.acct_packaged is False
    br.run(pkg=True)
    assert br.acct_packaged is True
    br.reset()


@pytest.mark.backup_routine
def test_backup_archive_upload(br, mocker):
    """Test whether archive is uploaded when expected"""
    assert br.upload_complete is False
    br.run()
    assert br.upload_complete is True