Exit with a nonzero status if the tests fail.

1 files changed, 15 insertions, 5 deletions

diff --git a/test.py b/test.py
index 8913c1f..3d7161b 100644
--- a/test.py
+++ b/test.py
@@ -3,18 +3,28 @@ import doctest
 import sys
 if len(sys.argv) > 1:
     match = False
+    libbe_failures = libbe_tries = becommands_failures = becommands_tries = 0
     mod = plugin.get_plugin("libbe", sys.argv[1])
     if mod is not None:
-        doctest.testmod(mod)
+        libbe_failures, libbe_tries = doctest.testmod(mod)
         match = True
     mod = plugin.get_plugin("becommands", sys.argv[1])
     if mod is not None:
-        doctest.testmod(mod)
+        becommands_failures, becommands_tries = doctest.testmod(mod)
         match = True
     if not match:
         print "No modules match \"%s\"" % sys.argv[1]
-else:    
+        sys.exit(1)
+    else:
+        sys.exit(libbe_failures or becommands_failures)
+else:
+    failed = False
     for module in plugin.iter_plugins("libbe"):
-        doctest.testmod(module[1])
+        failures, tries = doctest.testmod(module[1])
+        if failures:
+            failed = True
     for module in plugin.iter_plugins("becommands"):
-        doctest.testmod(module[1])
+        failures, tries = doctest.testmod(module[1])
+        if failures:
+            failed = True
+    sys.exit(failed)