Add a test module that runs assertions against all modules loaded by the…

Add a test module that runs assertions against all modules loaded by the interpreter. Add one test for legal info keys.

routersploit/test/test_modules.py

+from inspect import getmodule
+from itertools import chain
+from unittest import main, TestCase, TestSuite
+
+from routersploit.interpreter import RoutersploitInterpreter
+
+
+class ModuleTestBase(TestCase):
+    """A test case that every module must pass.
+
+    Attributes:
+        module (module): The exploit instance of the module being tested.
+        metadata (Dict): The info associated with the module.
+    """
+
+    def test_legal_metadata_keys(self):
+
+        legal_keys = set([
+            "name",
+            "description",
+            "targets",
+            "authors",
+            "references"])
+
+        self.assertTrue(set(self.metadata.keys()).issubset(legal_keys))
+
+
+def load_tests(loader, tests, pattern):
+    """Map every module to a test case, and group them into a suite."""
+
+    def tests():
+
+        # let interpreter load modules as it sees fit
+        interpreter = RoutersploitInterpreter()
+
+        for module_path in interpreter.modules:
+
+            # use the given module
+            interpreter.command_use(module_path)
+
+            class ModuleTest(ModuleTestBase):
+
+                # bind module and metadata
+                module = interpreter.current_module
+                metadata = interpreter.module_metadata
+
+                def shortDescription(self):
+                    return 'testing %s' % getmodule(self.module).__name__
+
+            # yield the tests from this test case
+            yield loader.loadTestsFromTestCase(ModuleTest)
+
+    suite = TestSuite()
+    suite.addTests(chain(*tests()))
+    return suite
+
+if __name__ == '__main__':
+    main()