~abentley/bzrtools/bzrtools.dev

# Copyright (C) 2008 Canonical Limited.

#

#    This program is free software; you can redistribute it and/or modify

#    it under the terms of the GNU General Public License as published by

#    the Free Software Foundation; either version 2 of the License, or

#    (at your option) any later version.

#

#    This program is distributed in the hope that it will be useful,

#    but WITHOUT ANY WARRANTY; without even the implied warranty of

#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

#    GNU General Public License for more details.

#

#    You should have received a copy of the GNU General Public License

#    along with this program; if not, write to the Free Software

#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

from bzrlib.commands import plugin_cmds

commands = {

    'cmd_branches': [],

    'cmd_branch_history': [],

    'cmd_cbranch': [],

    'cmd_cdiff': [],

    'cmd_clean_tree': [],

    'cmd_fetch_ghosts': ['fetch-missing'],

    'cmd_graph_ancestry': [],

    'cmd_heads': [],

    'cmd_import': [],

    'cmd_link_tree': [],

    'cmd_multi_pull': [],

    'cmd_patch': [],

    'cmd_rspush': [],

    'cmd_shelf': [],

    'cmd_shell': [],

    'cmd_shelve1': ['shelve'],

    'cmd_trees': [],

    'cmd_unshelve1': ['unshelve'],

    'cmd_zap': [],

}

for cmd_name, aliases in commands.items():

    plugin_cmds.register_lazy(cmd_name, aliases,

                              'bzrlib.plugins.bzrtools.command_classes')

def test_suite():

    from bzrlib.tests.TestUtil import TestLoader

    import tests

    from doctest import DocTestSuite, ELLIPSIS

    from unittest import TestSuite

    import bzrtools

    import tests.clean_tree

    import tests.test_dotgraph

    import tests.is_clean

    import tests.test_cbranch

    import tests.test_link_tree

    import tests.test_patch

    import tests.test_rspush

    import tests.upstream_import

    import zap

    import tests.blackbox

    import tests.shelf_tests

    result = TestSuite()

    result.addTest(DocTestSuite(bzrtools, optionflags=ELLIPSIS))

    result.addTest(tests.clean_tree.test_suite())

    result.addTest(tests.test_suite())

    result.addTest(TestLoader().loadTestsFromModule(tests.shelf_tests))

    result.addTest(tests.blackbox.test_suite())

    result.addTest(tests.upstream_import.test_suite())

    result.addTest(zap.test_suite())

    result.addTest(TestLoader().loadTestsFromModule(tests.test_dotgraph))

    result.addTest(TestLoader().loadTestsFromModule(tests.is_clean))

    result.addTest(TestLoader().loadTestsFromModule(tests.test_link_tree))

    result.addTest(TestLoader().loadTestsFromModule(tests.test_patch))

    result.addTest(TestLoader().loadTestsFromModule(tests.test_rspush))

    result.addTest(TestLoader().loadTestsFromModule(tests.test_cbranch))

    return result