From 2f0f181852627fd0611781a8bfe5f1f1d6e78249 Mon Sep 17 00:00:00 2001
From: Gene Shuman <gene@valimail.com>
Date: Fri, 27 Jan 2017 15:06:49 -0800
Subject: [PATCH] basic dnsplug test

---
 dkim/tests/__init__.py     |  2 ++
 dkim/tests/test_dnsplug.py | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 37 insertions(+)
 create mode 100644 dkim/tests/test_dnsplug.py

diff --git a/dkim/tests/__init__.py b/dkim/tests/__init__.py
index e0c9d63..40cb64a 100644
--- a/dkim/tests/__init__.py
+++ b/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)
diff --git a/dkim/tests/test_dnsplug.py b/dkim/tests/test_dnsplug.py
new file mode 100644
index 0000000..102a151
--- /dev/null
+++ b/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__)
+