Add configurable lifetime of key change request.

2 files changed, 6 insertions, 0 deletions

diff --git a/src/mailman_pgp/pgp/tests/base.py b/src/mailman_pgp/pgp/tests/base.py
index 52e5ec4..47d333c 100644
--- a/src/mailman_pgp/pgp/tests/base.py
+++ b/src/mailman_pgp/pgp/tests/base.py
@@ -25,6 +25,8 @@ from mailman.email.message import Message
 from pgpy import PGPKey
 from pkg_resources import resource_string
 
+from mailman_pgp.testing.layers import PGPLayer
+
 
 def load_message(path):
     data = resource_string('mailman_pgp.pgp.tests',
@@ -40,6 +42,7 @@ def load_key(path):
 
 
 class WrapperTestCase(TestCase):
+    layer = PGPLayer
     wrapper = None
 
     def wrap(self, message):
diff --git a/src/mailman_pgp/pgp/tests/test_keygen.py b/src/mailman_pgp/pgp/tests/test_keygen.py
index f2f72a7..a3a5499 100644
--- a/src/mailman_pgp/pgp/tests/test_keygen.py
+++ b/src/mailman_pgp/pgp/tests/test_keygen.py
@@ -25,9 +25,12 @@ from pgpy import PGPKey
 from pgpy.constants import EllipticCurveOID, PubKeyAlgorithm
 
 from mailman_pgp.pgp.keygen import ListKeyGenerator
+from mailman_pgp.testing.layers import PGPLayer
 
 
 class TestKeygen(TestCase):
+    layer = PGPLayer
+
     def setUp(self):
         self.display_name = 'Display Name'
         self.posting_address = 'posting@address.com'