From e75bafa43908d9f66f94ffae1393f18c9262f5a1 Mon Sep 17 00:00:00 2001
From: Daniel Wilcox <dmw@yubasolutions.com>
Date: Thu, 28 Apr 2016 12:27:49 -0700
Subject: [PATCH] Add basic test for get_all_certs_keys IInstaller interface
 method.

---
 .../TestPostfixConfigGenerator.py             | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/letsencrypt-postfix/TestPostfixConfigGenerator.py b/letsencrypt-postfix/TestPostfixConfigGenerator.py
index e1d921c6f..ebcf12a4c 100644
--- a/letsencrypt-postfix/TestPostfixConfigGenerator.py
+++ b/letsencrypt-postfix/TestPostfixConfigGenerator.py
@@ -24,6 +24,12 @@ mydomain = fubard.org
 myorigin = fubard.org"""
 
 
+certs_only_config = """
+smtpd_tls_cert_file = /etc/letsencrypt/live/www.fubard.org/fullchain.pem
+smtpd_tls_key_file = /etc/letsencrypt/live/www.fubard.org/privkey.pem
+"""
+
+
 def GetFakeOpen(fake_file_contents):
     fake_file = io.StringIO()
     # cast this to unicode for py2
@@ -40,6 +46,7 @@ class TestPostfixConfigGenerator(unittest.TestCase):
 
     def setUp(self):
         self.fopen_names_only_config = GetFakeOpen(names_only_config)        
+        self.fopen_certs_only_config = GetFakeOpen(certs_only_config)
         #self.config = Config.Config()
         self.config = None
         self.postfix_dir = 'tests/'
@@ -57,6 +64,18 @@ class TestPostfixConfigGenerator(unittest.TestCase):
         )
         self.assertEqual(sorted_names, postfix_config_gen.get_all_names())
 
+    def testGetAllCertAndKeys(self):
+        return_vals = ('/etc/letsencrypt/live/www.fubard.org/fullchain.pem',
+                       '/etc/letsencrypt/live/www.fubard.org/privkey.pem',
+                       None)
+        postfix_config_gen = pcg.PostfixConfigGenerator(
+            self.config,
+            self.postfix_dir,
+            fixup=True,
+            fopen=self.fopen_certs_only_config
+        )
+        self.assertEqual(return_vals, postfix_config_gen.get_all_certs_keys())
+
 
 if __name__ == '__main__':
     unittest.main()