from libbe import plugin
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:
        libbe_failures, libbe_tries = doctest.testmod(mod)
        match = True
    mod = plugin.get_plugin("becommands", sys.argv[1])
    if mod is not None:
        becommands_failures, becommands_tries = doctest.testmod(mod)
        match = True
    if not match:
        print "No modules match \"%s\"" % sys.argv[1]
        sys.exit(1)
    else:
        sys.exit(libbe_failures or becommands_failures)
else:
    failed = False
    for module in plugin.iter_plugins("libbe"):
        failures, tries = doctest.testmod(module[1])
        if failures:
            failed = True
    for module in plugin.iter_plugins("becommands"):
        failures, tries = doctest.testmod(module[1])
        if failures:
            failed = True
    sys.exit(failed)