gva/gnuviechadmin/managemails/tests/test_models.py

"""
This module contains tests for :py:mod:`managemails.models`
"""
from __future__ import unicode_literals

from mock import patch

from django.test import TestCase, TransactionTestCase
from django.test.utils import override_settings
from django.contrib.auth import get_user_model

from passlib.hash import sha512_crypt

from domains.models import MailDomain
from osusers.models import User

from managemails.models import (
    MailAddress,
    Mailbox,
)

Customer = get_user_model()


@override_settings(
    CELERY_ALWAYS_EAGER=True,
    CELERY_CACHE_BACKEND='memory',
    BROKER_BACKEND='memory'
)
class MailboxTest(TestCase):

    def setUp(self):
        super(MailboxTest, self).setUp()
        self.customer = Customer.objects.create_user('test')

    def test_set_password(self):
        user = User.objects.create_user(self.customer)
        mb = Mailbox.objects.create(username='test', osuser=user)
        mb.set_password('test')
        self.assertTrue(sha512_crypt.verify('test', mb.password))

    def test___str__(self):
        user = User.objects.create_user(self.customer)
        mb = Mailbox.objects.create(username='test', osuser=user)
        mb.set_password('test')
        self.assertEqual(str(mb), 'test')

    @patch('managemails.models.create_file_mailbox')
    def test_save(self, create_file_mailbox_task):
        user = User.objects.create_user(self.customer)
        mb = Mailbox.objects.create_mailbox(user)
        self.assertIsNotNone(mb.pk)
        create_file_mailbox_task.delay.assert_called_with(
            user.username, mb.username)

    @patch('managemails.models.delete_file_mailbox')
    def test_delete(self, delete_file_mailbox_task):
        user = User.objects.create_user(self.customer)
        mb = Mailbox.objects.create_mailbox(user)
        mb.delete()
        self.assertIsNone(mb.pk)
        delete_file_mailbox_task.delay.assert_called_with(
            user.username, mb.username)

    def test_get_mailaddresses(self):
        user = User.objects.create_user(self.customer)
        mb = Mailbox.objects.create_mailbox(user)
        md = MailDomain.objects.create(domain='example.org')
        address = MailAddress.objects.create(localpart='test', domain=md)
        address.set_mailbox(mb)
        mailaddresses = mb.get_mailaddresses()
        self.assertEqual(len(mailaddresses), 1)
        self.assertIn(address, mailaddresses)


@override_settings(
    CELERY_ALWAYS_EAGER=True,
    CELERY_CACHE_BACKEND='memory',
    BROKER_BACKEND='memory'
)
class MailAddressTest(TransactionTestCase):

    def test__str__(self):
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        self.assertEqual(str(ma), 'test@example.org')

    def test_set_mailbox_fresh(self):
        customer = Customer.objects.create_user('test')
        user = User.objects.create_user(customer)
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        mb = Mailbox.objects.create_mailbox(user)
        ma.set_mailbox(mb)
        self.assertIn(ma, mb.get_mailaddresses())

    def test_set_mailbox_reassing(self):
        customer = Customer.objects.create_user('test')
        user = User.objects.create_user(customer)
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        mb = Mailbox.objects.create_mailbox(user)
        ma.set_mailbox(mb)
        mb2 = Mailbox.objects.create_mailbox(user)
        ma.set_mailbox(mb2)
        self.assertIn(ma, mb2.get_mailaddresses())
        self.assertNotIn(ma, mb.get_mailaddresses())

    def test_set_mailbox_with_forwards(self):
        customer = Customer.objects.create_user('test')
        user = User.objects.create_user(customer)
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        mb = Mailbox.objects.create_mailbox(user)
        ma.set_forward_addresses(['test2@example.org'])
        ma.set_mailbox(mb)
        self.assertEqual(ma.mailaddressforward_set.count(), 0)
        self.assertIn(ma, mb.get_mailaddresses())

    def test_set_mailbox_with_unsaved_address(self):
        customer = Customer.objects.create_user('test')
        user = User.objects.create_user(customer)
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress(localpart='test', domain=md)
        mb = Mailbox.objects.create_mailbox(user)
        ma.set_mailbox(mb)
        self.assertIn(ma, mb.get_mailaddresses())

    def test_set_mailbox_fresh_no_commit(self):
        customer = Customer.objects.create_user('test')
        user = User.objects.create_user(customer)
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        mb = Mailbox.objects.create_mailbox(user)
        ma.set_mailbox(mb, commit=False)
        self.assertNotIn(ma, mb.get_mailaddresses())

    def test_set_mailbox_with_unsaved_address_no_commit(self):
        customer = Customer.objects.create_user('test')
        user = User.objects.create_user(customer)
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress(localpart='test', domain=md)
        mb = Mailbox.objects.create_mailbox(user)
        ma.set_mailbox(mb, commit=False)
        self.assertNotIn(ma, mb.get_mailaddresses())

    def test_set_forward_addresses_fresh(self):
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        ma.set_forward_addresses(['test2@example.org'])
        self.assertQuerysetEqual(
            ma.mailaddressforward_set.all(), ['test2@example.org'],
            lambda(maf): maf.target)

    def test_set_forward_addresses_unsaved(self):
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress(localpart='test', domain=md)
        ma.set_forward_addresses(['test2@example.org'])
        self.assertQuerysetEqual(
            ma.mailaddressforward_set.all(), ['test2@example.org'],
            lambda(maf): maf.target)

    def test_set_forward_addresses_replace_forwards(self):
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        ma.set_forward_addresses(['test2@example.org'])
        ma.set_forward_addresses(['test3@example.org'])
        self.assertQuerysetEqual(
            ma.mailaddressforward_set.all(), ['test3@example.org'],
            lambda(maf): maf.target)

    def test_set_forward_addresses_add_forwards(self):
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        ma.set_forward_addresses(['test2@example.org'])
        ma.set_forward_addresses(['test2@example.org', 'test3@example.org'])
        self.assertQuerysetEqual(
            ma.mailaddressforward_set.all(),
            ['test2@example.org', 'test3@example.org'],
            lambda(maf): maf.target,
            ordered=False)

    def test_set_forward_addresses_replace_mailbox(self):
        customer = Customer.objects.create_user('test')
        user = User.objects.create_user(customer)
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        mb = Mailbox.objects.create_mailbox(user)
        ma.set_mailbox(mb)
        ma.set_forward_addresses(['test2@example.org'])
        self.assertNotIn(ma, mb.get_mailaddresses())
        self.assertQuerysetEqual(
            ma.mailaddressforward_set.all(), ['test2@example.org'],
            lambda(maf): maf.target)

    def test_set_forward_addresses_fresh_no_commit(self):
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress.objects.create(localpart='test', domain=md)
        mafwds = ma.set_forward_addresses(['test2@example.org'], commit=False)
        self.assertEqual(ma.mailaddressforward_set.count(), 0)
        self.assertEqual(mafwds[0].target, 'test2@example.org')

    def test_set_forward_address_unsaved_no_commit(self):
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress(localpart='test', domain=md)
        mafwds = ma.set_forward_addresses(['test2@example.org'], commit=False)
        self.assertEqual(ma.mailaddressforward_set.count(), 0)
        self.assertEqual(mafwds[0].target, 'test2@example.org')


@override_settings(
    CELERY_ALWAYS_EAGER=True,
    CELERY_CACHE_BACKEND='memory',
    BROKER_BACKEND='memory'
)
class MailboxManagerTest(TransactionTestCase):

    def setUp(self):
        super(MailboxManagerTest, self).setUp()
        self.customer = Customer.objects.create_user('test')
        self.user = User.objects.create_user(self.customer)

    def test_get_next_mailbox_name_fresh(self):
        mailboxname = Mailbox.objects.get_next_mailbox_name(self.user)
        self.assertEqual(mailboxname, '{}p01'.format(self.user.username))

    def test_get_next_mailbox_name_second(self):
        Mailbox.objects.create_mailbox(self.user)
        mailboxname = Mailbox.objects.get_next_mailbox_name(self.user)
        self.assertEqual(mailboxname, '{}p02'.format(self.user.username))

    def test_get_next_mailbox_name_gap_detection(self):
        mailboxes = [
            Mailbox.objects.create_mailbox(self.user) for i in range(3)
        ]
        mailboxes[1].delete()
        mailboxname = Mailbox.objects.get_next_mailbox_name(self.user)
        self.assertEqual(mailboxname, '{}p02'.format(self.user.username))

    def test_unused_or_own_fresh(self):
        md = MailDomain.objects.create(domain='example.org')
        address = MailAddress.objects.create(localpart='test', domain=md)
        mailboxes = Mailbox.objects.unused_or_own(address, self.user)
        self.assertQuerysetEqual(mailboxes, [])

    def test_unused_or_own_unassigned(self):
        md = MailDomain.objects.create(domain='example.org')
        address = MailAddress.objects.create(localpart='test', domain=md)
        mailboxes = [
            Mailbox.objects.create_mailbox(self.user) for i in range(2)
        ]
        assignable = Mailbox.objects.unused_or_own(address, self.user)
        self.assertQuerysetEqual(assignable, [repr(mb) for mb in mailboxes])

    def test_unused_or_own_assigned(self):
        md = MailDomain.objects.create(domain='example.org')
        address = MailAddress.objects.create(localpart='test', domain=md)
        mailboxes = [
            Mailbox.objects.create_mailbox(self.user) for i in range(2)
        ]
        address.set_mailbox(mailboxes[0])
        assignable = Mailbox.objects.unused_or_own(address, self.user)
        self.assertQuerysetEqual(assignable, [repr(mb) for mb in mailboxes])

    def test_unused_or_own_assigned_other(self):
        md = MailDomain.objects.create(domain='example.org')
        address = MailAddress.objects.create(localpart='test', domain=md)
        address2 = MailAddress.objects.create(localpart='test2', domain=md)
        mailboxes = [
            Mailbox.objects.create_mailbox(self.user) for i in range(2)
        ]
        address2.set_mailbox(mailboxes[0])
        assignable = Mailbox.objects.unused_or_own(address, self.user)
        self.assertQuerysetEqual(assignable, [repr(mailboxes[1])])

    def test_unused_fresh(self):
        mailboxes = Mailbox.objects.unused(self.user)
        self.assertQuerysetEqual(mailboxes, [])

    def test_unused_unassigned(self):
        mailbox = Mailbox.objects.create_mailbox(self.user)
        mailboxes = Mailbox.objects.unused(self.user)
        self.assertQuerysetEqual(mailboxes, [repr(mailbox)])

    def test_unused_assigned(self):
        md = MailDomain.objects.create(domain='example.org')
        address = MailAddress.objects.create(localpart='test', domain=md)
        mailboxes = [
            Mailbox.objects.create_mailbox(self.user) for i in range(2)
        ]
        address.set_mailbox(mailboxes[0])
        assignable = Mailbox.objects.unused(self.user)
        self.assertQuerysetEqual(assignable, [repr(mailboxes[1])])

    def test_create_mailbox_no_password(self):
        mailbox = Mailbox.objects.create_mailbox(self.user)
        self.assertEqual(mailbox.osuser, self.user)
        self.assertEqual(mailbox.username, '{}p01'.format(self.user.username))
        self.assertEqual(mailbox.password, '')

    def test_create_mailbox_with_password(self):
        mailbox = Mailbox.objects.create_mailbox(self.user, 'test')
        self.assertEqual(mailbox.osuser, self.user)
        self.assertEqual(mailbox.username, '{}p01'.format(self.user.username))
        self.assertTrue(sha512_crypt.verify('test', mailbox.password))


@override_settings(
    CELERY_ALWAYS_EAGER=True,
    CELERY_CACHE_BACKEND='memory',
    BROKER_BACKEND='memory'
)
class MailAddressMailboxTest(TestCase):

    def setUp(self):
        super(MailAddressMailboxTest, self).setUp()
        self.customer = Customer.objects.create_user('test')

    def test___str__(self):
        user = User.objects.create_user(self.customer)
        md = MailDomain.objects.create(domain='example.org')
        ma = MailAddress(localpart='test', domain=md)
        mb = Mailbox.objects.create_mailbox(user)
        ma.set_mailbox(mb)
        self.assertEqual(str(ma.mailaddressmailbox), mb.username)