From: Bob Halley <halley@dnspython.org>
Date: Tue, 4 Aug 2020 13:05:00 +0000 (-0700)
Subject: add enumeration checking helpers
X-Git-Tag: v2.1.0rc1~112^2~1
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=b70e664ff050b30d2fe40ee041b102fa240827eb;p=thirdparty%2Fdnspython.git

add enumeration checking helpers
---

diff --git a/tests/util.py b/tests/util.py
index df736df0..644a88a1 100644
--- a/tests/util.py
+++ b/tests/util.py
@@ -15,7 +15,26 @@
 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
 # OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 
+import enum
+import inspect
 import os.path
 
 def here(filename):
     return os.path.join(os.path.dirname(__file__), filename)
+
+def enumerate_module(module, super_class):
+    """Yield module attributes which are subclasses of given class"""
+    for attr_name in dir(module):
+        attr = getattr(module, attr_name)
+        if inspect.isclass(attr) and issubclass(attr, super_class):
+            yield attr
+
+def check_enum_exports(module, eq_callback, only=None):
+    """Make sure module exports all mnemonics from enums"""
+    for attr in enumerate_module(module, enum.Enum):
+        if only is not None and attr not in only:
+            print('SKIP', attr)
+            continue
+        for flag, value in attr.__members__.items():
+            print(module, flag, value)
+            eq_callback(getattr(module, flag), value)