Newer
Older
postorius / src / postorius / tests / mailman_api_tests / test_models.py
@Simon Hanna Simon Hanna on 15 Mar 2016 2 KB Fix imports
# -*- coding: utf-8 -*-
# Copyright (C) 2012-2015 by the Free Software Foundation, Inc.
#
# This file is part of Postorius.
#
# Postorius is free software: you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
# Postorius is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# Postorius.  If not, see <http://www.gnu.org/licenses/>.

from __future__ import absolute_import, print_function, unicode_literals

from django.contrib.auth.models import User
try:
    from urllib2 import HTTPError
except ImportError:
    from urllib.error import HTTPError

from postorius.tests.utils import ViewTestCase


class ModelTest(ViewTestCase):
    """Tests for the list index page."""

    def setUp(self):
        super(ModelTest, self).setUp()
        self.domain = self.mm_client.create_domain('example.com')
        self.foo_list = self.domain.create_list('foo')

    def tearDown(self):
        for mlist in self.mm_client.lists:
            mlist.delete()
        for user in self.mm_client.users:
            user.delete()
        User.objects.all().delete()
        self.domain.delete()

    def test_mailman_user_not_created_when_flag_is_off(self):
        with self.settings(AUTOCREATE_MAILMAN_USER=False):
            User.objects.create_user('testuser', 'test@example.com', 'testpass')
            with self.assertRaises(HTTPError):
                self.mm_client.get_user('test@example.com')

    def test_mailman_user_created_when_flag_is_on(self):
        with self.settings(AUTOCREATE_MAILMAN_USER=True):
            User.objects.create_user('testuser', 'test@example.com', 'testpass')
            user = self.mm_client.get_user('test@example.com')
            self.assertEqual(str(user.addresses[0]), 'test@example.com')