basic dnsplug test

dkim/tests/__init__.py

@@ -30,6 +30,7 @@ def test_suite():
         test_dkim,
         test_util,
         test_arc,
+        test_dnsplug,
         )
     modules = [
         test_canonicalization,
@@ -37,6 +38,7 @@ def test_suite():
         test_dkim,
         test_util,
         test_arc,
+        test_dnsplug,
         ]
     suites = [x.test_suite() for x in modules]
     return unittest.TestSuite(suites)

dkim/tests/test_dnsplug.py

@@ -0,0 +1,35 @@
+# This software is provided 'as-is', without any express or implied
+# warranty.  In no event will the author be held liable for any damages
+# arising from the use of this software.
+#
+# Permission is granted to anyone to use this software for any purpose,
+# including commercial applications, and to alter it and redistribute it
+# freely, subject to the following restrictions:
+#
+# 1. The origin of this software must not be misrepresented; you must not
+#    claim that you wrote the original software. If you use this software
+#    in a product, an acknowledgment in the product documentation would be
+#    appreciated but is not required.
+# 2. Altered source versions must be plainly marked as such, and must not be
+#    misrepresented as being the original software.
+# 3. This notice may not be removed or altered from any source distribution.
+#
+# Copyright (c) 2017 Valimail Inc
+# Contact: Gene Shuman <gene@valimail.com>
+#
+
+import unittest
+import dkim.dnsplug
+
+class TestDNSPlug(unittest.TestCase):
+    
+    def test_get_txt(self):
+        dkim.dnsplug._get_txt = {"in": "out"}.get
+        res = dkim.dnsplug.get_txt(b"in")
+        
+        self.assertEqual(res, b"out")
+
+def test_suite():
+    from unittest import TestLoader
+    return TestLoader().loadTestsFromName(__name__)
+