view example/example.py @ 256:9ec55d2b3e99

adding some .test's
author Catherine Devlin <catherine.devlin@gmail.com>
date Thu, 02 Apr 2009 13:41:58 -0400
parents 07dff0ba981e
children d62bb3dd58a0
line wrap: on
line source

'''A sample application for cmd2.'''

from cmd2 import Cmd, make_option, options, Cmd2TestCase
import unittest, optparse, sys

class CmdLineApp(Cmd):
    multilineCommands = ['orate']
    Cmd.shortcuts.update({'&': 'speak'})
    maxrepeats = 3
    Cmd.settable.append('maxrepeats')

    @options([make_option('-p', '--piglatin', action="store_true", help="atinLay"),
              make_option('-s', '--shout', action="store_true", help="N00B EMULATION MODE"),
              make_option('-r', '--repeat', type="int", help="output [n] times")
             ])
    def do_speak(self, arg, opts=None):
        """Repeats what you tell me to."""
        arg = ''.join(arg)
        if opts.piglatin:
            arg = '%s%say' % (arg[1:], arg[0])
        if opts.shout:
            arg = arg.upper()
        repetitions = opts.repeat or 1
        for i in range(min(repetitions, self.maxrepeats)):
            self.stdout.write(arg)
            self.stdout.write('\n')
            # self.stdout.write is better than "print", because Cmd can be
            # initialized with a non-standard output destination

    do_say = do_speak     # now "say" is a synonym for "speak"
    do_orate = do_speak   # another synonym, but this one takes multi-line input

class TestMyAppCase(Cmd2TestCase):
    CmdApp = CmdLineApp
    transcriptExtension = 'test'

parser = optparse.OptionParser()
parser.add_option('-a', '--alltests', dest='test', action="store_true", 
                  help='Run all transcript tests')
parser.add_option('-t', '--testfile', dest='testfile', metavar='FILE',
                  help='Run a single transcript from file FILE')
(callopts, callargs) = parser.parse_args()
if callopts.testfile or callopts.test:
    CmdLineApp.testfile = callopts.testfile
    sys.argv = [sys.argv[0]]  # the --test argument upsets unittest.main()
    unittest.main()
else:
    app = CmdLineApp()
    app.cmdloop()