duplitab/tests/test_collection.py

import pytest

import os
import duplitab
import datetime

project_root_path = os.path.realpath(os.path.join(__file__, '..', '..'))
test_data_dir_path = os.path.join(project_root_path, 'tests', 'data')
test_collections_dir_path = os.path.join(test_data_dir_path, 'collections')


@pytest.mark.parametrize(('init_kwargs', 'expected_attr'), [
    [
        {
            'url': 'file://media/backup/a',
        },
        {
            'url': 'file://media/backup/a',
        },
    ],
    [
        {
            'url': 'sftp://user@server//media/backup/阿',
        },
        {
            'url': 'sftp://user@server//media/backup/阿',
        },
    ],
])
def test_collection_init(init_kwargs, expected_attr):
    c = duplitab.Collection(**init_kwargs)
    for name, value in expected_attr.items():
        assert getattr(c, name) == value


@pytest.mark.parametrize(('init_kwargs', 'ex_class'), [
    [
        {
        },
        TypeError,
    ],
])
def test_collection_init_fail(init_kwargs, ex_class):
    with pytest.raises(ex_class):
        duplitab.Collection(**init_kwargs)


@pytest.mark.parametrize(('url', 'expected_status'), [
    [
        'file://{}'.format(os.path.join(test_collections_dir_path, 'empty', 'only-full')),
        duplitab._CollectionStatus(
            primary_chain=duplitab._ChainStatus(
                sets=[
                    duplitab._SetStatus(backup_time=datetime.datetime(2016, 10, 27, 19, 57, 33)),
                ]),
        ),
    ],
    [
        'file://{}'.format(os.path.join(test_collections_dir_path, 'empty', 'single-full')),
        duplitab._CollectionStatus(
            primary_chain=duplitab._ChainStatus(
                sets=[
                    duplitab._SetStatus(backup_time=datetime.datetime(2016, 10, 27, 19, 57, 33)),
                    duplitab._SetStatus(backup_time=datetime.datetime(2016, 10, 27, 19, 57, 35)),
                    duplitab._SetStatus(backup_time=datetime.datetime(2016, 10, 27, 19, 57, 39)),
                ]),
        ),
    ],
    [
        'file://{}'.format(os.path.join(test_collections_dir_path, 'empty', 'multiple-full')),
        duplitab._CollectionStatus(
            primary_chain=duplitab._ChainStatus(
                sets=[
                    duplitab._SetStatus(backup_time=datetime.datetime(2016, 10, 27, 19, 57, 47)),
                    duplitab._SetStatus(backup_time=datetime.datetime(2016, 10, 27, 19, 57, 54)),
                ]),
        ),
    ],
    [
        'file://{}'.format(os.path.join(test_collections_dir_path, 'none')),
        duplitab._CollectionStatus(
            primary_chain=None,
        ),
    ],
])
def test_collection_request_status(url, expected_status):
    c = duplitab.Collection(url=url)
    assert expected_status == c.request_status()