Added tests for set-root

1 files changed, 21 insertions, 0 deletions

diff --git a/becommands/set_root.py b/becommands/set_root.py
index ff7662a..87e4bda 100644
--- a/becommands/set_root.py
+++ b/becommands/set_root.py
@@ -2,6 +2,27 @@
 from libbe import bugdir, cmdutil, rcs
 
 def execute(args):
+    """
+    >>> from libbe import tests
+    >>> dir = tests.Dir()
+    >>> try:
+    ...     bugdir.tree_root(dir.name)
+    ... except bugdir.NoBugDir, e:
+    ...     True
+    True
+    >>> execute([dir.name])
+    No revision control detected.
+    Directory initialized.
+    >>> bd = bugdir.tree_root(dir.name)
+    >>> bd.root = dir.name
+    >>> dir = tests.arch_dir()
+    >>> execute([dir.name+"/{arch}"])
+    Using Arch for revision control.
+    Directory initialized.
+    >>> bd = bugdir.tree_root(dir.name+"/{arch}")
+    >>> bd.root = dir.name
+    >>> tests.clean_up()
+    """
     if len(args) != 1:
         raise cmdutil.UserError("Please supply a directory path")
     dir_rcs = rcs.detect(args[0])