RD tests: use established list-missing-modules mechanism from defaults

NovapaX · Nov 4, 2017 · 11043bc · 11043bc
1 parent 04f9400
commit 11043bc
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/test_rd_examples.py b/tests/test_rd_examples.py
@@ -17,13 +17,11 @@
 
 from .test_server import WithAsyncLoop, Destructing, WithClient
 
-try:
+linkheader_modules = aiocoap.defaults.linkheader_missing_modules()
+_skip_unless_linkheader = unittest.skipIf(linkheader_modules, "Modules missing for running RD tests: %s"%(linkheader_modules,))
+if not linkheader_modules:
     import link_header
     import aiocoap.cli.rd
-except ImportError:
-    link_header = None
-
-_skip_unless_linkheader = unittest.skipIf(link_header is None, "RD tests require link header extension")
 
 class WithResourceDirectory(WithAsyncLoop, Destructing):
     rd_address = '::1'