Basic removal of duplicate code through using a base class

.pylintrc

@@ -38,7 +38,7 @@ load-plugins=linter_plugin
 # --enable=similarities". If you want to run only the classes checker, but have
 # no Warning level messages displayed, use"--disable=all --enable=classes
 # --disable=W"
-disable=fixme,locally-disabled,abstract-class-not-used,duplicate-code
+disable=fixme,locally-disabled,abstract-class-not-used
 # abstract-class-not-used cannot be disabled locally (at least in pylint 1.4.1)
 
 

letsencrypt/tests/cli_test.py

@@ -6,14 +6,13 @@ import traceback
 import tempfile
 import unittest
 
-import configobj
 import mock
 
 from letsencrypt import account
 from letsencrypt import configuration
 from letsencrypt import errors
-from letsencrypt import storage
 
+from letsencrypt.tests import renewer_test
 from letsencrypt.tests import test_util
 
 
@@ -166,40 +165,13 @@ class DetermineAccountTest(unittest.TestCase):
         self.assertEqual("other email", self.args.email)
 
 
-class DuplicativeCertsTest(unittest.TestCase):
+class DuplicativeCertsTest(renewer_test.BaseRenewableCertTest):
+    """Test to avoid duplicate lineages."""
 
     def setUp(self):
-        # The stuff below is taken from renewer_test.py
-        self.tempdir = tempfile.mkdtemp()
-        self.cli_config = configuration.RenewerConfiguration(
-            namespace=mock.MagicMock(config_dir=self.tempdir))
-        os.makedirs(os.path.join(self.tempdir, "live", "example.org"))
-        os.makedirs(os.path.join(self.tempdir, "archive", "example.org"))
-        os.makedirs(os.path.join(self.tempdir, "configs"))
-        config = configobj.ConfigObj()
-        for kind in storage.ALL_FOUR:
-            config[kind] = os.path.join(self.tempdir, "live", "example.org",
-                                        kind + ".pem")
-        config.filename = os.path.join(self.tempdir, "configs",
-                                       "example.org.conf")
-        config.write()
-        self.config = config
-        self.defaults = configobj.ConfigObj()
-        self.test_rc = storage.RenewableCert(
-            self.config, self.defaults, self.cli_config)
-        for kind in storage.ALL_FOUR:
-            where = getattr(self.test_rc, kind)
-            os.symlink(os.path.join("..", "..", "archive", "example.org",
-                                    "{0}12.pem".format(kind)), where)
-            with open(where, "w") as f:
-                f.write(kind)
-            os.unlink(where)
-            os.symlink(os.path.join("..", "..", "archive", "example.org",
-                                    "{0}11.pem".format(kind)), where)
-            with open(where, "w") as f:
-                f.write(kind)
-
-        # Here we will use test_rc to create duplicative stuff
+        super(DuplicativeCertsTest, self).setUp()
+        self.config.write()
+        self._write_out_ex_kinds()
 
     def tearDown(self):
         shutil.rmtree(self.tempdir)

letsencrypt/tests/renewer_test.py

@@ -32,9 +32,8 @@ def fill_with_sample_data(rc_object):
             f.write(kind)
 
 
-class RenewableCertTests(unittest.TestCase):
-    # pylint: disable=too-many-public-methods
-    """Tests for letsencrypt.renewer.*."""
+class BaseRenewableCertTest(unittest.TestCase):
+
     def setUp(self):
         from letsencrypt import storage
         self.tempdir = tempfile.mkdtemp()
@@ -52,10 +51,30 @@ class RenewableCertTests(unittest.TestCase):
                                         kind + ".pem")
         config.filename = os.path.join(self.tempdir, "configs",
                                        "example.org.conf")
+        self.config = config
 
         self.defaults = configobj.ConfigObj()
         self.test_rc = storage.RenewableCert(
-            config, self.defaults, self.cli_config)
+            self.config, self.defaults, self.cli_config)
+
+    def _write_out_ex_kinds(self):
+        for kind in ALL_FOUR:
+            where = getattr(self.test_rc, kind)
+            os.symlink(os.path.join("..", "..", "archive", "example.org",
+                                    "{0}12.pem".format(kind)), where)
+            with open(where, "w") as f:
+                f.write(kind)
+            os.unlink(where)
+            os.symlink(os.path.join("..", "..", "archive", "example.org",
+                                    "{0}11.pem".format(kind)), where)
+            with open(where, "w") as f:
+                f.write(kind)
+
+class RenewableCertTests(BaseRenewableCertTest):
+    # pylint: disable=too-many-public-methods
+    """Tests for letsencrypt.renewer.*."""
+    def setUp(self):
+        super(RenewableCertTests, self).setUp()
 
     def tearDown(self):
         shutil.rmtree(self.tempdir)
@@ -341,17 +360,8 @@ class RenewableCertTests(unittest.TestCase):
         """Test should_autodeploy() and should_autorenew() on the basis
         of expiry time windows."""
         test_cert = test_util.load_vector("cert.pem")
-        for kind in ALL_FOUR:
-            where = getattr(self.test_rc, kind)
-            os.symlink(os.path.join("..", "..", "archive", "example.org",
-                                    "{0}12.pem".format(kind)), where)
-            with open(where, "w") as f:
-                f.write(kind)
-            os.unlink(where)
-            os.symlink(os.path.join("..", "..", "archive", "example.org",
-                                    "{0}11.pem".format(kind)), where)
-            with open(where, "w") as f:
-                f.write(kind)
+        self._write_out_ex_kinds()
+
         self.test_rc.update_all_links_to(12)
         with open(self.test_rc.cert, "w") as f:
             f.write(test_cert)