test_transfer_mail.py

from collab_admin_kit import transfer_mail
import os
import pytest
import yaml

data_dir = os.path.join(os.path.dirname(__file__) + '/data/')
config = data_dir + 'test_cak_config.yml'


def test_transfer_label_list():
    args = {
        'current_owner': 'testuser',
        'new_owner': 'testuser',
        'labels': ['label1', 'label2'],
        'label_prefix': None,
        'email_directory': None
    }
    mover = transfer_mail.Mover(
        args,
        yaml.load(open(config), Loader=yaml.BaseLoader)
    )
    mover.transfer_mail()
    mover.cleanup()


def test_transfer_label_file():
    args = {
        'current_owner': 'testuser',
        'new_owner': 'testuser',
        'label_file': data_dir + '/mock_gyb/test_transfer_mail.txt',
        'label_prefix': None,
        'email_directory': None
    }
    mover = transfer_mail.Mover(
        args,
        yaml.load(open(config), Loader=yaml.BaseLoader)
    )
    mover.transfer_mail()
    mover.cleanup()


def test_prefix_assignment():
    args = {
        'current_owner': 'testuser',
        'new_owner': 'testuser',
        'labels': ['label1', 'label2'],
        'label_prefix': 'prefix',
        'email_directory': None
    }
    mover = transfer_mail.Mover(
        args,
        yaml.load(open(config), Loader=yaml.BaseLoader)
    )
    assert mover.transfer['label1']['destination'] == 'prefix-label1'
    mover.cleanup()


def test_search_translation():
    args = {
        'current_owner': 'testuser',
        'new_owner': 'testuser',
        'label_file': data_dir + '/mock_gyb/test_transfer_mail.txt',
        'label_prefix': None,
        'email_directory': None
    }
    mover = transfer_mail.Mover(
        args,
        yaml.load(open(config), Loader=yaml.BaseLoader)
    )
    print(mover.transfer)
    expected = 'weir---l-bel-'
    assert mover.transfer['weir|) l^bel(']['dest_searchable'] == expected
    mover.cleanup()


def test_with_existing_custom_dir():
    args = {
        'current_owner': 'testuser',
        'new_owner': 'testuser',
        'labels': ['label1', 'label2'],
        'label_prefix': None,
        'email_directory': '/var/tmp'
    }
    mover = transfer_mail.Mover(
        args,
        yaml.load(open(config), Loader=yaml.BaseLoader)
    )
    mover.transfer_mail()
    mover.cleanup()


def test_with_new_custom_dir():
    args = {
        'current_owner': 'testuser',
        'new_owner': 'testuser',
        'labels': ['label1', 'label2'],
        'label_prefix': None,
        'email_directory': '/var/tmp/gtm1a2b3c4d'
    }
    mover = transfer_mail.Mover(
        args,
        yaml.load(open(config), Loader=yaml.BaseLoader)
    )
    mover.transfer_mail()
    if not os.path.isdir('/var/tmp/gtm1a2b3c4d'):
        assert False
    mover.cleanup()
    if os.path.isdir('/var/tmp/gtm1a2b3c4d'):
        assert False


def test_transfer_with_bad_current_owner():
    with pytest.raises(SystemExit):
        args = {
            'current_owner': 'notauser',
            'new_owner': 'testuser',
            'labels': ['label1', 'label2'],
            'label_prefix': None
        }
        mover = transfer_mail.Mover(
            args,
            yaml.load(open(config), Loader=yaml.BaseLoader)
        )
        mover.transfer_mail()
        mover.cleanup()


def test_transfer_with_bad_new_owner():
    with pytest.raises(SystemExit):
        args = {
            'current_owner': 'testuser',
            'new_owner': 'notauser',
            'labels': ['label1', 'label2'],
            'label_prefix': None
        }
        mover = transfer_mail.Mover(
            args,
            yaml.load(open(config), Loader=yaml.BaseLoader)
        )
        mover.transfer_mail()
        mover.cleanup()