Added unit testing framework

2 files changed, 45 insertions, 3 deletions

diff --git a/libbe/cmdutil.py b/libbe/cmdutil.py
index 891e273..3fbfb50 100644
--- a/libbe/cmdutil.py
+++ b/libbe/cmdutil.py
@@ -55,6 +55,17 @@ def iter_commands():
         yield name.replace("_", "-"), module
 
 def get_command(command_name):
+    """Retrieves the module for a user command
+
+    >>> get_command("asdf")
+    Traceback (most recent call last):
+    File "<stdin>", line 1, in ?
+    File "/home/abentley/be/libbe/cmdutil.py", line 60, in get_command
+      raise UserError("Unknown command %s" % command_name)
+    UserError: Unknown command asdf
+    >>> get_command("list")
+    <module 'becommands.list' from '/home/abentley/be/becommands/list.pyc'>
+    """
     cmd = plugin.get_plugin("becommands", command_name.replace("-", "_"))
     if cmd is None:
         raise UserError("Unknown command %s" % command_name)
@@ -65,3 +76,11 @@ def execute(cmd, args):
 
 def help(cmd, args):
     return get_command(cmd).help()
+
+def _test():
+    import doctest
+    import sys
+    doctest.testmod()
+
+if __name__ == "__main__":
+    _test()
diff --git a/libbe/plugin.py b/libbe/plugin.py
index 2dedac4..5f0fa4d 100644
--- a/libbe/plugin.py
+++ b/libbe/plugin.py
@@ -9,7 +9,13 @@ def my_import(mod_name):
     return module
 
 def iter_plugins(prefix):
-    modfiles = os.listdir(os.path.join(sys.path[0], prefix))
+    """
+    >>> "list" in [n for n,m in iter_plugins("becommands")]
+    True
+    >>> "plugin" in [n for n,m in iter_plugins("libbe")]
+    True
+    """
+    modfiles = os.listdir(os.path.join(plugin_path, prefix))
     modfiles.sort()
     for modfile in modfiles:
         if modfile.endswith(".py") and modfile != "__init__.py":
@@ -17,9 +23,26 @@ def iter_plugins(prefix):
 
 
 def get_plugin(prefix, name):
+    """
+    >>> get_plugin("becommands", "asdf") is None
+    True
+    >>> get_plugin("becommands", "list")
+    <module 'becommands.list' from '/home/abentley/be/becommands/list.pyc'>
+    """
     dirprefix = '/'.join(prefix.split('.'))
-    command_path = os.path.join(sys.path[0], dirprefix, name+".py")
+    command_path = os.path.join(plugin_path, dirprefix, name+".py")
     if os.path.isfile(command_path):
         return my_import(prefix + "." + name)
     return None
- 
+
+plugin_path = sys.path[0]
+while not os.path.isfile(os.path.join(plugin_path, "libbe/plugin.py")):
+    plugin_path = os.path.realpath(os.path.dirname(plugin_path))
+if plugin_path not in sys.path:
+    sys.path.append(plugin_path)
+def _test():
+    import doctest
+    doctest.testmod()
+
+if __name__ == "__main__":
+    _test()