blob: 07eb3864d17c218372a15241a1f82f7c78afe094 (
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
|
# Copyright (C) 2017 Jan Jancar
#
# This file is a part of the Mailman PGP plugin.
#
# This program 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.
#
# This program 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
# this program. If not, see <http://www.gnu.org/licenses/>.
import contextlib
import os
from os.path import isfile
from mailman.testing.layers import (ConfigLayer, RESTLayer, SMTPLayer)
from sqlalchemy.exc import SQLAlchemyError
from mailman_pgp.config import config
from mailman_pgp.database import transaction
from mailman_pgp.model.base import Base
def reset_rollback():
config.db.session.rollback()
def reset_pgp_dirs():
for keydir in (config.pgp.keydir_config.values()):
for path in os.listdir(keydir):
full_path = os.path.join(keydir, path)
if isfile(full_path):
os.remove(full_path)
def reset_pgp_hard():
reset_rollback()
reset_pgp_dirs()
with transaction():
Base.metadata.drop_all(config.db.engine)
Base.metadata.create_all(config.db.engine)
def reset_pgp_soft():
reset_rollback()
reset_pgp_dirs()
with contextlib.closing(config.db.engine.connect()) as con:
trans = con.begin()
for table in reversed(Base.metadata.sorted_tables):
try:
con.execute(table.delete())
except SQLAlchemyError:
pass
trans.commit()
# It's weird that ws have to do this, but for some reason nose2 test layers
# don't work when ws create a mixin class with the two classmethods
# and subclass both it and the respective Mailman Core test layer.
class PGPConfigLayer(ConfigLayer):
@classmethod
def tearDown(cls):
reset_pgp_soft()
@classmethod
def testTearDown(cls):
reset_pgp_soft()
class PGPMigrationLayer(ConfigLayer):
@classmethod
def tearDown(cls):
reset_pgp_hard()
@classmethod
def testTearDown(cls):
reset_pgp_hard()
class PGPSMTPLayer(SMTPLayer):
@classmethod
def tearDown(cls):
reset_pgp_soft()
@classmethod
def testTearDown(cls):
reset_pgp_soft()
class PGPRESTLayer(RESTLayer):
@classmethod
def tearDown(cls):
reset_pgp_soft()
@classmethod
def testTearDown(cls):
reset_pgp_soft()
|
