summaryrefslogtreecommitdiff
path: root/src/mailman/database/tests/test_migrations.py
blob: 5333083f2cb9938398d0d57f19adc2d9c6f3b1b2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
# Copyright (C) 2012 by the Free Software Foundation, Inc.
#
# This file is part of GNU Mailman.
#
# GNU Mailman 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.
#
# GNU Mailman 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
# GNU Mailman.  If not, see <http://www.gnu.org/licenses/>.

"""Test schema migrations."""

from __future__ import absolute_import, print_function, unicode_literals

__metaclass__ = type
__all__ = [
    'TestMigration20120407',
    ]


import os
import shutil
import sqlite3
import tempfile
import unittest

from zope.component import getUtility

from mailman.app.lifecycle import create_list
from mailman.config import config
from mailman.interfaces.domain import IDomainManager
from mailman.testing.helpers import configuration, temporary_db
from mailman.testing.layers import ConfigLayer
from mailman.utilities.modules import call_name



class TestMigration20120407(unittest.TestCase):
    """Test the dated migration (LP: #971013)

    Circa: 3.0b1 -> 3.0b2

    table mailinglist:
    * news_moderation -> newsgroup_moderation
    * news_prefix_subject_too -> nntp_prefix_subject_too
    * ADD archive_policy
    * REMOVE archive
    * REMOVE archive_private
    * REMOVE archive_volume_frequency
    * REMOVE nntp_host
    """

    layer = ConfigLayer

    def setUp(self):
        self._tempdir = tempfile.mkdtemp()

    def tearDown(self):
        shutil.rmtree(self._tempdir)

    def test_sqlite_base(self):
        # Test the SQLite base schema.
        url = 'sqlite:///' + os.path.join(self._tempdir, 'mailman.db')
        database_class = config.database['class']
        database = call_name(database_class)
        with configuration('database', url=url):
            database.initialize()
        # Load all the database SQL to just before ours.
        database.load_migrations('20120406999999')
        # Populate the test database with a domain and a mailing list.
        with temporary_db(database):
            getUtility(IDomainManager).add(
                'example.com', 'An example domain.',
                'http://lists.example.com', 'postmaster@example.com')
            mlist = create_list('test@example.com')
            del mlist
        database.commit()
        # Verify that the database has not yet been migrated.
        for missing in ('archive_policy',
                        'nntp_prefix_subject_too'):
            self.assertRaises(sqlite3.OperationalError,
                              database.store.execute,
                              'select {0} from mailinglist;'.format(missing))
        for present in ('archive',
                        'archive_private',
                        'archive_volume_frequency',
                        'news_moderation',
                        'news_prefix_subject_too',
                        'nntp_host'):
            # This should not produce an exception.  Is there some better test
            # that we can perform?
            database.store.execute(
                'select {0} from mailinglist;'.format(present))

    def test_sqlite_migration(self):
        # Test the SQLite base schema.
        url = 'sqlite:///' + os.path.join(self._tempdir, 'mailman.db')
        database_class = config.database['class']
        database = call_name(database_class)
        with configuration('database', url=url):
            database.initialize()
        # Load all the database SQL to just before ours.
        database.load_migrations('20120406999999')
        database.commit()
        # Load all migrations, up to and including this one.
        database.load_migrations('20120407000000')
        # Verify that the database has been migrated.
        for present in ('archive_policy',
                        'nntp_prefix_subject_too'):
            # This should not produce an exception.  Is there some better test
            # that we can perform?
            database.store.execute(
                'select {0} from mailinglist;'.format(present))
        for missing in ('archive',
                        'archive_private',
                        'archive_volume_frequency',
                        'news_moderation',
                        'news_prefix_subject_too',
                        'nntp_host'):
            self.assertRaises(sqlite3.OperationalError,
                              database.store.execute,
                              'select {0} from mailinglist;'.format(missing))